| 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) |
