Update new modes parser arguments

nod-ai · Jun 21, 2024 · fcceac8 · fcceac8
1 parent ec97d91
commit fcceac8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tuning/autotune.py b/tuning/autotune.py
@@ -66,7 +66,7 @@ def parse_arguments() -> argparse.Namespace:
 
     # Required arguments
     parser.add_argument(
-        "mode", choices=["default", "winograd"], help="Compilation mode"
+        "mode", choices=["default", "winograd", "misa", "hybrid"], help="Compilation mode"
     )
     parser.add_argument(
         "input_file", type=Path, help="Path to the input benchmark file (.mlir)"