未验证 提交 65adfecf 编写于 作者: W wanghuancoder 提交者: GitHub

[Eager] [Bug Fix] fix eager trace op bug (#40402)

* fix some slice bug, test=develop

* refine, test=develop
上级 c6ec8b9f
......@@ -140,7 +140,12 @@ class Tracer(core.Tracer):
outputs[retname][j].reconstruct_from_(returns[i][j],
False)
else:
outputs[retname][0].reconstruct_from_(returns[i], False)
if isinstance(outputs[retname], list):
outputs[retname][0].reconstruct_from_(returns[i],
False)
else:
outputs[retname].reconstruct_from_(returns[i],
False)
elif isinstance(returns, list):
assert len(outputs.keys()) == 1
key = list(outputs.keys())[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册