diff --git a/paddle/phi/kernels/gpu/gather_kernel.cu b/paddle/phi/kernels/gpu/gather_kernel.cu index 7e0c6cc168564e94c5af2e26a8f9ba4acc0594ed..8ba9edb49fac523fb0359b9ccd7e87496d281095 100644 --- a/paddle/phi/kernels/gpu/gather_kernel.cu +++ b/paddle/phi/kernels/gpu/gather_kernel.cu @@ -66,5 +66,8 @@ PD_REGISTER_KERNEL(gather, int64_t, int, int16_t, + bool, + uint8_t, + int8_t, phi::dtype::float16, phi::dtype::bfloat16) {}