提交 25e6df0f authored 作者: abergeron's avatar abergeron

Merge pull request #3881 from nouiz/gpu_dot22

Fix opt warning/not moving to GPU some node.
...@@ -8,7 +8,6 @@ from theano.tensor.basic import as_tensor_variable ...@@ -8,7 +8,6 @@ from theano.tensor.basic import as_tensor_variable
from theano.tensor.opt import in2out from theano.tensor.opt import in2out
from .basic_ops import as_gpuarray_variable, infer_context_name from .basic_ops import as_gpuarray_variable, infer_context_name
from .opt_util import inplace_allocempty from .opt_util import inplace_allocempty
try: try:
...@@ -281,7 +280,9 @@ class GpuDot22(BlasOp): ...@@ -281,7 +280,9 @@ class GpuDot22(BlasOp):
assert x.ndim == 2 assert x.ndim == 2
assert y.ndim == 2 assert y.ndim == 2
assert x.dtype == y.dtype assert x.dtype == y.dtype
return Apply(self, [x, y], [x.type()]) otype = x.type.clone(
broadcastable=(x.type.broadcastable[0], y.type.broadcastable[1]))
return Apply(self, [x, y], [otype()])
def perform(self, node, inputs, outputs): def perform(self, node, inputs, outputs):
x, y = inputs x, y = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论