diff --git a/backends/metax_gpu/kernels/cuda_kernels/argsort_grad_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/argsort_grad_kernel_register.cu
index b91c02a310..aaba44dfdc 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/argsort_grad_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/argsort_grad_kernel_register.cu
@@ -13,8 +13,8 @@
 // limitations under the License.
 
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/argsort_grad_kernel.h"
 #include "paddle/phi/kernels/argsort_kernel.h"
-#include "paddle/phi/kernels/gpu/argsort_grad_kernel.cu"  //NOLINT
 
 PD_CUSTOM_KERNEL_REGISTER(argsort_grad,
                           metax_gpu,