diff --git a/tuning/libtuner.py b/tuning/libtuner.py index f1c208e..1d85ff3 100755 --- 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 index d79a10f..15b29d4 100644 --- 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