diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
index 0502c3ce169..9309bb74f50 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/class_center_sample_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/class_center_sample_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(class_center_sample,
                           iluvatar_gpu,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
index 6f3b9ef1352..bf43587b6a4 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/class_center_sample_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/class_center_sample_kernel.cu"  //NOLINT
+#include "paddle/phi/kernels/class_center_sample_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(class_center_sample,
                           metax_gpu,