diff options
Diffstat (limited to 'verif')
-rw-r--r-- | verif/tosa_test_gen.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index efc819c..d41d6e2 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -784,8 +784,13 @@ class TosaArgGen: # enormous tensors multiples = [] for i in range(rank): - multiples.append(testGen.randInt(1, 4)) - + if ifm_shape[i] > 1000: + # Multiple of 1 if ifm_shape dimension is large to reduce tensor size + multiples.append(1) + elif max(ifm_shape) > 1000: + multiples.append(2) + else: + multiples.append(testGen.randInt(1, 4)) arg_list.append(("perm{}".format(p), [multiples])) return arg_list |