It is only EquilibriumOptimizer that do this, not the object created by
local_optimizer.


On Fri, Mar 28, 2014 at 1:32 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))]
>
> Normally local_optimizer makes sure of that.
>
> --
> Reply to this email directly or view it on GitHub<https://github.com/Theano/Theano/pull/1781/files#r11076480>
> .
>