fix compile error on custom_devices (#53940)

b086404b · RedContritio · GitHub · 53d4e9e4 · b086404b · b086404b
显示空白变更内容
内联并排

Showing with 6 addition and 4 deletion

paddle/fluid/imperative/reducer.cc paddle/fluid/imperative/reducer.cc +3 -2

paddle/fluid/imperative/reducer.h paddle/fluid/imperative/reducer.h +3 -2

未找到文件。
--- a/paddle/fluid/imperative/reducer.cc
+++ b/paddle/fluid/imperative/reducer.cc
@@ -30,7 +30,8 @@ namespace paddle {
 namespace imperative {

 #if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) ||     \
-    defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO)
+    defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) || \
+    defined(PADDLE_WITH_CUSTOM_DEVICE)
 // div the nranks
 void Group::DivNRanks(const platform::DeviceContext &context, int64_t nranks) {
  phi::DenseTensor *tensor =

--- a/paddle/fluid/imperative/reducer.h
+++ b/paddle/fluid/imperative/reducer.h
@@ -45,7 +45,8 @@ namespace paddle {
 namespace imperative {

 #if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) ||     \
-    defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO)
+    defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) || \
+    defined(PADDLE_WITH_CUSTOM_DEVICE)

 template <typename T>
 struct DivNRanksFunctor {