mirror of
https://github.com/vortexgpgpu/vortex.git
synced 2025-04-24 22:07:41 -04:00
54 lines
1.2 KiB
Python
Executable file
54 lines
1.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import sys
|
|
import pandas as pd
|
|
from os import path
|
|
from glob import glob
|
|
|
|
if len(sys.argv) < 2:
|
|
print('usage: python3 ' + sys.argv[0] + ' <path to test_outputs>')
|
|
exit()
|
|
|
|
output_dir = sys.argv[1]
|
|
|
|
|
|
config_names = []
|
|
test_names = []
|
|
cycle_counts = []
|
|
|
|
for filename in glob(path.join(output_dir, '*.log')):
|
|
cycle_line = None
|
|
with open(filename, 'r') as f:
|
|
for line in f:
|
|
line = line.strip()
|
|
if line.startswith('[sim] total cycles:'):
|
|
cycle_line = line
|
|
print(filename, cycle_line)
|
|
|
|
full_name, _, _ = path.basename(filename).partition('.')
|
|
|
|
if cycle_line is None:
|
|
count = None
|
|
else:
|
|
_, _, count = cycle_line.partition(':')
|
|
count = int(count.strip())
|
|
|
|
config, test = full_name.rsplit('-', 1)
|
|
config_names.append(config)
|
|
test_names.append(test)
|
|
cycle_counts.append(count)
|
|
|
|
df = pd.DataFrame({
|
|
'config': config_names,
|
|
'test': test_names,
|
|
'cycle_count': cycle_counts,
|
|
})
|
|
|
|
print(df.head())
|
|
|
|
pivot = pd.pivot_table(df, values='cycle_count', index=['config'], columns=['test'])
|
|
|
|
print(pivot.head())
|
|
|
|
pivot.to_csv('results.csv')
|
|
print('Table written to results.csv')
|
|
|