diff --git a/backends/iluvatar_gpu/kernels/cuda_kernels/uniform_kernel_register.cc b/backends/iluvatar_gpu/kernels/cuda_kernels/uniform_kernel_register.cc
index b30fbe1659a..760f11e21e2 100644
--- a/backends/iluvatar_gpu/kernels/cuda_kernels/uniform_kernel_register.cc
+++ b/backends/iluvatar_gpu/kernels/cuda_kernels/uniform_kernel_register.cc
@@ -21,4 +21,5 @@ PD_CUSTOM_KERNEL_REGISTER(uniform,
                           phi::UniformKernel,
                           float,
                           phi::dtype::float16,
-                          phi::dtype::bfloat16) {}
+                          phi::dtype::bfloat16,
+                          phi::dtype::complex<float>) {}