mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Apply UFMT to all non test/torch files (#106205)
Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/106205 Approved by: https://github.com/albanD
This commit is contained in:
committed by
PyTorch MergeBot
parent
1163800d0f
commit
e6ec0efaf8
@ -1,11 +1,13 @@
|
||||
import pandas
|
||||
import matplotlib.pyplot as plt
|
||||
import pandas
|
||||
|
||||
df = pandas.read_csv("perf.csv")
|
||||
|
||||
ops = pandas.unique(df["operator"])
|
||||
nops = len(ops)
|
||||
pivot_op_shape = df.pivot_table(values="time", index=["operator", "shape"], columns=["fuser"])
|
||||
pivot_op_shape = df.pivot_table(
|
||||
values="time", index=["operator", "shape"], columns=["fuser"]
|
||||
)
|
||||
pivot_speedups = (pivot_op_shape.T / pivot_op_shape["eager"]).T
|
||||
|
||||
plt.rcParams["figure.figsize"] = (20, 100)
|
||||
|
Reference in New Issue
Block a user