Skip to content

Commit 9128c80

Browse files
committed
fix
1 parent 70607ec commit 9128c80

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

paddle/phi/kernels/cpu/index_select_grad_kernel.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,4 +63,5 @@ PD_REGISTER_KERNEL(index_select_grad,
6363
phi::dtype::complex<float>,
6464
phi::dtype::complex<double>,
6565
int,
66-
int64_t) {}
66+
int64_t,
67+
bool) {}

paddle/phi/kernels/gpu/index_select_grad_kernel.cu

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,9 +136,11 @@ PD_REGISTER_KERNEL(index_select_grad,
136136
phi::IndexSelectGradKernel,
137137
float,
138138
double,
139+
phi::dtype::float8_e4m3fn,
139140
phi::dtype::float16,
140141
phi::dtype::bfloat16,
141142
phi::dtype::complex<float>,
142143
phi::dtype::complex<double>,
143144
int,
144-
int64_t) {}
145+
int64_t,
146+
bool) {}

0 commit comments

Comments
 (0)