From 32bf90181b8d4c5ba19416b9426f32ce2bdd9878 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Wed, 20 Mar 2024 16:32:23 +0000 Subject: Fix data range of int ADD/SUB ops Signed-off-by: Jeremy Johnson Change-Id: Iaeed119ff81c8d1363dd26be7468bb53f9d70eae --- verif/generator/tosa_arg_gen.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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: -- cgit v1.2.1