Fix compile_unet commands

nod-ai · Aug 21, 2024 · 145f768 · 145f768
1 parent 58a745b
commit 145f768
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/tuning/libtuner.py b/tuning/libtuner.py
@@ -1011,6 +1011,7 @@ def compile_models(
             args,
             candidate_id=i,
             command=tuning_client.get_model_compile_command(candidate_trackers[i]),
+            check=False
         )
         for i in candidates
         if i != 0

diff --git a/tuning/punet_autotune.py b/tuning/punet_autotune.py
@@ -66,12 +66,8 @@ def get_model_compile_command(
     ) -> list[str]:
         mlir_spec_path = candidate_tracker.spec_path
         assert mlir_spec_path is not None
-        script_dir = Path(__file__).resolve().parent
-        command = [
-            "./compile_unet_candidate.sh",
-            "winograd",
-            mlir_spec_path.as_posix(),
-        ]
+        target_dir = mlir_spec_path.resolve().parent.parent.parent
+        output_name = f"unet_candidate_{candidate_tracker.candidate_id}.vmfb"
         command = [
             "timeout",
             "300s",
@@ -81,9 +77,7 @@ def get_model_compile_command(
             f"{mlir_spec_path.resolve()}",
             "./punet.mlir",
             "-o",
-            (
-                script_dir / "unet_candidate_{candidate_tracker.candidate_id}.vmfb"
-            ).as_posix(),
+            (target_dir / output_name).as_posix()
         ]
         return command