未验证 提交 93fce181 编写于 作者: X xiayanming 提交者: GitHub

fix rccl bug (#32808)

上级 298f210d
......@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册