address review comment

whitneywhtsang · whitneywhtsang · commit b2adfdd06022 · 2024-10-17T15:31:38.000Z
Signed-off-by: Whitney Tsang &lt;whitney.tsang@intel.com&gt;
diff --git a/benchmarks/triton_kernels_benchmark/gemm_benchmark.py b/benchmarks/triton_kernels_benchmark/gemm_benchmark.py
@@ -342,6 +342,7 @@ def benchmark(B, M, N, K, provider):
             'gemm_shape_32_4096_4096_128': 'Test_32x4096x4096x128_row_row',
             'gemm_shape_4096_8_128_16384': 'Test_4096x8x128x16384_row_row',
             'gemm_shape_4096_8_16384_128': 'Test_4096x8x16384x128_row_row',
+            'gemm_streamk_shape_3072_4096_3072': 'stream_k_gemm_run',
         }
 
         # benchmark_suit.assert_close(xetla_fn(), torch_fn(), atol=1e-4, rtol=1.0, err_msg='xetla to torch')

Original file line number	Diff line number	Diff line change
`@@ -342,6 +342,7 @@ def benchmark(B, M, N, K, provider):`
`342`	`342`	`'gemm_shape_32_4096_4096_128': 'Test_32x4096x4096x128_row_row',`
`343`	`343`	`'gemm_shape_4096_8_128_16384': 'Test_4096x8x128x16384_row_row',`
`344`	`344`	`'gemm_shape_4096_8_16384_128': 'Test_4096x8x16384x128_row_row',`
	`345`	`+ 'gemm_streamk_shape_3072_4096_3072': 'stream_k_gemm_run',`
`345`	`346`	`}`
`346`	`347`
`347`	`348`	`# benchmark_suit.assert_close(xetla_fn(), torch_fn(), atol=1e-4, rtol=1.0, err_msg='xetla to torch')`