fix multinode test auc in xpu

zmxdream · Mar 12, 2024 · 2e9a451 · 2e9a451
1 parent 8ed91f0
commit 2e9a451
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/framework/fleet/box_wrapper.cc b/paddle/fluid/framework/fleet/box_wrapper.cc
@@ -1296,7 +1296,7 @@ void BoxWrapper::InitializeGPUAndLoadModel(
     }
 #if defined(PADDLE_WITH_XPU_KP)
     std::vector<void*> bkcl_ctx_vec(gpu_num_, nullptr);
-    if (gpu_num_ > 1) {
+    if (boxps::MPICluster::Ins().size() == 1 && gpu_num_ > 1) {
         for (int i = 0; i < gpu_num_; ++i) {
             auto place = platform::XPUPlace(i);
             auto comm = platform::BKCLCommContext::Instance().Get(0, place);