提交 a7ac77a5 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3695 from carriepl/scan_gpuarray2

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