67 lines
1.3 KiB
Python
67 lines
1.3 KiB
Python
import pandas as pd
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib
|
|
from matplotlib.ticker import ScalarFormatter, FormatStrFormatter
|
|
|
|
|
|
matplotlib.style.use('ggplot')
|
|
|
|
# Roofline Plot
|
|
|
|
#log x
|
|
i = 32
|
|
xlbl = []
|
|
while i > 1:
|
|
xlbl.append("1/"+repr(i))
|
|
i //= 2
|
|
xlbl.append(repr(i)) #1
|
|
i += 1
|
|
while i<=64:
|
|
xlbl.append(repr(i))
|
|
i *= 2
|
|
|
|
# memory
|
|
values = []
|
|
bandwidth = 10.6
|
|
peak = 86.4
|
|
ymem = []
|
|
ypeak = []
|
|
|
|
for i in np.arange(0,64,0.1):
|
|
if bandwidth*i < peak:
|
|
values.append(bandwidth*i)
|
|
else:
|
|
values.append(peak)
|
|
|
|
i=1/32
|
|
while i<=64:
|
|
if bandwidth*i < peak and bandwidth*i*2 < peak:
|
|
ymem.append(bandwidth*i)
|
|
ypeak.append(None)
|
|
elif bandwidth*i < peak and bandwidth*i*2 > peak:
|
|
ypeak.append(peak)
|
|
ymem.append(bandwidth*i)
|
|
else:
|
|
ypeak.append(peak)
|
|
ymem.append(None)
|
|
i*=2
|
|
|
|
#plot data
|
|
#data = pd.Series(data=values, name='Peak Memory Bandwidth', index=np.arange(0,64,0.1))
|
|
|
|
data = {'Peak Memory Bandwidth': pd.Series(ymem, index=xlbl), 'Peak Floating-Point Performance': pd.Series(ypeak, index=xlbl)}
|
|
df = pd.DataFrame(data)
|
|
|
|
ax = df.plot()
|
|
ax.set_xlabel("Operational Itensity (FLOP/Byte)")
|
|
ax.set_ylabel("Attainable GFLOP/s")
|
|
ax.set_yscale('log', basey=2)
|
|
ax.yaxis.set_major_formatter(FormatStrFormatter('%.0f'))
|
|
|
|
print(repr(data))
|
|
|
|
|
|
|
|
|
|
plt.show()
|