提交 ee245777 authored 作者: Frederic's avatar Frederic

crash fix

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