未验证 提交 c558dc30 编写于 作者: J jameszhang 提交者: GitHub

[kunlun] support allreduce sparse (#54172)

上级 060e4fab
......@@ -1097,6 +1097,15 @@ void EagerReducer::AllReduceSparse(EagerGroup *group,
PADDLE_THROW(platform::errors::PermissionDenied(
"Paddle can't concat grad tensors since it's not compiled with NCCL,"
"Please recompile or reinstall Paddle with NCCL support."));
#endif
} else if (platform::is_xpu_place(inner_place_)) {
#ifdef PADDLE_WITH_XPU_BKCL
dev_ctx = static_cast<platform::XPUDeviceContext *>(
platform::DeviceContextPool::Instance().Get(inner_place_));
#else
PADDLE_THROW(platform::errors::PermissionDenied(
"Paddle can't concat grad tensors since it's not compiled with XCCL,"
"Please recompile or reinstall Paddle with XCCL support."));
#endif
} else if (platform::is_custom_place(inner_place_)) {
#ifdef PADDLE_WITH_CUSTOM_DEVICE
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册