diff options
Diffstat (limited to 'verif/tosa_verif_build_tests.py')
-rwxr-xr-x | verif/tosa_verif_build_tests.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/verif/tosa_verif_build_tests.py b/verif/tosa_verif_build_tests.py index 343d8d4..02d1934 100755 --- a/verif/tosa_verif_build_tests.py +++ b/verif/tosa_verif_build_tests.py @@ -201,6 +201,14 @@ def parseArgs(): help="Allow constant input tensors for concat operator", ) + parser.add_argument( + "--test-type", + dest="test_type", + choices=['positive', 'negative', 'both'], + default="positive", + type=str, + help="type of tests produced, postive, negative, or both", + ) args = parser.parse_args() return args @@ -221,15 +229,19 @@ def main(): shapeFilter=args.target_shapes, rankFilter=args.target_ranks, dtypeFilter=args.target_dtypes, + testType=args.test_type ) ) print("{} matching tests".format(len(testList))) + results = [] for opName, testStr, dtype, shapeList, testArgs in testList: if args.verbose: print(testStr) - ttg.serializeTest(opName, testStr, dtype, shapeList, testArgs) - print("Done creating {} tests".format(len(testList))) + results.append(ttg.serializeTest(opName, testStr, dtype, shapeList, testArgs)) + + print(f"Done creating {len(results)} tests") + if __name__ == "__main__": |