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) )