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