File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
benchmarks/triton_kernels_benchmark Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff 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 ,
Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments