From 0b4268a627c6762dace33204c1bf061f317455f7 Mon Sep 17 00:00:00 2001
From: Siming Dai <908660116@qq.com>
Date: Tue, 16 Aug 2022 11:46:36 +0800
Subject: [PATCH] [Ops] Support more dtype for gather kernel (#45142)

---
 paddle/phi/kernels/gpu/gather_kernel.cu | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/phi/kernels/gpu/gather_kernel.cu b/paddle/phi/kernels/gpu/gather_kernel.cu
index 7e0c6cc168..8ba9edb49f 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) {}
-- 
GitLab