refactor(distributed/communication/stream): rename utils

PaddlePaddle · Sep 27, 2022 · f136361 · f136361
1 parent 18a4b3a
commit f136361
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...le/fluid/distributed/collective/SplitOp.h → paddle/fluid/distributed/collective/Utils.h b/...le/fluid/distributed/collective/SplitOp.h → paddle/fluid/distributed/collective/Utils.h
@@ -108,7 +108,7 @@ void SplitTensor(const phi::DeviceContext *dev_ctx,
 
   const auto &place = dev_ctx->GetPlace();
   if (platform::is_gpu_place(place)) {
-#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL)
+#ifdef PADDLE_WITH_GPU
     SplitDenseTensorWithType(static_cast<const phi::GPUContext *>(dev_ctx),
                              tensor,
                              &dense_list,

diff --git a/paddle/fluid/pybind/distributed_py.cc b/paddle/fluid/pybind/distributed_py.cc
@@ -23,8 +23,8 @@ limitations under the License. */
 
 #include "paddle/fluid/distributed/collective/ProcessGroup.h"
 #include "paddle/fluid/distributed/collective/ProcessGroupStream.h"
-#include "paddle/fluid/distributed/collective/SplitOp.h"
 #include "paddle/fluid/distributed/collective/Types.h"
+#include "paddle/fluid/distributed/collective/Utils.h"
 #include "paddle/fluid/distributed/collective/reducer.h"
 #include "paddle/fluid/framework/lod_tensor.h"
 #include "paddle/fluid/framework/tensor.h"