@@ -106,6 +106,10 @@ def run(args,mustPrint=False,dumpStdErr=True,timeout=20,printCmd=False):
106106parser .add_argument ('-dump' , action = 'store_true' , help = "Dump build output" )
107107parser .add_argument ('-s' , nargs = '?' ,type = int ,help = "Force subtest" )
108108
109+ parser .add_argument ('--dt' , nargs = '?' ,type = str , default = "ALL" ,help = "Dataype to run" )
110+ parser .add_argument ('--test' , nargs = '?' ,type = str , default = "ALL" ,help = "Test to run" )
111+
112+
109113args = parser .parse_args ()
110114
111115GHACTION = False
@@ -176,16 +180,23 @@ def cmd_args():
176180
177181AVHROOT = args .avh
178182
179-
180- # Fusion F64 not working. To be analyzed
181- TESTS = ["DOT_TEST" ,
183+ ALL_TESTS = ["DOT_TEST" ,
182184 "VECTOR_TEST" ,
183185 "ROW_TEST" ,
184186 "COL_TEST" ,
185187 "MATRIX_TEST" ,
186188# "FUSION_TEST"
187189 ]
188190
191+ # Fusion F64 not working. To be analyzed
192+ if args .test != "ALL" :
193+ if args .test not in ALL_TESTS :
194+ print (f"Test { args .test } not known" )
195+ exit (1 )
196+ TESTS = [args .test ]
197+ else :
198+ TESTS = ALL_TESTS
199+
189200# Some tests are too big (code size) and needs to be decomposed
190201# They contain SUBTEST1, SUBTEST2 ... #if in the code
191202# This script must know how many subtests are defined in each test
@@ -201,26 +212,8 @@ def is_only_test(n,i):
201212 return (i in ONLY_TESTS [n [0 ]])
202213 return False
203214
204- DATATYPES = ["COMPLEX_F64_DT" ,
205- "F64_DT" ,
206- "COMPLEX_F32_DT" ,
207- "F32_DT" ,
208- "COMPLEX_F16_DT" ,
209- "F16_DT" ,
210- "COMPLEX_Q31_DT" ,
211- "Q31_DT" ,
212- "COMPLEX_Q15_DT" ,
213- "Q15_DT" ,
214- "Q7_DT"
215- ]
216-
217- MODE = ["STATIC_TEST" ,
218- "DYNAMIC_TEST"
219- ]
220215
221- # Restricted tests for debugging
222- #TESTS=["DOT_TEST"]
223- DATATYPES = [ #"COMPLEX_F64_DT",
216+ ALL_DATATYPES = [ #"COMPLEX_F64_DT",
224217 "F64_DT" ,
225218 "COMPLEX_F32_DT" ,
226219 "F32_DT" ,
@@ -233,6 +226,22 @@ def is_only_test(n,i):
233226 #"COMPLEX_Q7_DT",
234227 "Q7_DT"
235228 ]
229+
230+ if args .dt != "ALL" :
231+ if args .dt not in ALL_DATATYPES :
232+ print (f"Datatype { args .dt } not known" )
233+ exit (1 )
234+ DATATYPES = [args .dt ]
235+ else :
236+ DATATYPES = ALL_DATATYPES
237+
238+ MODE = ["STATIC_TEST" ,
239+ "DYNAMIC_TEST"
240+ ]
241+
242+ # Restricted tests for debugging
243+ #TESTS=["DOT_TEST"]
244+
236245#DATATYPES=[
237246# "COMPLEX_F16_DT",
238247# "Q31_DT",
0 commit comments