1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# Script to make the comparison plot for the benchmark
import matplotlib.pyplot as plt
import numpy as np
(
size,
np_1d_min,
np_1d_mean,
np_1d_median,
fa_1d_min,
fa_1d_mean,
fa_1d_median,
np_2d_min,
np_2d_mean,
np_2d_median,
fa_2d_min,
fa_2d_mean,
fa_2d_median,
) = np.loadtxt("benchmark_times.txt", unpack=True)
fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)
ax.plot(size, np_1d_min / fa_1d_min, color=(34 / 255, 122 / 255, 181 / 255), label="1D")
ax.plot(size, np_2d_min / fa_2d_min, color=(255 / 255, 133 / 255, 25 / 255), label="2D")
ax.set_xscale("log")
ax.set_xlim(0.3, 3e8)
ax.set_ylim(1, 35)
ax.grid()
ax.set_xlabel("Array size")
ax.set_ylabel("Speedup (fast-histogram / numpy)")
ax.legend()
fig.savefig("speedup_compared.png", bbox_inches="tight")
|
