o H&i@sddlmZmZddgZddgZeddgeedd d ged d ed eeddedeeddeddddeddddeddddeddeddedd ed!d"ed#d$ed%d&dded'd(ed)d*ed+d,dded-d.ed/d0dded1d2ddd3ed4d5dded6d7ed8d9ddd:ed;dddded?d@ddedAdBddedCdDddEdFS)G)register_artifact register_logz%torch.fx.experimental.symbolic_shapesztorch.fx.experimental.sym_nodeztorch.distributedz"torch._dynamo.backends.distributedZdynamoz torch._dynamoZaotztorch._functorch.aot_autogradztorch._functorch._aot_autogradZinductorztorch._inductorZdynamicZtorch distributedZonnxz torch.onnxZguardszhThis prints the guards for every compiled Dynamo frame. It does not tell you where the guards come from.T)visibleZverbose_guards)off_by_defaultbytecodez{Prints the original and modified bytecode from Dynamo. Mostly useful if you're debugging our bytecode generation in Dynamo.graphzvPrints the dynamo traced graph (prior to AOTDispatch) in a table. If you prefer python code use `graph_code` instead. Z graph_codez4Like `graph`, but gives you the Python code instead.Z graph_sizesz5Prints the sizes of all FX nodes in the dynamo graph.Z trace_sourcezAs we execute bytecode, prints the file name / line number we are processing and the actual source code. Useful with `bytecode`Z trace_callzhLike trace_source, but it will give you the per-expression blow-by-blow if your Python is recent enough.Z aot_graphszPrints the FX forward and backward graph generated by AOTDispatch, after partitioning. Useful to understand what's being given to InductorZaot_joint_graphz_Print FX joint graph from AOTAutograd, prior to partitioning. Useful for debugging partitioningZpost_grad_graphsz}Prints the FX graph generated by post grad passes. Useful to understand what's being given to Inductor after post grad passesZcompiled_autogradzzPrints various logs in compiled_autograd, including but not limited to the graphs. Useful for debugging compiled_autograd.Z ddp_graphszOnly relevant for compiling DDP. DDP splits into multiple graphs to trigger comms early. This will print each individual graph here.Z recompilesz?Prints the reason why we recompiled a graph. Very, very useful.Zrecompiles_verbosezPrints all guard checks that fail during a recompilation. At runtime, Dynamo will stop at the first failed check for each failing guard. So not all logged failing checks are actually ran by Dynamo.)rrZ graph_breakszPrints whenever Dynamo decides that it needs to graph break (i.e. create a new graph). Useful for debugging why torch.compile has poor performancenot_implementedzPrints log messages whenever we return NotImplemented in a multi-dispatch, letting you trace through each object we attempted to dispatch toZ output_codez>Prints the code that Inductor generates (either Triton or C++))rrZschedulezIInductor scheduler information. Useful if working on Inductor fusion algoZ perf_hintsZonnx_diagnosticsZfusionzADetailed Inductor fusion decisions. More detailed than 'schedule'overlapz0Detailed Inductor compute/comm overlap decisionsZcustom_format_test_artifactz Testing only)Z log_formatN) _internalrrZDYNAMICZ DISTRIBUTEDr r HC:\wamp64\www\opt\env\Lib\site-packages\torch/_logging/_registrations.pys