This is on the current/old back-end. It is still limited to float32.


On Fri, Mar 28, 2014 at 2:04 PM, abergeron <notifications@github.com> wrote:

> In theano/sandbox/cuda/opt.py:
>
> > @@ -1535,6 +1535,15 @@ def local_gpu_extract_diagonal(node):
> >                  gpu_from_host(diag_node.inputs[0]))]
> >      return False
> >
> > +@register_opt()
> > +@local_optimizer([tensor.nnet.GroupDot])
> > +def gd_to_gpu(node):
> > +    return [host_from_gpu(GpuGroupDot(node.op.n_groups)(*node.inputs))]
>
> I'll rework it to add a check. Also, I should check for the dtypes of the
> inputs since we are no longer limited to float32 only.
>
> --
> Reply to this email directly or view it on GitHub<https://github.com/Theano/Theano/pull/1781/files#r11077935>
> .
>