diff --git a/backends/metax_gpu/kernels/cuda_kernels/elementwise_kernel_register.cu b/backends/metax_gpu/kernels/cuda_kernels/elementwise_kernel_register.cu index 098f3ec2fc..6137cd0f12 100644 --- a/backends/metax_gpu/kernels/cuda_kernels/elementwise_kernel_register.cu +++ b/backends/metax_gpu/kernels/cuda_kernels/elementwise_kernel_register.cu @@ -45,7 +45,8 @@ PD_CUSTOM_KERNEL_REGISTER(remainder, int, int64_t, phi::dtype::float16, - phi::dtype::bfloat16) {} + phi::dtype::bfloat16, + phi::dtype::complex) {} PD_CUSTOM_KERNEL_REGISTER(floor_divide, metax_gpu, ALL_LAYOUT,