From 40e53c828f647ae2bafe177930a694e7277b36aa Mon Sep 17 00:00:00 2001
From: Illia Silin <98187287+illsilin@users.noreply.github.com>
Date: Mon, 11 Nov 2024 09:25:08 -0800
Subject: [PATCH] restore collecting performance of mixed prec gemms (#1648)

[ROCm/composable_kernel commit: 5fb150dbe700eba180feb5b27973a8ba95fae2ce]
---
 script/process_perf_data.py | 4 ++--
 script/process_qa_data.sh   | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/script/process_perf_data.py b/script/process_perf_data.py
index b82a7c2891..3892206e42 100644
--- a/script/process_perf_data.py
+++ b/script/process_perf_data.py
@@ -133,12 +133,12 @@ def parse_logfile(logfile):
             if 'Best Perf' in line:
                 lst=line.split()
                 res.append(lst[4])
-    elif 'onnx_gemm' in logfile or 'mixed_gemm' in logfile:
+    elif 'onnx_gemm' in logfile:
         for line in open(logfile):
             if 'Best Perf' in line:
                 lst=line.split()
                 res.append(lst[33])
-    elif 'splitK_gemm' in logfile:
+    elif 'splitK_gemm' in logfile or 'mixed_gemm' in logfile:
         for line in open(logfile):
             if 'Best Perf' in line:
                 lst=line.split()
diff --git a/script/process_qa_data.sh b/script/process_qa_data.sh
index d6083d2fc7..c9a1645f6e 100755
--- a/script/process_qa_data.sh
+++ b/script/process_qa_data.sh
@@ -22,6 +22,7 @@ python3 process_perf_data.py perf_gemm_bilinear.log
 python3 process_perf_data.py perf_reduction.log
 python3 process_perf_data.py perf_splitK_gemm.log
 python3 process_perf_data.py perf_onnx_gemm.log
+python3 process_perf_data.py perf_mixed_gemm.log
 
 file=./perf_fmha_fwd_gfx942.log
 if [ -e "$file" ]; then