提交 70f0b2eb authored 作者: carriepl's avatar carriepl

Fix call to gpu_from_host

上级 9d6615eb
......@@ -156,7 +156,7 @@ def traverse(out, x, x_copy, d, visited=None):
d[out] = cuda.gpu_from_host(x_copy)
else:
assert isinstance(x.type, gpuarray.GpuArrayType)
d[out] = gpuarray.gpu_from_host(x_copy)
d[out] = gpuarray.GpuFromHost(x.type.context_name)(x_copy)
return d
elif out.owner is None:
return d
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论