P/ja UddlZddlmZddlmZmZmZddlmZgdZ ia ee efe d<da ed ed Zed d ee fd Zed d Zed dZed dee fdZed defdZed edZed dee effdZy)N)contextmanager)ListAnyDict) compatibility)preserve_node_metahas_preserved_node_metaset_stack_traceset_grad_fn_seq_nrreset_grad_fn_seq_nr format_stackset_current_metaget_current_meta current_metaF)is_backward_compatiblec#hKt}tj} dad|a|ay#|a|awxYww)NT)should_preserve_node_metarcopy)saved_should_preserve_node_metasaved_current_metas U/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/fx/traceback.pyr r sF '@#%**,*$(! $C!) %D!) s2)2/2stackcHtr|rdj|td<yyy)N stack_trace)rjoinr)rs rr r s#!U&(ggen ]#&+ ctrBtjdg|gztd<tjdddztd<yy)Ngrad_fn_seq_nr in_grad_fnrrrrget)seq_nrs rr r (sJ!)5)9)9:JB)OSYRZ)Z %&%1%5%5lA%F%J \"!rctrQtjdd}|dkDsJ|dk(rtd=td=y|dz td<tdddtd<yy)Nr!rrr r") current_levels rr r 2sl !$((q9 q   A \*-.)6):L &-9:J-KCR-PL) *!rreturnctrtjddgStjtj ddS)Nrrr&)rrr# traceback format_list extract_stackrrrrBs>   344$$Y%<%<%>s%CDDrctSN)rr-rrr r Ks $$rc#tKtr|jrt} |jjadtvr!|j|j fgtd<nFtddd|jk7r*td|j|j fgztd<d|aydy#|awxYww)N from_noder&r)rmetarrnametarget)node saved_metas rrrPs!TYY!  &99>>+L,..2ii-E,F [)k*2.q1TYY>,8,E$))UYU`U`IaHb,b [) %L &LsB8B B1)B81B55B8ctSr/)rr-rrrrfs r)r* contextlibrtypingrrr_compatibilityr__all__rstr__annotations__rr r r r rboolr rrr-rrr?sA%"") C " d38n!!e, *- *e,5DI5-5e,K-Ke, Q- Qe,Ed3iE-Ee,%%-%e,-(e,$sCx.-r