Skip to content

Commit

Permalink
add all option for running benchmarks
Browse files Browse the repository at this point in the history
  • Loading branch information
오원석 authored and 오원석 committed Nov 18, 2024
1 parent 1591252 commit c66d3dd
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 c66d3dd

Please sign in to comment.