support compelx64 dtype for remainder op (#2300)

HydrogenSulfate · web-flow · commit a36d51e59dc3 · 2025-12-23T17:08:06.000+08:00
diff --git 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<float>) {}
 PD_CUSTOM_KERNEL_REGISTER(floor_divide,
                           metax_gpu,
                           ALL_LAYOUT,
