From 0f565abe95bfd24f1e74a2d8b7cf0a8cd3eccfde Mon Sep 17 00:00:00 2001 From: Adam Mainz Date: Wed, 20 Aug 2025 12:04:19 -0700 Subject: [PATCH] Fixing naming convention of gemms Summary: amd and nvidia have the same function name for the gemm kernels they have. Renaming to fix Reviewed By: nmacchioni, karthik-man Differential Revision: D80640938 --- tritonbench/operators/gemm/operator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tritonbench/operators/gemm/operator.py b/tritonbench/operators/gemm/operator.py index f006c7cba..bdbd1c7e8 100644 --- a/tritonbench/operators/gemm/operator.py +++ b/tritonbench/operators/gemm/operator.py @@ -322,7 +322,7 @@ def pt2_matmul_maxautotune(self, a, b, bias) -> Callable: return lambda: compiled(a, b) @register_benchmark(enabled=not is_cuda()) - def streamk_matmul(self, a, b, bias) -> Callable: + def streamk_matmul_amd(self, a, b, bias) -> Callable: return ( lambda: streamk_amd_matmul(a, b, bias) if bias else streamk_amd_matmul(a, b) )