From 0ebade0f0bbb88e089cc332d565143a3bcd1a9bb Mon Sep 17 00:00:00 2001
From: PommesPeter <434596665@qq.com>
Date: Tue, 10 Oct 2023 14:44:25 +0800
Subject: [PATCH] :sparkles: Refactor: added int8 support for less_than gpu
 kernel

---
 paddle/phi/kernels/legacy/kps/compare_kernel.cu | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/phi/kernels/legacy/kps/compare_kernel.cu b/paddle/phi/kernels/legacy/kps/compare_kernel.cu
index 5ab9dd8fea2d33..8c89daa8538b81 100644
--- 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,                             \