【Hackathon 5th No.13】【关联 PR】Added int16 support for less_than

PaddlePaddle · Oct 26, 2023 · e9c8823 · e9c8823
1 parent 6f7cd08
commit e9c8823
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/python/paddle/tensor/logic.py b/python/paddle/tensor/logic.py
@@ -907,6 +907,7 @@ def less_than(x, y, name=None):
                 "float16",
                 "float32",
                 "float64",
+                "int16",
                 "int32",
                 "int64",
                 "uint16",
@@ -921,6 +922,7 @@ def less_than(x, y, name=None):
                 "float16",
                 "float32",
                 "float64",
+                "int16",
                 "int32",
                 "int64",
                 "uint16",

diff --git a/test/legacy_test/test_compare_op.py b/test/legacy_test/test_compare_op.py
@@ -615,6 +615,21 @@ def test_place_2(self):
         self.assertEqual((result.numpy() == np.array([False])).all(), True)
 
 
+class TestLessThanInt16(unittest.TestCase):
+    def test_less_than_int16(self):
+        # Create a tensor of type int8
+        x = paddle.to_tensor([1, 2, 3], dtype='int16')
+        y = paddle.to_tensor([1, 3, 2], dtype='int16')
+
+        result = paddle.less_than(x, y)
+
+        # desired output
+        expected = np.array([False, True, False])
+
+        # Verify output
+        self.assertTrue((result.numpy() == expected).all())
+
+
 if __name__ == '__main__':
     paddle.enable_static()
     unittest.main()
diff --git a/third_party/mkldnn b/third_party/mkldnn
+3 −9		src/cpu/matmul/gemm_based_common.hpp
+2 −14		src/cpu/x64/cpu_isa_traits.cpp
+2 −3		src/cpu/x64/jit_brgemm_conv_bwd_utils.cpp
+3 −4		src/cpu/x64/jit_brgemm_inner_product.cpp