Fix leftover get_best_config()

ROCm · Nov 7, 2023 · 148633a · 148633a
1 parent c3c2c29
commit 148633a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/python/tutorials/03-matrix-multiplication.py b/python/tutorials/03-matrix-multiplication.py
@@ -392,7 +392,7 @@ def benchmark(M, N, K, provider):
         ms, min_ms, max_ms = triton.testing.do_bench(lambda: matmul(a, b), quantiles=quantiles)
         global verbose
         if verbose:
-            print(f'SIZE: {M},{N},{K}   Best tuning config: ({matmul_kernel.get_best_config(M, N, K)})')
+            print(f'SIZE: {M},{N},{K}   Best tuning config: ({matmul_kernel.get_best_config()})')
     perf = lambda ms: 2 * M * N * K * 1e-12 / (ms * 1e-3)
     return perf(ms), perf(max_ms), perf(min_ms)
 

diff --git a/scripts/amd/gemm/matmul.py b/scripts/amd/gemm/matmul.py
@@ -236,10 +236,6 @@ def matmul(a, b, activation=""):
 
     return c
 
-def get_best_config(M, N, K):
-    best_config = matmul_kernel_splitK.get_best_config(M = M, N = N, K = K)
-    return best_config
-
 
 def test_correctness(M, N, K, datatype = torch.float16):
     torch.manual_seed(0)
@@ -341,7 +337,7 @@ def main():
 
         if args.compare:
             test_correctness(m, n, k, dtype)
-        best_config = get_best_config(m, n, k)
+        best_config = matmul_kernel_splitK.get_best_config()
 
         if use_rocprof:
             dtype_str = 'fp16' if (not args.specify_type) else args.dtype