restore collecting performance of mixed prec gemms (#1648)

This commit is contained in:
Illia Silin
2024-11-11 09:25:08 -08:00
committed by GitHub
parent 8ef8a994e7
commit 5fb150dbe7
2 changed files with 3 additions and 2 deletions

View File

@@ -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()