1 files changed, 10 insertions, 0 deletions
diff --git a/verif/frameworks/tosa_verif_framework_compiler_runner.py b/verif/frameworks/tosa_verif_framework_compiler_runner.py
index 71723ae..0d98c17 100755
--- a/verif/frameworks/tosa_verif_framework_compiler_runner.py
+++ b/verif/frameworks/tosa_verif_framework_compiler_runner.py
@@ -74,6 +74,14 @@ def parse_args():
         help="Comparison tolerance b value",
     )
     parser.add_argument(
+        "--tosa_level",
+        dest="tosa_level",
+        default="EIGHTK",
+        type=str,
+        help="A TOSA level defines operator parameter ranges that an implementation shall support."
+        "Config tosa_level for running the reference model only. Default is EIGHTK",
+    )
+    parser.add_argument(
         "--no-compiler",
         dest="no_compiler",
         action="store_true",
@@ -552,6 +560,8 @@ def run_test(args, test, framework):
             "-q",
         ] + ref_model_cmd
 
+    ref_model_cmd = ref_model_cmd + ["--tosa_level={}".format(args.tosa_level)]
+
     # Clean out any ref_model result first
     try:
         os.remove(os.path.join(test, flatbuffer_dir, "ref_model_*.npy"))