提交 7ac7ac40 authored 作者: Frederic's avatar Frederic

Fix crash in new gpu opt, the new_op return many outputs, so no need to add a list

上级 af29025c
......@@ -313,8 +313,8 @@ def local_gpu_split(node):
if host_input.owner and isinstance(host_input.owner.op, tensor.Split):
split_node = host_input.owner
new_op = GpuSplit(split_node.op.len_splits)
return [new_op(gpu_from_host(split_node.inputs[0]),
*split_node.inputs[1:])]
return new_op(gpu_from_host(split_node.inputs[0]),
*split_node.inputs[1:])
return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论