diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-01-27 14:17:52 +0000 |
---|---|---|
committer | Luke Hutton <luke.hutton@arm.com> | 2023-01-27 14:22:15 +0000 |
commit | 5c844210d6d6018f89a41463d4976a000136aee4 (patch) | |
tree | 5580dcca7cf0cbab97d0c02d9e8f6d42656f45d9 | |
parent | d713a4d841ad09a9466e43aa1bf0be09fe54ea22 (diff) | |
download | reference_model-5c844210d6d6018f89a41463d4976a000136aee4.tar.gz |
Add RFFT2d to optimized only kernel list
This fixes framework test generation when the "optimized"
kernel mode option is not selected. As a result, the
optimized RFFT2d kernel is used when generating tests in
the reference kernel mode to prevent "op not found"
errors.
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: Ice18bfa08c46e237e39c99f849622bf99d9b849c
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 742f991..c93fc88 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1204,7 +1204,7 @@ def run_unit_test( if tflite_model_filename: tflite_result_npy_filename = "tflite_result.npy" - ops_with_optimized_only_kernel = ["elu", "ceil", "gather"] + ops_with_optimized_only_kernel = ["elu", "ceil", "gather", "rfft2d"] if args.tflite_kernel_mode == "optimized" or ( op_name in ops_with_optimized_only_kernel |