diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/complex_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/complex_kernel_register.cu
index b9dafdf489..0424539bb7 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/complex_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/complex_kernel_register.cu
@@ -12,8 +12,9 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#include "paddle/phi/common/data_type.h"
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/complex_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/complex_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(conj,
                           iluvatar_gpu,
@@ -34,7 +35,7 @@ PD_CUSTOM_KERNEL_REGISTER(real,
                           phi::RealKernel,
                           phi::dtype::complex<float>,
                           phi::dtype::complex<double>) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToReal(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToRealType(kernel_key.dtype()));
 }
 
 PD_CUSTOM_KERNEL_REGISTER(imag,
@@ -43,10 +44,10 @@ PD_CUSTOM_KERNEL_REGISTER(imag,
                           phi::ImagKernel,
                           phi::dtype::complex<float>,
                           phi::dtype::complex<double>) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToReal(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToRealType(kernel_key.dtype()));
 }
 
 PD_CUSTOM_KERNEL_REGISTER(
     complex, iluvatar_gpu, ALL_LAYOUT, phi::ComplexKernel, float, double) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToComplex(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToComplexType(kernel_key.dtype()));
 }
diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/fused_softmax_mask_upper_triangle_kernel_register.cu b/backends/iluvatar_gpu/kernels/cuda_kernels/fused_softmax_mask_upper_triangle_kernel_register.cu
index 1562bb77f9..7cb81e92f8 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/fused_softmax_mask_upper_triangle_kernel_register.cu
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/fused_softmax_mask_upper_triangle_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 #include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/kernels/fused_softmax_mask_upper_triangle_kernel.h"
-#include "paddle/phi/kernels/fusion/gpu/fused_softmax_mask_upper_triangle_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/fusion/fused_softmax_mask_upper_triangle_kernel.h"
 #include "paddle/phi/kernels/fusion/gpu/fused_softmax_mask_upper_triangle_utils.h"
 
 PD_CUSTOM_KERNEL_REGISTER(fused_softmax_mask_upper_triangle,
diff --git a/backends/metax_gpu/kernels/cuda_kernels/complex_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/complex_kernel_register.cu
index 5598aab7b8..199239d973 100644
--- a/backends/metax_gpu/kernels/cuda_kernels/complex_kernel_register.cu
+++ b/backends/metax_gpu/kernels/cuda_kernels/complex_kernel_register.cu
@@ -12,8 +12,9 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#include "paddle/phi/common/data_type.h"
 #include "paddle/phi/core/kernel_registry.h"
-#include "paddle/phi/kernels/gpu/complex_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/complex_kernel.h"
 
 PD_CUSTOM_KERNEL_REGISTER(conj,
                           metax_gpu,
@@ -34,7 +35,7 @@ PD_CUSTOM_KERNEL_REGISTER(real,
                           phi::RealKernel,
                           phi::dtype::complex<float>,
                           phi::dtype::complex<double>) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToReal(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToRealType(kernel_key.dtype()));
 }
 
 PD_CUSTOM_KERNEL_REGISTER(imag,
@@ -43,10 +44,10 @@ PD_CUSTOM_KERNEL_REGISTER(imag,
                           phi::ImagKernel,
                           phi::dtype::complex<float>,
                           phi::dtype::complex<double>) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToReal(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToRealType(kernel_key.dtype()));
 }
 
 PD_CUSTOM_KERNEL_REGISTER(
     complex, metax_gpu, ALL_LAYOUT, phi::ComplexKernel, float, double) {
-  kernel->OutputAt(0).SetDataType(phi::dtype::ToComplex(kernel_key.dtype()));
+  kernel->OutputAt(0).SetDataType(phi::ToComplexType(kernel_key.dtype()));
 }
diff --git a/backends/metax_gpu/kernels/fusion/fused_softmax_mask_upper_triangle_kernel_register.cu b/backends/metax_gpu/kernels/fusion/fused_softmax_mask_upper_triangle_kernel_register.cu
index 72176aaaf9..1cd4363818 100644
--- a/backends/metax_gpu/kernels/fusion/fused_softmax_mask_upper_triangle_kernel_register.cu
+++ b/backends/metax_gpu/kernels/fusion/fused_softmax_mask_upper_triangle_kernel_register.cu
@@ -13,7 +13,7 @@
 // limitations under the License.
 #include "paddle/phi/core/kernel_registry.h"
 #include "paddle/phi/kernels/fused_softmax_mask_upper_triangle_kernel.h"
-#include "paddle/phi/kernels/fusion/gpu/fused_softmax_mask_upper_triangle_kernel.cu"  // NOLINT
+#include "paddle/phi/kernels/fusion/fused_softmax_mask_upper_triangle_kernel.h"
 #include "paddle/phi/kernels/fusion/gpu/fused_softmax_mask_upper_triangle_utils.h"
 
 PD_CUSTOM_KERNEL_REGISTER(fused_softmax_mask_upper_triangle,