Skip to content

Commit a0988d2

Browse files
committed
fix kernel name for xetla streamk
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 59de19a commit a0988d2

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

benchmarks/triton_kernels_benchmark/gemm_streamk_benchmark.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -293,9 +293,8 @@ def benchmark(M, N, K, provider):
293293
torch_fn = lambda: torch.matmul(a, b).to(torch.float32)
294294

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

0 commit comments

Comments
 (0)