提交 9e2eb687 authored 作者: sentient07's avatar sentient07

Added ._props_dict to GPU OP

上级 1e2f2860
......@@ -379,7 +379,7 @@ def local_gpu_split(node):
if (input.owner and isinstance(input.owner.op, HostFromGpu) or
any(c != 'output' and isinstance(c.op, GpuFromHost) for c, idx
in outs_clients)):
new_op = GpuSplit(node.op.len_splits)
new_op = GpuSplit(**node.op._props_dict())
split_res = new_op(as_cuda_ndarray_variable(input),
*node.inputs[1:], return_list=True)
return [host_from_gpu(o) for o in split_res]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论