o ^i+@sUddlZddlZddlZddlmZddlmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZeeZgd Ziaeeefed <d aed d GdddeZed d GdddZed d ed)ddZed d deefddZ ed d ddZ!ed d ddZ"ed d deefddZ#ed d de$fdd Z%ed d ed*d"d#Z&ed d deeeffd$d%Z'ed d d&edeeeffd'd(Z(dS)+N)contextmanager)Enum)AnyOptionalUnion)signpost_event) 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_compatiblec@seZdZdZdZdS)rcreatereplaceN)__name__ __module__ __qualname__CREATEREPLACEr r M/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/torch/fx/traceback.pyr$src @s.eZdZUdZGdddZeed<eded<eded<eded<ee ee fed <eeed < d(d ee dedee dedffddZ edefddZedefddZedefddZddZddZd)ddZde fddZd efd!d"Zd#d$Zed%ee dedfd&d'Zd 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@s"eZdZdededefddZdS)zNodeSource.NodeInfonametargetgraph_idcCs||_||_||_dSNr"r#r$)selfr"r#r$r r r!__init__2s zNodeSource.NodeInfo.__init__N)rrrstrintr(r r r r!NodeInfo1sr+ pass_nameraction from_node node_info_dict_action_stringNnodecCs||_|dur g}nt|ts|g}|D] }t|tsJq||_|rD|j|jt|jt |j d|_ d|j vr@t |j dng|_nd|_ g|_d|_d|_dS)Nr&r.)r, isinstancelistrr-r+r"r)r#idgraphr/metacopydeepcopyr.r1r0)r'r3r,r-ar r r!r(>s(   zNodeSource.__init__returncC|jr|jjSdSNr2)r/r"r'r r r!r"^zNodeSource.namecCr=r>)r/r#r?r r r!r#br@zNodeSource.targetcCr=)N)r/r$r?r r r!r$fr@zNodeSource.graph_idcCs|Sr%)print_readabler?r r r!__repr__jszNodeSource.__repr__cCs(|jdurddd|jD|_|jS)N+cSsg|]}|jqSr )r"lower).0r;r r r! osz1NodeSource._get_action_string..)r1joinr-r?r r r!_get_action_stringms zNodeSource._get_action_stringrc Csn|dkrdSd}|}|d|dd|jd|jd|d|jd 7}|jD] }|||d 7}q)|S) N r2 z(name=z , pass_name=z , action=z , graph_id=z) r)rIr"r,r$r.rB)r'indentresult action_stringitemr r r!rBrs " zNodeSource.print_readablecCsN|jdur|}|j|j|j|j|dd|jDd|_|jdus$J|jS)NcSg|]}|qSr to_dict)rFr3r r r!rGz&NodeSource.to_dict..)r"r#r$r,r-r.)r0rIr"r#r$r,r.)r'rOr r r!rSs  zNodeSource.to_dictothercCst|tsdS||kS)NF)r4rrS)r'rUr r r!__eq__s zNodeSource.__eq__csfddt|S)NcsLt|trttfdd|DSt|tr$tfdd|DS|S)Nc3s |] \}}||fVqdSr%r )rFkv_make_hashabler r! sz>NodeSource.__hash__.._make_hashable..c3s|]}|VqdSr%r )rFrPrYr r!r[s)r4dicttuplesorteditemsr5)objrYr r!rZs  z+NodeSource.__hash__.._make_hashable)hashrSr?r rYr!__hash__s zNodeSource.__hash__dcs0|durdSt|tsJdt|tt}d|_d|_|dd|_|dd}g}|rQ| dD]}| dkrD| t j q5| dkrP| t jq5||_d |vrwd |vrwd |vrwt|d d|d d|d d }||_nd|_|d ddurfdd|d gD|_|Sg|_|S)z Recursively deserialize from_node metadata from dictionary data. It is used to deserialize the from_node field from serialized metadata. Please use constructor NodeSource(node, ...) to create a NodeSource object. NzExpected a dict, got r,r2r-rDrrr"r#r$rAr.cs"g|] }|durqSr%) _from_dict)rFfnclsrNr r!rGs z)NodeSource._from_dict..)r4r\typer__new__r1r0getr,splitupperappendrrrr-r+r/r.)rgrc node_source action_stractions action_namer/r rfr!rds<       zNodeSource._from_dict)r2N)r)rrr__doc__r+r)__annotations__r5rr\rr rr(propertyr"r#r*r$rCrIrBrSobjectrVrb classmethodrdr r r r!r*s@          rTccs@|sdVdSt}t}z dadVW|a|adS|a|aw)NT)should_preserve_node_metarr9)enablesaved_should_preserve_node_metasaved_current_metar r r!r s r stackcCs"tr |rd|td<dSdSdS)Nr2 stack_trace)rwrHr)r{r r r!rsrcCs6trtdg|gtd<tdddtd<dSdS)Ngrad_fn_seq_nr in_grad_fnrrrwrrj)seq_nrr r r!rs   rcCs\tr,tdd}|dksJ|dkrtd=td=dS|dtd<tdddtd<dSdS)Nr~rrr}rAr) current_levelr r r!rs    rr<cCs(tr tddgSttddS)Nr|r2rA)rwrrj traceback format_list extract_stackr r r r!rsrcCtSr%)rwr r r r!r r r2ccsPtr#|jr#t}z|jat||tjgtd<dVW|adS|awdVdS)Nr.)rwr8rr9rrr)r3r, saved_metar r r!rs    rcCrr%)rr r r r!r5rrr7c Csz$i}|jD]}|jdkr!d|jvrdd|jdDng||j<q|WStyF}ztdddt|tdiWYd }~Sd }~ww) za Given an fx.Graph, return a json that contains the provenance information of each node. call_functionr.cSrQr rR)rFsourcer r r!rGDrTz-get_graph_provenance_json..inductorprovenance_tracking_errorr)function error_msgr|N) nodesopr8r" Exceptionrr)r format_exc)r7provenance_tracking_jsonr3er r r!r:s,    r)T)r2))r9loggingr contextlibrenumrtypingrrrtorch._utils_internalr_compatibilityr r7r r3r getLoggerrlog__all__rr\r)rsrwrrr r5rrrrboolr rrrr r r r!sN        .