.

PaddlePaddle · Apr 12, 2022 · 409cb71 · 409cb71 · paddle-bot-old · Apr 12, 2022
1 parent 385ff3b
commit 409cb71
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/paddle/fluid/distributed/ps/service/heter_client.h b/paddle/fluid/distributed/ps/service/heter_client.h
@@ -177,7 +177,7 @@ class HeterClient {
     VLOG(4) << "peer role is: " << peer_role
             << ", addr is: " << peer_endpoints[0];
     switch_s_instance_.SetPeerSwitchList(peer_endpoints);
-    switch_s_instance_.InitClientChannels(true, peer_endpoints, peer_role);
+    switch_s_instance_.InitClientChannels(false, peer_endpoints, peer_role);
     return switch_s_instance_;
   }
 

diff --git a/paddle/fluid/operators/pscore/heter_cloud_comm_cpu_test.cc b/paddle/fluid/operators/pscore/heter_cloud_comm_cpu_test.cc
@@ -116,7 +116,7 @@ void StartSwitchInterServer(
     std::vector<std::string> peer_endpoints) {
   switch_server_ptr->SetPeerEndPoints(peer_endpoints);
   switch_server_ptr->SetInterEndpoint(endpoints[1]);
-  switch_server_ptr->StartHeterInterService(true);
+  switch_server_ptr->StartHeterInterService(false);
 }
 
 TEST(HETERSENDANDRECV, CPU) {

diff --git a/paddle/fluid/operators/pscore/switch_server_test.cc b/paddle/fluid/operators/pscore/switch_server_test.cc
@@ -50,7 +50,7 @@ void StartSwitchInterServer(
   LOG(INFO) << "switch heter service started";
   switch_server_ptr->SetPeerEndPoints(peer_endpoints);
   switch_server_ptr->SetInterEndpoint(endpoints[0]);
-  switch_server_ptr->StartHeterInterService(true);
+  switch_server_ptr->StartHeterInterService(false);
 }
 
 int main(int argc, char* argv[]) {