提交 efabc175 authored 作者: Reyhane Askari's avatar Reyhane Askari

minor change

上级 53e9b5fc
......@@ -262,12 +262,10 @@ def op_lifter(OP, cuda_only=False):
new_outputs = [new_op]
to_cpu_fn = lambda x: x.transfer('cpu')
# copy stack traces onto gpu outputs
for old_output, new_output in zip(node.outputs, new_outputs):
copy_stack_trace(old_output, new_output)
new_outputs = [to_cpu_fn(o) for o in new_outputs]
# also copy the stack traces onto HostFromGpu outputs
for old_output, new_output in zip(node.outputs, new_outputs):
copy_stack_trace(old_output, new_output)
copy_stack_trace(old_output, to_cpu_fn(new_output))
return new_outputs
return False
local_opt.__name__ = maker.__name__
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论