Skip to content

Commit 0aff621

Browse files
support metax and iluvatar uniform complex64 (#2138)
1 parent 9cb1400 commit 0aff621

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/uniform_kernel_register.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,5 @@ PD_CUSTOM_KERNEL_REGISTER(uniform,
2121
phi::UniformKernel,
2222
float,
2323
phi::dtype::float16,
24-
phi::dtype::bfloat16) {}
24+
phi::dtype::bfloat16,
25+
phi::dtype::complex<float>) {}

0 commit comments

Comments
 (0)