fix windows compile error

PaddlePaddle · Aug 7, 2023 · 675080a · 675080a
1 parent 2085a82
commit 675080a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/paddle/fluid/pybind/communication.cc b/paddle/fluid/pybind/communication.cc
@@ -111,7 +111,7 @@ void BindTCPStore(py::module *m) {
            py::arg("timeout") = 900,
            py::call_guard<py::gil_scoped_release>());
 
-  m->def("create_or_get_tcp_store",
+  m->def("create_or_get_global_tcp_store",
          &phi::distributed::CreateOrGetGlobalTCPStore);
 }
 

diff --git a/paddle/phi/core/distributed/auto_parallel/reshard_utils.cc b/paddle/phi/core/distributed/auto_parallel/reshard_utils.cc
@@ -17,7 +17,6 @@
 #include <cstdlib>
 #include "glog/logging.h"
 #include "paddle/phi/core/distributed/auto_parallel/process_mesh.h"
-#include "paddle/phi/core/distributed/store/tcp_store.h"
 
 namespace phi {
 namespace distributed {

diff --git a/paddle/phi/core/distributed/auto_parallel/reshard_utils.h b/paddle/phi/core/distributed/auto_parallel/reshard_utils.h
@@ -20,10 +20,10 @@
 #include <string>
 #include <vector>
 
+#include "paddle/phi/core/distributed/store/tcp_store.h"
+
 namespace phi {
 namespace distributed {
-class TCPStore;
-
 namespace auto_parallel {
 
 class ProcessMesh;

diff --git a/python/paddle/distributed/collective.py b/python/paddle/distributed/collective.py
@@ -319,7 +319,7 @@ def is_available():
 
 
 def _init_parallel_env(backend):
-    store = core.get_global_tcp_store()
+    store = core.create_or_get_global_tcp_store()
     global_env = _get_global_env()
     rank = global_env.rank
     world_size = global_env.world_size