diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
index 0cd36e6448..5c4f880512 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
@@ -13,7 +13,6 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/weighted_sample_neighbors_kernel.cu"  //NOLINT
 #include "paddle/phi/kernels/weighted_sample_neighbors_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(weighted_sample_neighbors,
diff --git a/backends/metax_gpu/CMakeLists.txt b/backends/metax_gpu/CMakeLists.txt
index ddadacdaf7..a00dc4514b 100755
--- a/backends/metax_gpu/CMakeLists.txt
+++ b/backends/metax_gpu/CMakeLists.txt
@@ -541,6 +541,7 @@ file(
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/meshgrid_grad_kernel.cu.cc
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pad3d_grad_kernel.cu
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/pad3d_kernel.cu
+  ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/weighted_sample_neighbors_kernel.cu
   # ############################################################################
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/array_grad_kernel.cc
   ${PADDLE_SOURCE_DIR}/paddle/phi/kernels/set_kernel.cc
diff --git a/backends/metax_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
index c96fdc8bb6..2840f69261 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/weighted_sample_neighbors_kernel_register.cu
@@ -13,7 +13,6 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/weighted_sample_neighbors_kernel.cu"  //NOLINT
 #include "paddle/phi/kernels/weighted_sample_neighbors_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(weighted_sample_neighbors,