hUddlZddlZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZgd Ziaeeefed <d ae d Gd deZe d GddZe d eddZe d deefdZe d dZe d dZe d deefdZe d defdZe d ed dZ e d deeeffdZ!e d de deeeffdZ"dS)!N)contextmanager)Enum)AnyOptionalUnion) compatibility)Graph)Node) preserve_node_metahas_preserved_node_metaset_stack_traceset_grad_fn_seq_nrreset_grad_fn_seq_nr format_stackset_current_metaget_current_meta NodeSourceNodeSourceActionget_graph_provenance_json current_metaF)is_backward_compatibleceZdZdZdZdS)rcreatereplaceN)__name__ __module__ __qualname__CREATEREPLACEZ/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/torch/fx/traceback.pyrrs FGGGr"rc <eZdZUdZGddZeed<eded<eded<eded< dd ee dedee dedffd Z e d efdZ e d efdZe d efdZdZdZddZd efdZd S)rz NodeSource is a data structure that contains the provenance information of a node. If node `a` is created from node `b`, then `a.meta["from_node"]` may contain NodeSource(b). c"eZdZdededefdZdS)NodeSource.NodeInfonametargetgraph_idc0||_||_||_dSNr'r(r))selfr'r(r)s r#__init__zNodeSource.NodeInfo.__init__-sDI DK$DMMMr"N)rrrstrintr.r!r"r#NodeInfor&,s= % %c %S % % % % % %r"r1 pass_nameraction from_node node_infoNnodec||_|g}nt|ts|g}|D]}t|tsJ||_|rw||jt|jt|j |_ d|j vrtj|j dng|_dSd|_ g|_dS)Nr,r4)r2 isinstancelistrr3r1r'r/r(idgraphr5metacopydeepcopyr4)r-r7r2r3as r#r.zNodeSource.__init__7s # >FFFD)) XF 3 3Aa!122 2 2 2 2  !]]Ys4;'7'7"TZ..+DN $)++ di 4555 NNN "DNDNNNr"returnc,|jr |jjndSNr6)r5r'r-s r#r'zNodeSource.nameSs&*nt~$$B>r"c,|jr |jjndS)N)r5r)rDs r#r)zNodeSource.graph_id[s*..@t~&&b@r"c*|Sr+)print_readablerDs r#__repr__zNodeSource.__repr___s""$$$r"cJdd|jDS)N+c@g|]}|jSr!)r'lower).0r@s r# z1NodeSource._get_action_string..cs"===A===r")joinr3rDs r#_get_action_stringzNodeSource._get_action_stringbs%xx=====>>>r"rc |dkrdSd}|}|d|zdzd|jd|jd|d|jd zz }|jD]}|||d zz }|S) N r6 z(name=z , pass_name=z , action=z , graph_id=z) r)rRr'r2r)r4rI)r-indentresult action_stringitems r#rIzNodeSource.print_readablees A::2//11  &L1 styssdnss}ssaeansss t N 6 6D d))&1*55 5FF r"c|}|j|j|j|j|d|jDdS)Nc6g|]}|Sr!to_dict)rOr7s r#rPz&NodeSource.to_dict..{s DDDT$,,..DDDr")r'r(r)r2r3r4)rRr'r(r)r2r4)r-rYs r#r^zNodeSource.to_dictrsL//11 Ik #DDT^DDD    r")r6N)r)rrr__doc__r1r/__annotations__r:rr rr.propertyr'r(r0r)rJrRrIdictr^r!r"r#rr%s %%%%%%%% NNN # $$$$L!!!! #### PT   tn  148J3KKLM     8=c===X=????X?A#AAAXA%%%???             r"rTc#K|sdVdSt}t} dadV|a|adS#|a|awxYw)NT)should_preserve_node_metarr>)enablesaved_should_preserve_node_metasaved_current_metas r#r r ss  . *C')..00 .(, % EEE(G %-LLL)H %-L - - - -s8>stackcZtr!|r!d|td<dSdSdS)Nr6 stack_trace)rdrQr)rhs r#rrsA!5U5&(ggenn ]###5555r"ctrOtdg|gztd<tdddztd<dSdS)Ngrad_fn_seq_nr in_grad_fnrrrdrget)seq_nrs r#rrsj!K)5)9)9:JB)O)O S * %&&2%5%5lA%F%F%J \""" KKr"ctrgtdd}|dksJ|dkrtd=td=dS|dz td<tdddtd<dSdS)NrmrrrlrGrn) current_levels r#rrs !Q$((q99 q    A  \*-...)6):L &-9:J-KCRC-PL) * * *QQr"rActrtddgStjtjddS)Nrjr6rG)rdrro traceback format_list extract_stackr!r"r#rrsI E  3344$Y%<%>%>ss%CDDDr"ctSr+)rdr!r"r#r r s $$r"r6c#Ktr[|jrTt} |jat ||t jgtd<dV|adS#|awxYwdVdS)Nr4)rdr=rr>rrr)r7r2 saved_metas r#rrs!TY!  &9>>++L4,<,CDD)L % EEE%LLL:L % % % % s AAA"ctSr+)rr!r"r#rrs r"r<ci}|jD]7}|jdkr*d|jvrd|jdDng||j<8|S)za Given an fx.Graph, return a json that contains the provenance information of each node. call_functionr4c6g|]}|Sr!r])rOsources r#rPz-get_graph_provenance_json..s"GGGf!!GGGr")nodesopr=r')r<provenance_tracking_jsonr7s r#rrsm "  7o % %$)++HG +0FGGGG %TY / $#r")T)r6)#r>rt contextlibrenumrtypingrrr_compatibilityr r<r r7r __all__rrbr/r`rdrrr r:rrrrboolr rrrr!r"r#rsS %%%%%%''''''''''))))))    " d38n!!!!e,,,t-, e,,,V V V V V V V -,V re,,,...-,.$e,,,549555-,5e,,,KK-,Ke,,, Q Q-, Qe,,,Ed3iEEE-,Ee,,,%%%%-,%e,,,-,*e,,,$sCx.-,e,,, $U $tCH~ $ $ $-, $ $ $r"