[tuner] clean up the code. NFC. (#841)

- Fix mypy warning issues.
- remove unused parameter for benchmark function.
- remove unused import.

Signed-off-by: Bangtian Liu <liubangtian@gmail.com>

tuner/examples/simple/simple_tuner.py

@@ -126,7 +126,6 @@ def main():
         simple_tuner.benchmark_flags = ["--input=1", "--benchmark_repetitions=3"]
         top_candidates = libtuner.benchmark(
             args,
-            path_config,
             compiled_candidates,
             candidate_trackers,
             simple_tuner,
@@ -159,7 +158,6 @@ def main():
         simple_tuner.benchmark_timeout = 60
         top_model_candidates = libtuner.benchmark(
             args,
-            path_config,
             compiled_model_candidates,
             candidate_trackers,
             simple_tuner,

tuner/tuner/candidate_gen.py

@@ -291,7 +291,7 @@ def strip_compilation_info(input_path: Path) -> str:
     return result.process_res.stdout
 
 
-def main():
+def main() -> None:
     parser = argparse.ArgumentParser()
     parser.add_argument("input", help="Input mlir file", type=str)
     parser.add_argument(
@@ -369,4 +369,4 @@ def main():
 
 
 if __name__ == "__main__":
-    args = main()
+    main()

tuner/tuner/libtuner.py

@@ -892,7 +892,6 @@ def get_speedup(result: BenchmarkResult) -> float:
 
 def benchmark(
     args: argparse.Namespace,
-    path_config: PathConfig,
     compiled_candidates: list[int],
     candidate_trackers: list[CandidateTracker],
     tuning_client: TuningClient,

tuner/tuner/libtuner_test.py

@@ -6,7 +6,6 @@
 
 import argparse
 import math
-from subprocess import CompletedProcess
 from unittest.mock import call, patch, MagicMock
 from . import libtuner