diff --git a/figures/plots.py b/figures/plots.py index 61c15ae..f66ac7e 100644 --- a/figures/plots.py +++ b/figures/plots.py @@ -5,12 +5,38 @@ import matplotlib.pyplot as plt plt.rcdefaults() fig, ax = plt.subplots() -systems = ('BW', "Minsky") -performance = (10, 100) +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, performance, align='center', color='gray', ecolor='black') +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("MLFMM Time (ms)") +ax.set_ylabel("Execution Time (ms)") # ax.set_title('How fast do you want to go today?') -plt.savefig('figures/test.pdf') \ No newline at end of file +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') \ No newline at end of file