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
@@ -29,8 +29,9 @@
 namespace paddle {
 namespace imperative {

-#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \
-    defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO)
+#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) ||     \
+    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
@@ -44,8 +44,9 @@ class VariableWrapper;
 namespace paddle {
 namespace imperative {

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

 template <typename T>
 struct DivNRanksFunctor {