commit 1adff8301ac7e4933382549ad22b3b47cdb01eda
author evacha01 <evan.chandler@arm.com> Wed Mar 06 17:33:44 2024 +0000
committer evacha01 <evan.chandler@arm.com> Thu Mar 07 12:52:13 2024 +0000
tree 1cbe95bb38a9f877e5bbc89294064239e3e4b8b4
parent 9c96eefbaca6c85be79529bce7ff04fd7dfe3a0d

Add INT48 and INT4 datatype support to IDENTITY

Added support to ref model and testing

Signed-off-by: evacha01 <evan.chandler@arm.com>
Change-Id: Iece53d07c8986332fdd8f1ce5ed6265349df1b6a

verif/generator/tosa_test_gen.py

diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index a1f54c6..4309024 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -191,6 +191,8 @@
 
         if dtype == DType.BOOL:
             return np.bool_(self.rng.choice(a=[False, True], size=shape))
+        elif dtype == DType.INT4:
+            return np.int8(self.rng.integers(low=low, high=high, size=shape))
         elif dtype == DType.INT8:
             return np.int8(self.rng.integers(low=low, high=high, size=shape))
         elif dtype == DType.UINT8:
@@ -4652,7 +4654,7 @@
                 TosaTensorValuesGen.tvgLazyGenDefault,
                 TosaArgGen.agNone,
             ),
-            "types": TYPE_FIB,
+            "types": TYPE_FIB + [DType.INT4, DType.INT48],
             "data_gen": {
                 "fp": (gtu.DataGenType.PSEUDO_RANDOM,),
             },