From ab121e47556bd02c4057acc0b258aedb52f6005c Mon Sep 17 00:00:00 2001 From: HydrogenSulfate <490868991@qq.com> Date: Tue, 23 Dec 2025 15:00:17 +0800 Subject: [PATCH] support compelx64 dtype for remainder op --- .../kernels/cuda_kernels/elementwise_kernel_register.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 098f3ec2fcc..6137cd0f122 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,