From 31efec537b9231f0871192c23c735014de2d667e Mon Sep 17 00:00:00 2001 From: YuanRisheng Date: Fri, 31 Dec 2021 16:33:30 +0800 Subject: [PATCH] [Pten]Fix bugs of compilation when use pten::add/subtract (#38631) * change 'math' to 'math_kernel' * fix compile bugs * merge develop * fix compile bugs * fix compile bugs --- paddle/fluid/operators/lu_op.h | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/paddle/fluid/operators/lu_op.h b/paddle/fluid/operators/lu_op.h index 256219f2e3b..21839c263e4 100644 --- a/paddle/fluid/operators/lu_op.h +++ b/paddle/fluid/operators/lu_op.h @@ -20,7 +20,7 @@ limitations under the License. */ #include "paddle/fluid/operators/svd_helper.h" #include "paddle/fluid/operators/triangular_solve_op.h" #include "paddle/fluid/operators/tril_triu_op.h" -#include "paddle/pten/include/math.h" +#include "paddle/pten/kernels/math_kernel.h" namespace paddle { namespace operators { @@ -223,7 +223,8 @@ void Tensor_Add(const DeviceContext& dev_ctx, const framework::Tensor& src1, auto pt_x = paddle::experimental::MakePtenDenseTensor(src1); auto pt_y = paddle::experimental::MakePtenDenseTensor(src2); auto pt_z = paddle::experimental::MakePtenDenseTensor(*out); - pten::Add(dev_ctx, *pt_x.get(), *pt_y.get(), -1, pt_z.get()); + pten::AddKernel(dev_ctx, *pt_x.get(), *pt_y.get(), -1, + pt_z.get()); } template @@ -234,7 +235,8 @@ void Tensor_Sub(const DeviceContext& dev_ctx, const framework::Tensor& src1, auto pt_x = paddle::experimental::MakePtenDenseTensor(src1); auto pt_y = paddle::experimental::MakePtenDenseTensor(src2); auto pt_z = paddle::experimental::MakePtenDenseTensor(*out); - pten::Subtract(dev_ctx, *pt_x.get(), *pt_y.get(), -1, pt_z.get()); + pten::SubtractKernel(dev_ctx, *pt_x.get(), *pt_y.get(), -1, + pt_z.get()); } template -- GitLab