提交 8a6e065f authored 作者: sentient07's avatar sentient07 提交者: Reyhane Askari

Reverted subtensor to old version

上级 3db235a7
...@@ -983,7 +983,7 @@ def local_gpua_subtensor(op, context_name, inputs, outputs): ...@@ -983,7 +983,7 @@ def local_gpua_subtensor(op, context_name, inputs, outputs):
for n, _ in outputs[0].clients]): for n, _ in outputs[0].clients]):
return return
else: else:
return [host_from_gpu(gpu_x.owner.op(outputs[0]))] return [gpu_x.owner.op(outputs[0]).transfer('cpu')]
return GpuSubtensor(op.idx_list) return GpuSubtensor(op.idx_list)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论