Skip to content

Commit ef174e3

Browse files
committed
Change get_benchmark params name usage
1 parent d4494b6 commit ef174e3

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

benchmarks/triton_kernels_benchmark/gemm_tensor_desc_benchmark.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,8 @@ def get_benchmark(
120120
):
121121
return gemm_benchmark.get_benchmark(
122122
providers_filter=providers_filter,
123-
matmul_kernel=matmul_kernel_with_tensor_descriptors,
124-
matmul_kernel_batched=matmul_kernel_with_tensor_descriptors_batched,
123+
triton_matmul_kernel=matmul_kernel_with_tensor_descriptors,
124+
triton_matmul_kernel_batched=matmul_kernel_with_tensor_descriptors_batched,
125125
plot_name='matmul-tensor-desc-performance',
126126
transpose_a=transpose_a,
127127
transpose_b=transpose_b,

benchmarks/triton_kernels_benchmark/gemm_tensor_of_ptr_benchmark.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,8 @@ def get_benchmark(
127127
):
128128
return gemm_benchmark.get_benchmark(
129129
providers_filter=providers_filter,
130-
matmul_kernel=matmul_kernel,
131-
matmul_kernel_batched=matmul_kernel_batched,
130+
triton_matmul_kernel=matmul_kernel,
131+
triton_matmul_kernel_batched=matmul_kernel_batched,
132132
plot_name='matmul-tensor-of-ptr-performance',
133133
transpose_a=transpose_a,
134134
transpose_b=transpose_b,

0 commit comments

Comments
 (0)