commit 32bf90181b8d4c5ba19416b9426f32ce2bdd9878
author Jeremy Johnson <jeremy.johnson@arm.com> Wed Mar 20 16:32:23 2024 +0000
committer Jeremy Johnson <jeremy.johnson@arm.com> Thu Mar 21 15:50:24 2024 +0000
tree bd66205172fdac05d350e23716a641888abc2864
parent ac8c0c83e52e0408191787e16b2b3bca1baf3ce7

Fix data range of int ADD/SUB ops

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Iaeed119ff81c8d1363dd26be7468bb53f9d70eae

verif/generator/tosa_arg_gen.py

diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 4878708..504bfa9 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -931,7 +931,9 @@
             ), "Op.ADD / Op.SUB must have 2 placeholders, 0 consts"
             tens_ser_list = []
             add = op["op"] in (Op.ADD, Op.ADD_SHAPE)
-            data_range = testGen.args.tensor_shape_range
+            data_range = None  # Use default
+            if op["op"] in (Op.ADD_SHAPE, Op.SUB_SHAPE):
+                data_range = testGen.args.tensor_shape_range
             a_arr = rng.randTensor(shapeList[0], dtypeList[0], data_range)
             b_arr = rng.randTensor(shapeList[1], dtypeList[1], data_range)
             if add: