提交 745f50f7 authored 作者: Frederic's avatar Frederic

remove optimization crash. This case didn't make sense

上级 fc104d7c
...@@ -300,7 +300,7 @@ def local_gpu_elemwise_1(node): ...@@ -300,7 +300,7 @@ def local_gpu_elemwise_1(node):
@register_opt() @register_opt()
@local_optimizer([tensor.Split, gpu_from_host]) @local_optimizer([tensor.Split])
def local_gpu_split(node): def local_gpu_split(node):
if isinstance(node.op, tensor.Split): if isinstance(node.op, tensor.Split):
input = node.inputs[0] input = node.inputs[0]
...@@ -308,13 +308,6 @@ def local_gpu_split(node): ...@@ -308,13 +308,6 @@ def local_gpu_split(node):
new_op = GpuSplit(node.op.len_splits) new_op = GpuSplit(node.op.len_splits)
split_res = new_op(gpu_from_host(input), *node.inputs[1:]) split_res = new_op(gpu_from_host(input), *node.inputs[1:])
return [host_from_gpu(o) for o in split_res] return [host_from_gpu(o) for o in split_res]
if isinstance(node.op, GpuFromHost):
host_input = node.inputs[0]
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 False return False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论