diff --git a/tuning/libtuner.py b/tuning/libtuner.py index 7fb86e3..ee41779 100755 --- a/tuning/libtuner.py +++ b/tuning/libtuner.py @@ -311,9 +311,6 @@ def parse_arguments() -> argparse.Namespace: parser = argparse.ArgumentParser(description="Autotune script") # Required arguments - parser.add_argument( - "mode", choices=["default", "winograd"], help="Compilation mode" - ) parser.add_argument( "input_file", type=Path, help="Path to the input benchmark file (.mlir)" ) @@ -379,7 +376,7 @@ def parse_arguments() -> argparse.Namespace: def setup_logging(args: argparse.Namespace, path_config: PathConfig): - log_file_name = f"autotune_{args.mode}_{args.input_file.stem}.log" + log_file_name = f"autotune_{args.input_file.stem}.log" run_log_path = path_config.base_dir / log_file_name path_config._set_run_log(run_log_path) diff --git a/tuning/punet_autotune.py b/tuning/punet_autotune.py index c281bbf..0c94139 100644 --- a/tuning/punet_autotune.py +++ b/tuning/punet_autotune.py @@ -11,17 +11,17 @@ """ Sample Usage: -python punet_autotune.py winograd 1286.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64 +python punet_autotune.py 1286.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64 Recommended Trial Run: -python punet_autotune.py winograd 1286.mlir --num-candidates=1 +python punet_autotune.py 1286.mlir --num-candidates=1 Dry Run Test (no gpu requried): -python punet_autotune.py winograd 1286.mlir --num-candidates=64 --num-model-candidates=10 --dry-run +python punet_autotune.py 1286.mlir --num-candidates=64 --num-model-candidates=10 --dry-run """