o h@sUddlZddlZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZgd Ziaeeefed <d ae d d Gd ddeZe d d GdddZe d d ed(ddZe d d deefddZe d d ddZe d d ddZe d d deefddZe d d defddZe d d ed)d!d"Z e d d deeeffd#d$Z!e d d d%e deeeffd&d'Z"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_compatiblec@seZdZdZdZdS)rcreatereplaceN)__name__ __module__ __qualname__CREATEREPLACErrf/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/fx/traceback.pyrsrc @seZdZUdZGdddZeed<eded<eded<eded< dd ee dedee dedffd d Z e defddZ e defddZe defddZddZddZdddZdefddZd 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#rrr __init__-s zNodeSource.NodeInfo.__init__N)rrrstrintr'rrrr NodeInfo,sr* pass_nameraction from_node node_infoNnodecCs||_|dur g}nt|ts|g}|D] }t|tsJq||_|rH|j|jt|jt |j d|_ d|j vrCt |j d|_dSg|_dSd|_ g|_dS)Nr%r-)r+ isinstancelistrr,r*r!r(r"idgraphr.metacopydeepcopyr-)r&r0r+r,arrr r'7s&   zNodeSource.__init__returncC|jr|jjSdSNr/)r.r!r&rrr r!SzNodeSource.namecCr:r;)r.r"r<rrr r"Wr=zNodeSource.targetcCr:)N)r.r#r<rrr r#[r=zNodeSource.graph_idcCs|Sr$)print_readabler<rrr __repr___szNodeSource.__repr__cCsddd|jDS)N+cSsg|]}|jqSr)r!lower).0r8rrr csz1NodeSource._get_action_string..)joinr,r<rrr _get_action_stringbszNodeSource._get_action_stringrc Csn|dkrdSd}|}|d|dd|jd|jd|d|jd 7}|jD] }|||d 7}q)|S) N r/ z(name=z , pass_name=z , action=z , graph_id=z) r)rFr!r+r#r-r?)r&indentresult action_stringitemrrr r?es " zNodeSource.print_readablecCs.|}|j|j|j|j|dd|jDdS)NcSg|]}|qSrto_dict)rCr0rrr rD{z&NodeSource.to_dict..)r!r"r#r+r,r-)rFr!r"r#r+r-)r&rLrrr rPrszNodeSource.to_dict)r/N)r)rrr__doc__r*r(__annotations__r2rr rr'propertyr!r"r)r#r@rFr?dictrPrrrr r%s4       rTccs@|sdVdSt}t}z dadVW|a|adS|a|aw)NT)should_preserve_node_metarr6)enablesaved_should_preserve_node_metasaved_current_metarrr r s r stackcCs"tr |rd|td<dSdSdS)Nr/ stack_trace)rVrEr)rZrrr r sr cCs6trtdg|gtd<tdddtd<dSdS)Ngrad_fn_seq_nr in_grad_fnrrrVrget)seq_nrrrr rs   rcCs\tr,tdd}|dksJ|dkrtd=td=dS|dtd<tdddtd<dSdS)Nr]rrr\r>r^) current_levelrrr rs    rr9cCs(tr tddgSttddS)Nr[r/r>)rVrr_ traceback format_list extract_stackrrrr rsrcCtSr$)rVrrrr r r r/ccsPtr#|jr#t}z|jat||tjgtd<dVW|adS|awdVdS)Nr-)rVr5rr6rrr)r0r+ saved_metarrr rs    rcCrer$)rrrrr rrfrr4cCsFi}|jD]}|jdkr d|jvrdd|jdDng||j<q|S)za Given an fx.Graph, return a json that contains the provenance information of each node. call_functionr-cSrNrrO)rCsourcerrr rDrQz-get_graph_provenance_json..)nodesopr5r!)r4provenance_tracking_jsonr0rrr rs   r)T)r/)#r6rb contextlibrenumrtypingrrr_compatibilityrr4r r0r __all__rrUr(rSrVrrr r2r rrrboolr rrrrrrr sF      Y