提交 d24e99ff authored 作者: Frederic's avatar Frederic 提交者: Arnaud Bergeron

Fix gpuarray op_lifter when the gpu op was inheriting from the OP it replace.

上级 1eb1adc7
......@@ -56,7 +56,8 @@ def op_lifter(OP):
all([c != 'output' and c.op == gpu_from_host
for c, idx in node.outputs[0].clients])):
new_op = maker(node)
if new_op:
# This is needed as sometimes new_op inherit from OP.
if new_op and new_op != node.op:
return [host_from_gpu(new_op(*node.inputs))]
return False
local_opt.__name__ = maker.__name__
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论