MD-Bench/util/plot_gather_data.py

107 lines
3.1 KiB
Python
Raw Normal View History

import matplotlib.pyplot as plt
import sys
filename = sys.argv[1]
plot_output_file = filename.replace(".txt", ".pdf")
raw_output_file = filename.replace(".txt", ".csv")
fig = plt.figure()
ax = plt.axes()
plot_data = {}
status = 0 # No header found
md_case = False
stride = None
dims = None
freq = None
cl_size = None
vector_width = None
cache_lines_per_gather = None
with open(filename, 'r') as fp:
for line in fp.readlines():
line = line.strip()
if len(line) <= 0 or "likwid-pin" in line:
continue
if line.startswith("ISA,"):
status = 1
md_case = True if "Dims" in line else False
continue
if line.startswith("N,"):
status = 2
continue
assert status == 1 or status == 2, "Invalid input!"
if status == 1:
if md_case:
isa, layout, stride, dims, freq, cl_size, vector_width, cache_lines_per_gather = line.split(',')
else:
isa, stride, freq, cl_size, vector_width, cache_lines_per_gather = line.split(',')
stride = int(stride)
continue
if md_case:
N, size, total_time, time_per_it, cy_per_iter, cy_per_gather, cy_per_elem = line.split(',')
else:
N, size, total_time, time_per_it, cy_per_gather, cy_per_elem = line.split(',')
size = float(size)
cycles = float(cy_per_iter) if md_case else float(cy_per_gather)
if stride not in plot_data:
plot_data[stride] = {}
plot_data[stride][size] = cycles if size not in plot_data[stride] \
else min(cycles, plot_data[stride][size])
all_sizes = set()
all_strides = set()
for stride in plot_data:
sizes = list(plot_data[stride].keys())
sizes.sort()
cycles = [plot_data[stride][size] for size in sizes]
ax.plot(sizes, cycles, marker='.', label=str(stride))
for size in sizes:
all_sizes.add(size)
all_strides.add(stride)
all_sizes = list(all_sizes)
all_sizes.sort()
all_strides = list(all_strides)
all_strides.sort()
with open(raw_output_file, 'w') as wp:
wp.write(" size\stride")
for stride in all_strides:
wp.write(",{0:14}".format(stride))
wp.write("\n")
for size in all_sizes:
wp.write("{:14.6f}".format(size))
for stride in all_strides:
try:
cycles = plot_data[stride][size]
except:
cycles = ''
wp.write(",{:14.6f}".format(cycles))
wp.write("\n")
cy_label = "Cycles per iteration" if md_case else "Cycles per gather"
ax.vlines([32, 1000], 0, 1, transform=ax.get_xaxis_transform(), linestyles='dashed', color=['#444444', '#777777'])
#ax.vlines([32, 1000, 28000], 0, 1, transform=ax.get_xaxis_transform(), linestyles='dashed', color=['#444444', '#777777', '#aaaaaa'])
ax.set(xlabel='Array size (kB)', ylabel=cy_label)
ax.set_xscale('log')
#ax.set_xticks([32, 1000, 28000])
#ax.set_xlim(0, 200000)
plt.legend(title="Stride")
fig.savefig(plot_output_file, bbox_inches = 'tight', pad_inches = 0)