Apply suggestions from code review

anmyachev · whitneywhtsang · web-flow · commit c21c92affed3 · 2024-12-18T18:36:11.000+01:00
Co-authored-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
@@ -117,7 +117,7 @@ def matmul_kernel_with_block_pointers(
             num_stages=s, num_warps=4) for s in [2]
     ],
     key=['M', 'N', 'K'],
-    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),
+    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),
 )
 @triton.jit
 def matmul_kernel_with_block_pointers_batched(
diff --git a/benchmarks/triton_kernels_benchmark/gemm_postop_addmatrix_benchmark.py b/benchmarks/triton_kernels_benchmark/gemm_postop_addmatrix_benchmark.py
@@ -110,7 +110,7 @@ def matmul_kernel_with_block_pointers(
             num_stages=2, num_warps=4),
     ],
     key=['M', 'N', 'K'],
-    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),
+    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),
 )
 @triton.jit
 def matmul_kernel_with_block_pointers_batched(
diff --git a/benchmarks/triton_kernels_benchmark/gemm_postop_gelu_benchmark.py b/benchmarks/triton_kernels_benchmark/gemm_postop_gelu_benchmark.py
@@ -123,7 +123,7 @@ def matmul_kernel_with_block_pointers(
             num_stages=2, num_warps=4),
     ],
     key=['M', 'N', 'K'],
-    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),
+    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),
 )
 @triton.jit
 def matmul_kernel_with_block_pointers_batched(
diff --git a/benchmarks/triton_kernels_benchmark/gemm_preop_exp_benchmark.py b/benchmarks/triton_kernels_benchmark/gemm_preop_exp_benchmark.py
@@ -108,7 +108,7 @@ def matmul_kernel_with_block_pointers(
             num_stages=2, num_warps=4),
     ],
     key=['M', 'N', 'K'],
-    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),
+    do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),
 )
 @triton.jit
 def matmul_kernel_with_block_pointers_batched(

Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ def matmul_kernel_with_block_pointers(`
`117`	`117`	`num_stages=s, num_warps=4) for s in [2]`
`118`	`118`	`],`
`119`	`119`	`key=['M', 'N', 'K'],`
`120`		`- do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),`
	`120`	`+ do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),`
`121`	`121`	`)`
`122`	`122`	`@triton.jit`
`123`	`123`	`def matmul_kernel_with_block_pointers_batched(`
Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def matmul_kernel_with_block_pointers(`
`110`	`110`	`num_stages=2, num_warps=4),`
`111`	`111`	`],`
`112`	`112`	`key=['M', 'N', 'K'],`
`113`		`- do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),`
	`113`	`+ do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),`
`114`	`114`	`)`
`115`	`115`	`@triton.jit`
`116`	`116`	`def matmul_kernel_with_block_pointers_batched(`
Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def matmul_kernel_with_block_pointers(`
`123`	`123`	`num_stages=2, num_warps=4),`
`124`	`124`	`],`
`125`	`125`	`key=['M', 'N', 'K'],`
`126`		`- do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),`
	`126`	`+ do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),`
`127`	`127`	`)`
`128`	`128`	`@triton.jit`
`129`	`129`	`def matmul_kernel_with_block_pointers_batched(`
Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ def matmul_kernel_with_block_pointers(`
`108`	`108`	`num_stages=2, num_warps=4),`
`109`	`109`	`],`
`110`	`110`	`key=['M', 'N', 'K'],`
`111`		`- do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers'),`
	`111`	`+ do_bench=benchmark_suit.make_do_bench_for_autotune(kernel_name='matmul_kernel_with_block_pointers_batched'),`
`112`	`112`	`)`
`113`	`113`	`@triton.jit`
`114`	`114`	`def matmul_kernel_with_block_pointers_batched(`