未验证 提交 2b479e17 编写于 作者: L Leo Chen 提交者: GitHub

fix import error of GlooParallelContext (#37892)

上级 c7cb7eec
...@@ -92,8 +92,10 @@ HeterParallelContext::HeterParallelContext(const ParallelStrategy &strategy, ...@@ -92,8 +92,10 @@ HeterParallelContext::HeterParallelContext(const ParallelStrategy &strategy,
inter_strategy_.local_rank_ = inter_rank; inter_strategy_.local_rank_ = inter_rank;
inter_strategy_.current_endpoint_ = strategy_.current_endpoint_; inter_strategy_.current_endpoint_ = strategy_.current_endpoint_;
inter_strategy_.trainer_endpoints_ = inter_endpoints; inter_strategy_.trainer_endpoints_ = inter_endpoints;
#ifdef PADDLE_WITH_GLOO
inter_parallel_ctx_ = std::make_shared<GLOOParallelContext>( inter_parallel_ctx_ = std::make_shared<GLOOParallelContext>(
inter_strategy_, platform::CPUPlace()); inter_strategy_, platform::CPUPlace());
#endif
} }
VLOG(0) << "init inter size " << inter_endpoints.size() << " rank " VLOG(0) << "init inter size " << inter_endpoints.size() << " rank "
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册