diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
index 0f119143eb7..309f8e0ce54 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
@@ -14,7 +14,7 @@
 
 #include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/core/utils/data_type.h"
-#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(fusion_group,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
index 39ca317b271..09fcfb3999b 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/fusion_group_kernel_register.cu
@@ -14,7 +14,7 @@
 
 #include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/core/utils/data_type.h"
-#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/fusion/gpu/fusion_group_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(fusion_group,
                           metax_gpu,