commit 5c844210d6d6018f89a41463d4976a000136aee4
author Luke Hutton <luke.hutton@arm.com> Fri Jan 27 14:17:52 2023 +0000
committer Luke Hutton <luke.hutton@arm.com> Fri Jan 27 14:22:15 2023 +0000
tree 5580dcca7cf0cbab97d0c02d9e8f6d42656f45d9
parent d713a4d841ad09a9466e43aa1bf0be09fe54ea22

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

verif/frameworks/tosa_verif_framework_generator.py

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 @@
         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