diff options
author | Eric Kunze <eric.kunze@arm.com> | 2023-07-20 10:52:56 -0700 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-07-26 17:23:54 +0000 |
commit | 97b0027ca018ad9b5f91f41b413e843afb15c6d7 (patch) | |
tree | 9e57fd24214e63444444a4dab6353bb7ccc04dab /verif/frameworks/tosa_verif_framework_generator.py | |
parent | 95961ccffc3e3cac1e6d85c28ca361bcb0bd65bc (diff) | |
download | reference_model-97b0027ca018ad9b5f91f41b413e843afb15c6d7.tar.gz |
Update framework tools to add name in json
Name of the test can now come from the json description
rather than inferred from the directory. Old method remains to
keep functionality with older tests
Change-Id: I79f872103ae6a101b5cc799af91c933839f28dfb
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 124bf6e..ec009c6 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1288,6 +1288,8 @@ def run_unit_test( # Assume single result tensor now tflite_result_name = output_details[0]["name"] + _, test_name = os.path.split(test_dir) + # Write out test descriptor write_test_json( filename=os.path.join(test_dir, "test.json"), @@ -1302,6 +1304,7 @@ def run_unit_test( ifm_shape=placeholder_shapes, framework_exclusions=excluded_framework_list, quantized=is_quantized, + test_name=test_name, ) except Exception as e: msg = "Error running task: {}".format(e) |