Skip to content

Commit a56b524

Browse files
authored
[microbenchmarks] Fixed gbps formula for gemm-streamk (#5106)
Fixed gbps formula for gemm-streamk benchmark
1 parent 43fbabb commit a56b524

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
@@ -295,7 +295,7 @@ def benchmark(M, N, K, provider):
295295
raise NotImplementedError(f'Unsupported provider {provider}')
296296

297297
tflops = lambda mean: 2 * M * N * K * (1e-12) / (mean * 1e-3)
298-
gbps = lambda mean: 2 * (M * K + K * N) + 4.0 * (M * N) * (1e-9) / (mean * 1e-3)
298+
gbps = lambda mean: (2 * (M * K + K * N) + 4.0 * M * N) * (1e-9) / (mean * 1e-3)
299299

300300
return (gbps(mean_ms), gbps(max_ms), gbps(min_ms)), (tflops(mean_ms), tflops(max_ms), tflops(min_ms)), cv
301301

0 commit comments

Comments
 (0)