42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import matplotlib.pyplot as plt
|
|
plt.rcdefaults()
|
|
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
plt.rcdefaults()
|
|
|
|
fig, ax = plt.subplots()
|
|
systems = ('BW (32T)', "Minsky (160T)")
|
|
mlfmm = (8.65e4, 4.71e4)
|
|
total = (1.2e5, 5.6e4)
|
|
x_pos = np.arange(len(systems))
|
|
ax.bar(x_pos, mlfmm, color='gray')
|
|
ax.bar(x_pos, [i-j for i,j in zip(total, mlfmm)], color='lightgray', bottom=mlfmm )
|
|
ax.set_xticks(x_pos)
|
|
ax.set_xticklabels(systems)
|
|
ax.set_ylabel("Execution Time (ms)")
|
|
# ax.set_title('How fast do you want to go today?')
|
|
plt.savefig('figures/cpu_matvec.pdf')
|
|
|
|
fig, ax = plt.subplots()
|
|
systems = ('BW (1T)', "BW (32T)", "BW (1 GPU)" ,"BW (4 GPU)", "BW (16 GPU)")
|
|
mlfmm = (1.50e6, 8.64e4, 8.64e4, 8.64e4, 8.64e4)
|
|
x_pos = np.arange(len(systems))
|
|
ax.bar(x_pos, mlfmm, color='gray', log=True)
|
|
ax.set_xticks(x_pos)
|
|
ax.set_xticklabels(systems)
|
|
ax.set_ylabel("Total MLFMM Execution Time (ms)")
|
|
plt.ylim([1, 1e7])
|
|
# ax.set_title('How fast do you want to go today?')
|
|
plt.savefig('figures/mlfmm_bw.pdf')
|
|
|
|
fig, ax = plt.subplots()
|
|
systems = ('1T', "160T", "1 GPU" ,"4 GPU")
|
|
mlfmm = (1.25e6, 4.71e4, 4.71e4, 4.71e4)
|
|
x_pos = np.arange(len(systems))
|
|
ax.bar(x_pos, mlfmm, color='gray', log=True)
|
|
ax.set_xticks(x_pos)
|
|
ax.set_xticklabels(systems)
|
|
ax.set_ylabel("Total MLFMM Execution Time (ms)")
|
|
plt.ylim([1, 1e7])
|
|
# ax.set_title('How fast do you want to go today?')
|
|
plt.savefig('figures/mlfmm_minsky.pdf') |