File tree 2 files changed +6
-0
lines changed
2 files changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -183,6 +183,7 @@ if(WITH_PYTHON)
183
183
endif ()
184
184
set (PYBIND_SRCS ${PYBIND_SRCS} distributed_py.cc)
185
185
endif ()
186
+ set (PYBIND_DEPS ${PYBIND_DEPS} processgroup_comm_utils)
186
187
187
188
if (WITH_ASCEND)
188
189
set (PYBIND_DEPS ${PYBIND_DEPS} ascend_wrapper)
Original file line number Diff line number Diff line change @@ -59,6 +59,8 @@ limitations under the License. */
59
59
#include " paddle/fluid/distributed/store/tcp_store.h"
60
60
#endif
61
61
62
+ #include " paddle/phi/kernels/sync_batch_norm_kernel.h"
63
+
62
64
namespace py = pybind11;
63
65
64
66
namespace paddle {
@@ -90,6 +92,9 @@ using GlooOptions = paddle::distributed::ProcessGroupGloo::GlooOptions;
90
92
91
93
static std::string GLOO_SOCKET_IFNAME_ENV = " GLOO_SOCKET_IFNAME" ; // NOLINT
92
94
95
+ static UNUSED void *use_ccl_comm_func =
96
+ phi::detail::GetCCLComm (phi::CPUPlace());
97
+
93
98
void BindDistributed (py::module *m) {
94
99
py::enum_<distributed::ReduceOp>(*m, " ReduceOp" )
95
100
.value (" SUM" , distributed::ReduceOp::SUM)
You can’t perform that action at this time.
0 commit comments