diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-20 16:32:23 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-21 15:50:24 +0000 |
commit | 32bf90181b8d4c5ba19416b9426f32ce2bdd9878 (patch) | |
tree | bd66205172fdac05d350e23716a641888abc2864 | |
parent | ac8c0c83e52e0408191787e16b2b3bca1baf3ce7 (diff) | |
download | reference_model-32bf90181b8d4c5ba19416b9426f32ce2bdd9878.tar.gz |
Fix data range of int ADD/SUB ops
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Iaeed119ff81c8d1363dd26be7468bb53f9d70eae
-rw-r--r-- | verif/generator/tosa_arg_gen.py | 4 |
1 files changed, 3 insertions, 1 deletions
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 @@ class TosaTensorValuesGen: ), "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: |