U 3þjga ã@s&UddlZddlmZddlmZmZmZddlmZdddd d d d d gZ ia ee efe d<da eddedd„ƒƒZeddee dœdd„ƒZedddd „ƒZedddd „ƒZeddee dœdd „ƒZeddedœdd„ƒZeddedd „ƒƒZeddee efdœdd „ƒZdS)éN)Úcontextmanager)ÚListÚAnyÚDicté)Ú compatibilityÚpreserve_node_metaÚhas_preserved_node_metaÚset_stack_traceÚset_grad_fn_seq_nrÚreset_grad_fn_seq_nrÚ format_stackÚset_current_metaÚget_current_metaÚ current_metaF)Zis_backward_compatibleccs*t}t ¡}zdadVW5|a|aXdS)NT)Úshould_preserve_node_metarÚcopy)Zsaved_should_preserve_node_metaZsaved_current_meta©rú6/tmp/pip-unpacked-wheel-ttp2cnii/torch/fx/traceback.pyrs ©ÚstackcCstr|rd |¡td<dS)NÚÚ stack_trace)rÚjoinrrrrrr scCs2tr.t dg¡|gtd<t dd¡dtd<dS)NÚgrad_fn_seq_nrÚ in_grad_fnrr)rrÚget)Zseq_nrrrrr (scCsVtrRt dd¡}|dkst‚|dkr2td=td=n |dtd<tddd…td<dS)Nrrrréÿÿÿÿ)rrrÚAssertionError)Z current_levelrrrr 2s   )ÚreturncCs,trt dd¡gSt t ¡dd…¡SdS)Nrrr)rrrÚ tracebackÚ format_listÚ extract_stackrrrrr BscCstS©N)rrrrrr Ksccs‚trx|jrxt}z`|j ¡adtkr6|j|jfgtd<n0tddd|jkrftd|j|jfgtd<dVW5|aXndVdS)NZ from_noderr)rÚmetarrÚnameÚtarget)ÚnodeZ saved_metarrrrPs   cCstSr#)rrrrrrfs)r Ú contextlibrÚtypingrrrZ_compatibilityrÚ__all__rÚstrÚ__annotations__rrr r r r Úboolr rrrrrrÚs@  þ