Use the same heuristics for mfma type as PR#352 (#366)

ROCm · Oct 19, 2023 · f963c04 · f963c04
1 parent 20f316b
commit f963c04
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/scripts/amd/gemm/tune_gemm.py b/scripts/amd/gemm/tune_gemm.py
@@ -45,10 +45,7 @@ def get_full_tuning_space():
 def prune_configs(M, N, K, configs):
     pruned_configs = []
 
-    ## TODO: improve how we deal with mfma16 vs mfma32
-    ## after it becomes a tuning parameter
-    mfma_type = os.getenv('MFMA_TYPE')
-    if mfma_type == '16':
+    if M < 32 or N < 32:
         mfma = 16
     else:
         mfma = 32