diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
index bbeaa10273..6ee6c05beb 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/global_gather_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/global_gather_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(global_gather,
                           iluvatar_gpu,
diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
index e2fe3370d2..3ffa0f4030 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/global_scatter_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/global_scatter_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(global_scatter,
                           iluvatar_gpu,
diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
index 201f4caee6..042e5d5b34 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/l1_norm_grad_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/gpu/l1_norm_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(
     l1_norm_grad, iluvatar_gpu, ALL_LAYOUT, phi::L1NormGradKernel, float) {}
diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
index 5535637b35..9ad91bf58a 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/lookup_table_grad_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/lookup_table_grad_kernel.h"
 PD_CUSTOM_KERNEL_REGISTER(lookup_table_grad,
                           iluvatar_gpu,
                           ALL_LAYOUT,
diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
index 30b58c16f6..b0a96e2bf0 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/lookup_table_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/lookup_table_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(lookup_table,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt
index 9fabd549a3..8831bdb9a0 100755
--- a/backends/metax_gpu/CMakeLists.txt
+++ b/backends/metax_gpu/CMakeLists.txt
@@ -124,6 +124,11 @@ file(
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/math/*.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/funcs/eigen/*.cu
   # kernels/gpu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/lookup_table_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/lookup_table_grad_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/l1_norm_grad_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/global_scatter_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/global_gather_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/yolo_box_post_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/spectral_norm_kernel.cu
diff --git a/backends/metax_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
index c24af944a1..1cc3e79e11 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/global_gather_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/global_gather_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/global_gather_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(global_gather,
                           metax_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
index 2494d27dd3..8a178fe1bf 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/global_scatter_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/global_scatter_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/global_scatter_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(global_scatter,
                           metax_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
index 1ce5a01485..8ec95a15d6 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/l1_norm_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/l1_norm_grad_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/gpu/l1_norm_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(
     l1_norm_grad, metax_gpu, ALL_LAYOUT, phi::L1NormGradKernel, float) {}
diff --git a/backends/metax_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
index 484d7cfe1e..6f9a9c0b51 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/lookup_table_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/lookup_table_grad_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/lookup_table_grad_kernel.h"
 PD_CUSTOM_KERNEL_REGISTER(lookup_table_grad,
                           metax_gpu,
                           ALL_LAYOUT,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
index 42d83c3adf..1caffe5644 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/lookup_table_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/lookup_table_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/lookup_table_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(lookup_table,
                           metax_gpu,