diff --git a/tests/kernel/wave/wave_e2e_test.py b/tests/kernel/wave/wave_e2e_test.py index 19bfca28..64714e43 100644 --- a/tests/kernel/wave/wave_e2e_test.py +++ b/tests/kernel/wave/wave_e2e_test.py @@ -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,