From 2642caaa94f837472c3b4cf34a831bcbb09ecf0b Mon Sep 17 00:00:00 2001
From: wangzhen38 <41941775+wangzhen38@users.noreply.github.com>
Date: Mon, 9 Jan 2023 19:37:48 +0800
Subject: [PATCH] [gpubox fix] benchmark2 (#49649)

---
 paddle/fluid/framework/fleet/ps_gpu_wrapper.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/framework/fleet/ps_gpu_wrapper.h b/paddle/fluid/framework/fleet/ps_gpu_wrapper.h
index 4a452f4d36..119aebf4b2 100644
--- a/paddle/fluid/framework/fleet/ps_gpu_wrapper.h
+++ b/paddle/fluid/framework/fleet/ps_gpu_wrapper.h
@@ -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];
     }
-- 
GitLab