diff options
Diffstat (limited to 'verif/generator/tosa_verif_build_tests.py')
-rw-r--r-- | verif/generator/tosa_verif_build_tests.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/verif/generator/tosa_verif_build_tests.py b/verif/generator/tosa_verif_build_tests.py index 0d532c0..954c6e9 100644 --- a/verif/generator/tosa_verif_build_tests.py +++ b/verif/generator/tosa_verif_build_tests.py @@ -3,7 +3,9 @@ import argparse import re import sys +from pathlib import Path +import conformance.model_files as cmf from generator.tosa_test_gen import TosaTestGen from serializer.tosa_serializer import dtype_str_to_val from serializer.tosa_serializer import DTypeNames @@ -86,6 +88,13 @@ def parseArgs(argv): help="Tensor data generation is delayed til test running", ) + parser.add_argument( + "--generate-lib-path", + dest="generate_lib_path", + type=Path, + help="Path to TOSA generate library.", + ) + # Constraints on tests tens_group.add_argument( "--tensor-dim-range", @@ -268,6 +277,17 @@ def main(argv=None): args = parseArgs(argv) + if not args.lazy_data_gen: + if args.generate_lib_path is None: + args.generate_lib_path = cmf.find_tosa_file( + cmf.TosaFileType.GENERATE_LIBRARY, Path("reference_model"), False + ) + if not args.generate_lib_path.is_file(): + print( + f"Argument error: Generate library (--generate-lib-path) not found - {str(args.generate_lib_path)}" + ) + exit(2) + ttg = TosaTestGen(args) if args.test_type == "both": |