diff options
author | evacha01 <evan.chandler@arm.com> | 2024-03-06 17:33:44 +0000 |
---|---|---|
committer | evacha01 <evan.chandler@arm.com> | 2024-03-07 12:52:13 +0000 |
commit | 1adff8301ac7e4933382549ad22b3b47cdb01eda (patch) | |
tree | 1cbe95bb38a9f877e5bbc89294064239e3e4b8b4 /verif/generator/tosa_test_gen.py | |
parent | 9c96eefbaca6c85be79529bce7ff04fd7dfe3a0d (diff) | |
download | reference_model-1adff8301ac7e4933382549ad22b3b47cdb01eda.tar.gz |
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
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 4 |
1 files changed, 3 insertions, 1 deletions
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 @@ class TosaTestGen: 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 @@ class TosaTestGen: TosaTensorValuesGen.tvgLazyGenDefault, TosaArgGen.agNone, ), - "types": TYPE_FIB, + "types": TYPE_FIB + [DType.INT4, DType.INT48], "data_gen": { "fp": (gtu.DataGenType.PSEUDO_RANDOM,), }, |