From 82ad4d628cdb2e20975d8b7b43b41bbf381b015f Mon Sep 17 00:00:00 2001 From: Matthew Haddon Date: Fri, 20 Aug 2021 15:02:39 +0100 Subject: Set TILE multiple to one if input dimension is large * Sets multiple of input dimension to one if the dimension of the input tensor is large to reduce ouput tensor size. Change-Id: I2ebcecf438282de032a33e0cf4b3847cbf94440d --- verif/tosa_test_gen.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'verif') 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 -- cgit v1.2.1