diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
index c44a556b0dc..c4484e68cf6 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/affine_channel_grad_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/affine_channel_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(affine_channel_grad,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt
index bca1ce7aad4..5ab9fc412d3 100755
--- a/backends/metax_gpu/CMakeLists.txt
+++ b/backends/metax_gpu/CMakeLists.txt
@@ -559,6 +559,7 @@ file(
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/transpose_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/mask_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/sync_batch_norm_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/affine_channel_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/unary_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/sparse/gpu/sum_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/transfer_layout_kernel.cc
diff --git a/backends/metax_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
index 236ce34b9ec..1e6c747d523 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/affine_channel_grad_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/affine_channel_grad_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/affine_channel_grad_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(affine_channel_grad,
                           metax_gpu,