Skip to content

Commit 8f5cbf4

Browse files
committed
fix
1 parent 0a308ce commit 8f5cbf4

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

paddle/phi/kernels/gpu/expand_kernel.cu

+2
Original file line numberDiff line numberDiff line change
@@ -99,5 +99,7 @@ PD_REGISTER_KERNEL(expand,
9999
int8_t,
100100
phi::dtype::float16,
101101
phi::dtype::bfloat16,
102+
phi::dtype::float8_e4m3fn,
103+
phi::dtype::float8_e5m2,
102104
phi::dtype::complex<float>,
103105
phi::dtype::complex<double>) {}

paddle/phi/kernels/gpu/tile_kernel.cu

+2
Original file line numberDiff line numberDiff line change
@@ -109,5 +109,7 @@ PD_REGISTER_KERNEL(tile,
109109
int64_t,
110110
phi::dtype::float16,
111111
phi::dtype::bfloat16,
112+
phi::dtype::float8_e4m3fn,
113+
phi::dtype::float8_e5m2,
112114
phi::dtype::complex<float>,
113115
phi::dtype::complex<double>) {}

0 commit comments

Comments
 (0)