Skip to content

Commit 1dc0e4a

Browse files
support metax and iluvatar uniform complex64
1 parent ab114ce commit 1dc0e4a

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
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>) {}

backends/metax_gpu/kernels/cuda_kernels/uniform_kernel_register.cu

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

0 commit comments

Comments
 (0)