address review comment

Signed-off-by: Whitney Tsang <[email protected]>

Author: whitneywhtsang

benchmarks/triton_kernels_benchmark/gemm_benchmark.py

@@ -329,7 +329,7 @@ def benchmark(B, M, N, K, provider):
             'gemm_shape_1_512_32768_8192': 'Test_1x512x32768x8192_row_row',
             'gemm_shape_1_1024_16384_8192': 'Test_1x1024x16384x8192_row_row',
             'gemm_shape_1_1024_28672_8192': 'Test_1x1024x28672x8192_row_row',
-            'gemm_shape_1_3072_4096_3072': 'Test_1x3072x4096x3072_row_row',
+            'gemm_streamk_shape_3072_4096_3072': 'Test_1x3072x4096x3072_row_row',
             'gemm_shape_1_4096_16384_8192': 'Test_1x4096x16384x8192_row_row',
             'gemm_shape_1_8192_16384_1024': 'Test_1x8192x16384x1024_row_row',
             'gemm_shape_1_8192_16384_4096': 'Test_1x8192x16384x4096_row_row',