未验证 提交 17a99760 编写于 作者: R ronnywang 提交者: GitHub

fix npu_finalize (#34857)

上级 507ea06f
......@@ -2221,6 +2221,7 @@ All parameter, weight, gradient are variables in Paddle.
auto &pool = platform::DeviceContextPool::Instance();
auto devices = platform::GetSelectedNPUDevices();
for (size_t i = 0; i < devices.size(); ++i) {
platform::NPUDeviceGuard guard(devices[i]);
pool.Get(platform::NPUPlace(devices[i]))->Wait();
}
platform::AclInstance::Instance().Finalize();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册