[Ops] Support more dtype for gather kernel (#45142)

0b4268a6 · Siming Dai · GitHub · 5bcabf78 · 0b4268a6
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/phi/kernels/gpu/gather_kernel.cu paddle/phi/kernels/gpu/gather_kernel.cu +3 -0

未找到文件。
--- 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) {}