Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle
未验证 提交 b086404b 编写于 作者: R RedContritio 提交者： GitHub
浏览文件
操作

fix compile error on custom_devices (#53940)

上级 53d4e9e4
隐藏空白更改
内联 并排
Showing with 6 addition and 4 deletion
paddle/fluid/imperative/reducer.cc
浏览文件 @ b086404b
...@@ -29,8 +29,9 @@ ...@@ -29,8 +29,9 @@
namespace paddle { namespace paddle {
namespace imperative { namespace imperative {
#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \ #if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \
defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) || \
defined(PADDLE_WITH_CUSTOM_DEVICE)
// div the nranks // div the nranks
void Group::DivNRanks(const platform::DeviceContext &context, int64_t nranks) { void Group::DivNRanks(const platform::DeviceContext &context, int64_t nranks) {
phi::DenseTensor *tensor = phi::DenseTensor *tensor =
......
paddle/fluid/imperative/reducer.h
浏览文件 @ b086404b
...@@ -44,8 +44,9 @@ class VariableWrapper; ...@@ -44,8 +44,9 @@ class VariableWrapper;
namespace paddle { namespace paddle {
namespace imperative { namespace imperative {
#if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \ #if defined(PADDLE_WITH_NCCL) || defined(PADDLE_WITH_RCCL) || \
defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) defined(PADDLE_WITH_XPU_BKCL) || defined(PADDLE_WITH_GLOO) || \
defined(PADDLE_WITH_CUSTOM_DEVICE)
template <typename T> template <typename T>
struct DivNRanksFunctor { struct DivNRanksFunctor {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或