diff --git a/paddle/fluid/distributed/collective/ProcessGroupHeter.cc b/paddle/fluid/distributed/collective/ProcessGroupHeter.cc index fdce542954aacc17a35d8355a3f4946f28795d09..2963d65f2611f68c6eb68735893a8256e0f1bc94 100644 --- a/paddle/fluid/distributed/collective/ProcessGroupHeter.cc +++ b/paddle/fluid/distributed/collective/ProcessGroupHeter.cc @@ -73,15 +73,15 @@ ProcessGroupHeter::ProcessGroupHeter(const std::shared_ptr& store, src_rank_(src_rank), dst_rank_(dst_rank) { return; -#if defined(PADDLE_WITH_NCCL) +#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) inner_pg_ = std::make_shared( store, local_rank, local_size, place_, IGNORE_ID); #elif defined(PADDLE_WITH_ASCEND_CL) inner_pg_ = std::make_shared( store, local_rank, local_size, place_, IGNORE_ID); #else - PADDLE_THROW(platform::errors::Fatal( - "ProcessGroupHeter only supports NCCL and HCCL now."); + PADDLE_THROW(platform::errors::Unavailable( + "ProcessGroupHeter only supports NCCL, RCCL and HCCL now.")); #endif if (local_rank_ == 0 && !with_switch_) { auto opts = ProcessGroupGloo::GlooOptions::create(); @@ -104,7 +104,7 @@ std::shared_ptr ProcessGroupHeter::AllReduce( std::vector& in_tensors, std::vector& out_tensors, const AllreduceOptions& opts) { -#if defined(PADDLE_WITH_NCCL) +#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) PADDLE_ENFORCE_EQ( CheckTensorsInCudaPlace(in_tensors), true, @@ -213,7 +213,7 @@ std::shared_ptr ProcessGroupHeter::Broadcast( std::vector& in_tensors, std::vector& out_tensors, const BroadcastOptions& opts) { -#if defined(PADDLE_WITH_NCCL) +#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) PADDLE_ENFORCE_EQ( CheckTensorsInCudaPlace(in_tensors), true, diff --git a/paddle/fluid/distributed/collective/ProcessGroupHeter.h b/paddle/fluid/distributed/collective/ProcessGroupHeter.h index 0385391d364632ad324dae0b87174292c0b8891a..2b0146e6ec016f0502789626fa9a8877b6724a19 100644 --- a/paddle/fluid/distributed/collective/ProcessGroupHeter.h +++ b/paddle/fluid/distributed/collective/ProcessGroupHeter.h @@ -35,11 +35,10 @@ #include "paddle/fluid/platform/place.h" #include "paddle/fluid/platform/stream/cuda_stream.h" -#if defined(PADDLE_WITH_NCCL) +#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) #include "paddle/fluid/distributed/collective/NCCLTools.h" #include "paddle/fluid/distributed/collective/ProcessGroupNCCL.h" #include "paddle/fluid/platform/cuda_device_guard.h" -#include "paddle/fluid/platform/dynload/nccl.h" #endif #if defined(PADDLE_WITH_ASCEND_CL) @@ -48,7 +47,8 @@ #endif #if defined(PADDLE_WITH_DISTRIBUTE) && defined(PADDLE_WITH_PSCORE) && \ - (defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_ASCEND_CL)) + (defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \ + defined(PADDLE_WITH_ASCEND_CL)) #include "paddle/fluid/distributed/ps/service/heter_client.h" #endif