mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 13:41:24 +00:00
a3c910ac6c
* add performance test to jenkins pipeline * fix typo * fix the syntax in conv_fwd_util.cpp * fix the error message syntax spacing * fix the error message syntax spacing again * run profile_gemm and archive results * fix typo * try to figure out the paths * try to figure out the paths one more time * skip the copying step * build ckProfiler release only once * change directory using dir * fix dir syntax * change the gemm parameters * do not pipe script output to file * try running ckProfiler directly * fix typo * use set +e * run profile_gemm.sh || true * run multiple gemms and parse results * fix typo in jenkinsfile * fix syntax * add new gemm sizes, update scripts * put all jenkins steps in original order Co-authored-by: Chao Liu <chao.liu2@amd.com> Co-authored-by: Chao Liu <lc.roy86@gmail.com>
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
import os, io
|
|
import argparse
|
|
|
|
def print_to_string(*args, **kwargs):
|
|
output = io.StringIO()
|
|
print(*args, file=output, **kwargs)
|
|
contents = output.getvalue()
|
|
output.close()
|
|
return contents
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description='Parse results from tf benchmark runs')
|
|
parser.add_argument('filename', type=str, help='Log file to prase or directory containing log files')
|
|
args = parser.parse_args()
|
|
files = []
|
|
if os.path.isdir(args.filename):
|
|
all_files = os.listdir(args.filename)
|
|
for name in all_files:
|
|
if not 'log' in name:
|
|
continue
|
|
files.append(os.path.join(args.filename, name))
|
|
else:
|
|
files = [args.filename]
|
|
args.files = files
|
|
return args
|
|
|
|
def main():
|
|
args = parse_args()
|
|
results = []
|
|
#parse results
|
|
glue=""
|
|
for filename in args.files:
|
|
for line in open(filename):
|
|
if 'Best Perf' in line:
|
|
lst=line.split()
|
|
results.append(print_to_string(glue.join(lst[8:]),lst[4]))
|
|
|
|
#sort results
|
|
|
|
#read baseline results for the latest develop branch
|
|
|
|
#write new results to the db
|
|
|
|
#compare the results to the baseline
|
|
|
|
#return 0 if performance criteria met, otherwise return 1
|
|
|
|
print(results)
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
main() |