未验证 提交 2642caaa 编写于 作者: W wangzhen38 提交者: GitHub

[gpubox fix] benchmark2 (#49649)

上级 1a0b3661
......@@ -232,9 +232,11 @@ class PSGPUWrapper {
if (s_instance_ == nullptr) {
return;
}
#if defined(PADDLE_WITH_GPU_GRAPH) && defined(PADDLE_WITH_HETERPS)
if (FLAGS_gpugraph_storage_mode == GpuGraphStorageMode::WHOLE_HBM) {
this->EndPass();
}
#endif
for (size_t i = 0; i < hbm_pools_.size(); i++) {
delete hbm_pools_[i];
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册