提交 baff091a authored 作者: Frederic Bastien's avatar Frederic Bastien

small fix to put more stuff on the gpu.

上级 741e24dc
......@@ -343,7 +343,7 @@ def local_gpu_incsubtensor(node):
x, y = host_output.owner.inputs[0:2]
coords = host_output.owner.inputs[2:]
return [GpuIncSubtensor(incsubt.idx_list, inplace=incsubt.inplace,
set_instead_of_inc=incsubt.op.set_instead_of_inc)(
set_instead_of_inc=incsubt.set_instead_of_inc)(
gpu_from_host(x),
gpu_from_host(y),
*coords)]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论