未验证 提交 ed7bc2bd 编写于 作者: G Ghost Screaming 提交者: GitHub

Fix bug of reduce sum kp operator compilation (#46204)

上级 7bce7949
...@@ -16,11 +16,14 @@ ...@@ -16,11 +16,14 @@
#include <limits> #include <limits>
#include "paddle/phi/core/enforce.h" #include "paddle/phi/core/enforce.h"
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/funcs/eigen/common.h"
#include "paddle/phi/kernels/gpu/reduce.h" #include "paddle/phi/kernels/gpu/reduce.h"
#ifndef PADDLE_WITH_XPU_KP
#include "paddle/phi/kernels/funcs/eigen/common.h"
#endif
namespace phi { namespace phi {
#ifndef PADDLE_WITH_XPU_KP
template <typename T, template <typename T,
int EigenDimSize = 5, int EigenDimSize = 5,
int ReducedDimSize = 1, int ReducedDimSize = 1,
...@@ -66,6 +69,7 @@ void ReduceSumEigen(const KPDevice& dev_ctx, ...@@ -66,6 +69,7 @@ void ReduceSumEigen(const KPDevice& dev_ctx,
eigen_x_tensor.sum(eigen_reduce_dim); eigen_x_tensor.sum(eigen_reduce_dim);
out->Resize(origin_out_dims); out->Resize(origin_out_dims);
} }
#endif
template <typename T, typename Context> template <typename T, typename Context>
void SumRawKernel(const Context& dev_ctx, void SumRawKernel(const Context& dev_ctx,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册