diff --git a/paddle/fluid/operators/lu_op.h b/paddle/fluid/operators/lu_op.h
index 256219f2e3befe1e345ecb6a364941f2a36962e0..21839c263e4a84349959670a6cd674f0f39846d2 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<T>(dev_ctx, *pt_x.get(), *pt_y.get(), -1, pt_z.get());
+  pten::AddKernel<T, DeviceContext>(dev_ctx, *pt_x.get(), *pt_y.get(), -1,
+                                    pt_z.get());
 }
 
 template <typename DeviceContext, typename T>
@@ -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<T>(dev_ctx, *pt_x.get(), *pt_y.get(), -1, pt_z.get());
+  pten::SubtractKernel<T, DeviceContext>(dev_ctx, *pt_x.get(), *pt_y.get(), -1,
+                                         pt_z.get());
 }
 
 template <typename DeviceContext, typename T, size_t D>