✨ Refactor: added int8 support for less_than gpu kernel

PaddlePaddle · Oct 10, 2023 · 0ebade0 · 0ebade0
1 parent ad5b059
commit 0ebade0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/phi/kernels/legacy/kps/compare_kernel.cu b/paddle/phi/kernels/legacy/kps/compare_kernel.cu
@@ -161,6 +161,7 @@ PD_REGISTER_KERNEL(less_than_raw,
                    ALL_LAYOUT,
                    phi::LessThanRawKernel,
                    bool,
+                   int8_t,
                    int16_t,
                    int,
                    int64_t,
@@ -177,6 +178,7 @@ PD_REGISTER_KERNEL(less_than_raw,
                      ALL_LAYOUT,                          \
                      phi::func##RawKernel,                \
                      bool,                                \
+                     int8_t,                              \
                      int16_t,                             \
                      int,                                 \
                      int64_t,                             \