From 03bec734775dad8e45145b4d0dae4584199c1b84 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Thu, 7 Oct 2021 12:06:00 +0100 Subject: Fix rank and dtype filtering for ops like conv3d & fully_connected Change-Id: Ic2aebe40b5cce61d4576a64f4f48ff87b36475c2 Signed-off-by: Jeremy Johnson --- verif/tosa_test_gen.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) (limited to 'verif') diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index 2478331..8d69831 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -2694,21 +2694,19 @@ class TosaTestGen: if rank >= rmin and rank <= rmax: cleanRankFilter.append(rank) elif rankFilter is None and shapeFilter[0] is None: - cleanRankFilter = [] - # Ensure default behaviour is bounded by default range or by operator, whichever is smaller. - rankRange = range(rmin, rmax + 1) - for rank in rankRange: - if rank >= min(default_test_rank_range) and rank <= max(default_test_rank_range): - cleanRankFilter.append(rank) + # Ensure default behaviour is bounded by default range or by operator, + # whichever is the smaller range of ranks. + opRankRange = range(rmin, rmax + 1) + cleanRankFilter = opRankRange if len(opRankRange) <= len(default_test_rank_range) else default_test_rank_range else: cleanRankFilter = range(rmin, rmax + 1) dtypes = op["types"] if dtypeFilter is not None: cleanDtypeFilter = [] - # Ensure filtered dtypes are allowed by operator - for dtype in dtypeFilter: - if dtype in dtypes: + # Create list of operator dtypes filtered by requested dtypes + for dtype in dtypes: + if dtype in dtypeFilter or (isinstance(dtype, list) and dtype[0] in dtypeFilter): cleanDtypeFilter.append(dtype) else: cleanDtypeFilter = dtypes -- cgit v1.2.1