mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
e0bf51d3bf
I want to be able to decouple DDP graph printing from the rest of dynamo DEBUG-level logging, since frequently these logs are particularly enlightening. Differential Revision: [D45290919](https://our.internmc.facebook.com/intern/diff/D45290919/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/100021 Approved by: https://github.com/wconstab, https://github.com/mlazos
18 lines
635 B
Python
18 lines
635 B
Python
from ._internal import register_artifact, register_log
|
|
|
|
register_log("dynamo", "torch._dynamo")
|
|
register_log("aot", "torch._functorch.aot_autograd")
|
|
register_log("inductor", "torch._inductor")
|
|
register_log("dynamic", "torch.fx.experimental.symbolic_shapes")
|
|
|
|
register_artifact("guards")
|
|
register_artifact("bytecode", off_by_default=True)
|
|
register_artifact("graph")
|
|
register_artifact("graph_code")
|
|
register_artifact("aot_graphs")
|
|
register_artifact("aot_joint_graph")
|
|
register_artifact("ddp_graphs")
|
|
register_artifact("recompiles")
|
|
register_artifact("output_code", off_by_default=True)
|
|
register_artifact("schedule", off_by_default=True)
|