提交 cdd84ba4 authored 作者: --global's avatar --global

Make gpuarray.GpuFromHost.r_Op return a list

上级 804f9114
...@@ -294,7 +294,7 @@ class GpuFromHost(Op): ...@@ -294,7 +294,7 @@ class GpuFromHost(Op):
if isinstance(ev, GpuArrayType): if isinstance(ev, GpuArrayType):
return [host_from_gpu(ev)] return [host_from_gpu(ev)]
else: else:
return ev return [ev]
def infer_shape(self, node, xshp): def infer_shape(self, node, xshp):
return xshp return xshp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论