From 6f51343370cde77db5ae416b592e0739849bf131 Mon Sep 17 00:00:00 2001 From: Lixun Zhang Date: Tue, 12 Nov 2024 10:18:38 -0600 Subject: [PATCH] fix min --> max --- python/perf-kernels/tools/tune_gemm/tune_gemm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/perf-kernels/tools/tune_gemm/tune_gemm.py b/python/perf-kernels/tools/tune_gemm/tune_gemm.py index f0eb5a55599d..a525a0424c00 100755 --- a/python/perf-kernels/tools/tune_gemm/tune_gemm.py +++ b/python/perf-kernels/tools/tune_gemm/tune_gemm.py @@ -147,7 +147,7 @@ def prune_configs(M, N, K, configs, elemBytes_a, elemBytes_b): LDSB = BLOCK_SIZE_K * BLOCK_SIZE_N * elemBytes_b if num_stages <= 1: # No pipeline, buffer A and buffer B can re-use each other - LDS = min(LDSA, LDSB) + LDS = max(LDSA, LDSB) else: # Pipeline, we need (num_stages - 1) buffers for both A and B at the same time LDS = (LDSA + LDSB) * (num_stages - 1)