Skip to content

GitLab

Unverified Commit 5fb150db authored by Illia Silin's avatar Illia Silin Committed by GitHub
Browse files

restore collecting performance of mixed prec gemms (#1648)

parent 8ef8a994
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
script/process_perf_data.py
View file @ 5fb150db
...@@ -133,12 +133,12 @@ def parse_logfile(logfile): ...@@ -133,12 +133,12 @@ def parse_logfile(logfile):
if 'Best Perf' in line: if 'Best Perf' in line:
lst=line.split() lst=line.split()
res.append(lst[4]) res.append(lst[4])
elif 'onnx_gemm' in logfile or 'mixed_gemm' in logfile: elif 'onnx_gemm' in logfile:
for line in open(logfile): for line in open(logfile):
if 'Best Perf' in line: if 'Best Perf' in line:
lst=line.split() lst=line.split()
res.append(lst[33]) res.append(lst[33])
elif 'splitK_gemm' in logfile: elif 'splitK_gemm' in logfile or 'mixed_gemm' in logfile:
for line in open(logfile): for line in open(logfile):
if 'Best Perf' in line: if 'Best Perf' in line:
lst=line.split() lst=line.split()
......
script/process_qa_data.sh
View file @ 5fb150db
...@@ -22,6 +22,7 @@ python3 process_perf_data.py perf_gemm_bilinear.log ...@@ -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_reduction.log
python3 process_perf_data.py perf_splitK_gemm.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_onnx_gemm.log
python3 process_perf_data.py perf_mixed_gemm.log
file=./perf_fmha_fwd_gfx942.log file=./perf_fmha_fwd_gfx942.log
if [ -e "$file" ]; then if [ -e "$file" ]; then
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment