提交 4b16b7fa authored 作者: Frederic's avatar Frederic

Fix optimization error

上级 0749674e
......@@ -637,6 +637,8 @@ def gpu_reconstruct_graph(inputs, outputs, tag=None):
@op_lifter([scan_op.Scan])
def local_scan_to_gpua(node):
info = copy.deepcopy(node.op.info)
if info.get('gpua', False):
return
info['gpua'] = True
nw_ins = [node.inputs[0]]
e = (1 +
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论