pytorch

1
import matplotlib.pyplot as plt
2
import pandas
3

4

5
df = pandas.read_csv("perf.csv")
6

7
ops = pandas.unique(df["operator"])
8
nops = len(ops)
9
pivot_op_shape = df.pivot_table(
10
    values="time", index=["operator", "shape"], columns=["fuser"]
11
)
12
pivot_speedups = (pivot_op_shape.T / pivot_op_shape["eager"]).T
13

14
plt.rcParams["figure.figsize"] = (20, 100)
15
fig, axs = plt.subplots(nops)
16
plt.subplots_adjust(hspace=0.5)
17
for idx, op in enumerate(ops):
18
    op_speedups = pivot_speedups.T[op].T
19
    op_speedups.plot(ax=axs[idx], kind="bar", ylim=(0, 5), rot=45)
20
    axs[idx].set_title(op)
21
    axs[idx].set_xlabel("")
22
plt.savefig("scorecard.svg")
23