Fix punet commands

nod-ai · Aug 22, 2024 · 5c44ea1 · 5c44ea1
1 parent 01f36ee
commit 5c44ea1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tuning/punet_autotune.py b/tuning/punet_autotune.py
@@ -72,7 +72,7 @@ def get_model_compile_command(
         command = [
             "timeout",
             "300s",
-            (script_dir / "../int8-model/...").as_posix(),
+            (script_dir / "../int8-model/compile-punet-base.sh").as_posix(),
             "./tools/iree-compile",
             "gfx942",
             f"{mlir_spec_path.resolve()}",