Skip to content

Commit 200c3ab

Browse files
authored
[tuner] Remove args.mode (#100)
Remove unused `args.mode`
1 parent ff47eab commit 200c3ab

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

tuning/libtuner.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -311,9 +311,6 @@ def parse_arguments() -> argparse.Namespace:
311311
parser = argparse.ArgumentParser(description="Autotune script")
312312

313313
# Required arguments
314-
parser.add_argument(
315-
"mode", choices=["default", "winograd"], help="Compilation mode"
316-
)
317314
parser.add_argument(
318315
"input_file", type=Path, help="Path to the input benchmark file (.mlir)"
319316
)
@@ -379,7 +376,7 @@ def parse_arguments() -> argparse.Namespace:
379376

380377

381378
def setup_logging(args: argparse.Namespace, path_config: PathConfig):
382-
log_file_name = f"autotune_{args.mode}_{args.input_file.stem}.log"
379+
log_file_name = f"autotune_{args.input_file.stem}.log"
383380
run_log_path = path_config.base_dir / log_file_name
384381
path_config._set_run_log(run_log_path)
385382

tuning/punet_autotune.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -11,17 +11,17 @@
1111
"""
1212
Sample Usage:
1313
14-
python punet_autotune.py winograd 1286.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64
14+
python punet_autotune.py 1286.mlir --lhs-dims=bmk --rhs-dims=bkn --tile-dims=*mnk --devices=hip://0,hip://1 --num-candidates=64
1515
1616
1717
Recommended Trial Run:
1818
19-
python punet_autotune.py winograd 1286.mlir --num-candidates=1
19+
python punet_autotune.py 1286.mlir --num-candidates=1
2020
2121
2222
Dry Run Test (no gpu requried):
2323
24-
python punet_autotune.py winograd 1286.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
24+
python punet_autotune.py 1286.mlir --num-candidates=64 --num-model-candidates=10 --dry-run
2525
2626
"""
2727

0 commit comments

Comments
 (0)