提交 ac62b206 authored 作者: Ziye Fan's avatar Ziye Fan

bug fixed

上级 f176bf07
...@@ -3603,8 +3603,10 @@ def local_fill_sink(node): ...@@ -3603,8 +3603,10 @@ def local_fill_sink(node):
# node_sub_c = client.op([v for v in client.inputs if v is not c]) # node_sub_c = client.op([v for v in client.inputs if v is not c])
# import ipdb; ipdb.set_trace() # import ipdb; ipdb.set_trace()
r = local_fill_sink.transform(client) r = local_fill_sink.transform(client)
if r: if isinstance(r, list):
return {client: r} return {client.outputs[0]: r[0]}
elif isinstance(r, dict):
return r
return [c] return [c]
register_canonicalize(local_fill_sink) register_canonicalize(local_fill_sink)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论