Skip to content

Commit

Permalink
Add all option for running benchmarks (#58)
Browse files Browse the repository at this point in the history
add all option for running benchmarks

Co-authored-by: 오원석 <[email protected]>
  • Loading branch information
marinelay and 오원석 authored Nov 18, 2024
1 parent 1591252 commit 94d007d
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion kaprese/bin/kaprese_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from rich.table import Table
from rich.text import Text

from kaprese.core.benchmark import Benchmark
from kaprese.core.benchmark import Benchmark, all_benchmarks
from kaprese.core.engine import Engine
from kaprese.core.runner import Runner
from kaprese.utils.console import PanelConsole, console
Expand Down Expand Up @@ -321,6 +321,12 @@ def main(
continue
engines.append(engine)
benchmarks: list[Benchmark] = []

# Load benchmarks
# If no benchmark is specified, run all benchmarks
if len(args.benchmark) == 0:
benchmarks = all_benchmarks()

for bench_name in args.benchmark:
bench = Benchmark.load(bench_name)
if bench is None:
Expand Down

0 comments on commit 94d007d

Please sign in to comment.