קga UddlZddlmZddlmZmZmZddlmZgdZ ia ee efe d<da ed ed Zed d ee fd Zed d Zed dZed dee fdZed defdZed edZed dee effdZdS)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#rKt}t} dadV|a|adS#|a|awxYw)NT)should_preserve_node_metarcopy)saved_should_preserve_node_metasaved_current_metas N/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/fx/traceback.pyr r s_ '@#%**,,*$(! $C!) %D!) ))))s06stackcZtr!|r!d|td<dSdSdS)N stack_trace)rjoinr)rs rr r sA!5U5&(ggenn ]###5555ctrOtdg|gztd<tdddztd<dSdS)Ngrad_fn_seq_nr in_grad_fnrrrrget)seq_nrs rr r (sc!K)5)9)9:JB)O)OSYRZ)Z %&%1%5%5lA%F%F%J \"""KKrctrgtdd}|dksJ|dkrtd=td=dS|dz td<tdddtd<dSdS)Nr!rrr r") current_levels rr r 2s !Q$((q99 q    A  \*-...)6):L &-9:J-KCRC-PL) * * *QQrreturnctrtddgStjtjddS)Nrrr&)rrr# traceback format_list extract_stackrrrrBsI E  3344$Y%<%>%>ss%CDDDrctSN)rr-rrr r Ks $$rc#^Ktr|jrt} |jadtvr|j|jfgtd<nGtddd|jkr%td|j|jfgztd<dV|adS#|awxYwdVdS)N from_noder&r)rmetarrnametarget)node saved_metas rrrPs!TY!  &9>>++L,...2i-E,F [))k*2.q1TY>>,8,E$)UYU`IaHb,b [) EEE%LLL:L % % % % s BB""B&ctSr/)rr-rrrrfs r)r* contextlibrtypingrrr_compatibilityr__all__rstr__annotations__rr r r r rboolr rrr-rrr?s;%%%%%%"""""""""")))))) C C C " d38n!!!!e,,, * *-, *e,,,5DI555-,5e,,,KK-,Ke,,, Q Q-, Qe,,,Ed3iEEE-,Ee,,,%%%%-,%e,,,-,(e,,,$sCx.-,r