Skip to content

Commit 27572e6

Browse files
Fix merge
1 parent 44e4b63 commit 27572e6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

benchmarks/triton_kernels_benchmark/gemm_streamk_benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def benchmark(M, N, K, provider):
294294

295295
# benchmark_suit.assert_close(xetla_fn(), torch_fn(), atol=1e-4, rtol=1.0, err_msg='xetla to torch')
296296
_, min_ms, max_ms, mean_ms, cv = benchmark_suit.do_bench(
297-
xetla_fn, warmup=10, rep=10, quantiles=quantiles,
297+
xetla_fn, n_warmup=10, n_repeat=10, quantiles=quantiles,
298298
kernel_name='gpu::xetla::kernel::gemm_universal_t<dispatch_stream_k')
299299
else:
300300
raise NotImplementedError(f'Unsupported provider {provider}')

0 commit comments

Comments
 (0)