Skip to content

Commit

Permalink
bench batch size
Browse files Browse the repository at this point in the history
Signed-off-by: Ivan Butygin <ivan.butygin@gmail.com>
  • Loading branch information
Hardcode84 committed Oct 3, 2024
1 parent 4399d66 commit 70eefe9
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/kernel/wave/wave_e2e_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -942,6 +942,10 @@ def repeat(acc: tkl.Register[M, NF, tkl.f32]) -> tkl.Register[M, NF, tkl.f32]:
config = {"backend": "rocm", "device": "hip", "target": "gfx942"}

run_bench = request.config.getoption("--runperf")
if run_bench:
config["benchmark_batch_size"] = 1000
config["benchmark_repetitions"] = 3

with tk.gen.TestLaunchContext(
{
N: n,
Expand Down

0 comments on commit 70eefe9

Please sign in to comment.