diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
index 210a41dfa9..c472d59403 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/ap_facade_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/ap_facade_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(ap_facade,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt
index 475074ced8..e765065079 100755
--- a/backends/metax_gpu/CMakeLists.txt
+++ b/backends/metax_gpu/CMakeLists.txt
@@ -145,6 +145,7 @@ file(
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/angle_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/angle_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/adamax_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/ap_facade_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/bincount_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/c_embedding_grad_kernel.cu
diff --git a/backends/metax_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
index 7d89b1d90f..121df4f7e0 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/ap_facade_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/ap_facade_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/gpu/ap_facade_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(ap_facade,
                           metax_gpu,