未验证 提交 e703a2ed 编写于 作者: D duanboqiang 提交者: GitHub

fix opt-offload save bug (#36433)

上级 12882b2f
...@@ -906,7 +906,7 @@ def save_persistables(exe, dirname, main_program, filename=None): ...@@ -906,7 +906,7 @@ def save_persistables(exe, dirname, main_program, filename=None):
"_velocity_0" "_velocity_0"
] ]
for check in checks: for check in checks:
if var.name.endswith(check): if var.name.endswith(check) and var.persistable:
return True return True
return False return False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册