aboutsummaryrefslogtreecommitdiff
path: root/verif/frameworks/tosa_verif_framework_generator.py
diff options
context:
space:
mode:
authorEric Kunze <eric.kunze@arm.com>2023-07-20 10:52:56 -0700
committerEric Kunze <eric.kunze@arm.com>2023-07-26 17:23:54 +0000
commit97b0027ca018ad9b5f91f41b413e843afb15c6d7 (patch)
tree9e57fd24214e63444444a4dab6353bb7ccc04dab /verif/frameworks/tosa_verif_framework_generator.py
parent95961ccffc3e3cac1e6d85c28ca361bcb0bd65bc (diff)
downloadreference_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-xverif/frameworks/tosa_verif_framework_generator.py3
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)