未验证 提交 97004f67 编写于 作者: R ronnywang 提交者: GitHub

[CustomDevice] fix undefined symbol GetCCLComm in the cpu version (#47717)

上级 14c95700
......@@ -183,6 +183,7 @@ if(WITH_PYTHON)
endif()
set(PYBIND_SRCS ${PYBIND_SRCS} distributed_py.cc)
endif()
set(PYBIND_DEPS ${PYBIND_DEPS} processgroup_comm_utils)
if(WITH_ASCEND)
set(PYBIND_DEPS ${PYBIND_DEPS} ascend_wrapper)
......
......@@ -59,6 +59,8 @@ limitations under the License. */
#include "paddle/fluid/distributed/store/tcp_store.h"
#endif
#include "paddle/phi/kernels/sync_batch_norm_kernel.h"
namespace py = pybind11;
namespace paddle {
......@@ -90,6 +92,9 @@ using GlooOptions = paddle::distributed::ProcessGroupGloo::GlooOptions;
static std::string GLOO_SOCKET_IFNAME_ENV = "GLOO_SOCKET_IFNAME"; // NOLINT
static UNUSED void *use_ccl_comm_func =
phi::detail::GetCCLComm(phi::CPUPlace());
void BindDistributed(py::module *m) {
py::enum_<distributed::ReduceOp>(*m, "ReduceOp")
.value("SUM", distributed::ReduceOp::SUM)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册