fix rccl bug (#32808)

93fce181 · xiayanming · GitHub · 298f210d · 93fce181
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/pipeline_trainer.cc paddle/fluid/framework/pipeline_trainer.cc +1 -1

未找到文件。
--- a/paddle/fluid/framework/pipeline_trainer.cc
+++ b/paddle/fluid/framework/pipeline_trainer.cc
@@ -35,7 +35,7 @@ void PipelineTrainer::Initialize(const TrainerDesc& trainer_desc,
  ParseDumpConfig(trainer_desc);
  const auto& section_config = section_params.section_config();
  int place_id = section_config.place_id();
-#if (defined PADDLE_WITH_NCCL)
+#if (defined PADDLE_WITH_NCCL) || (defined PADDLE_WITH_RCCL)
  place_ = platform::CUDAPlace(place_id);
 #elif (defined PADDLE_WITH_ASCEND_CL)  // NOLINT
  place_ = platform::NPUPlace(place_id);