o Zho @sUdZddlZddlZddlZddlmZddlmZmZddl Z iZ e e efed<eZee ed<dee defd d Zd e ddfd d Zdeejdee e effddZdedee e effddZ ddededee e effddZdS)zG Utilities for converting data types into structured JSON for dumping. N)Sequence)AnyOptional INTERN_TABLE DUMPED_FILESsreturncsRdurdStddur'ttt<tjjjdfddddS)NstrcsfSNr rrr H/var/www/auris/lib/python3.10/site-packages/torch/_logging/structured.pyszintern_string..T)Zsuppress_context)rgetlentorch_logging _internaltrace_structured)rr r r intern_strings rfilenamecsXdvrdStvr dStddlmtjjjdfddfddddS) NZ eval_with_keyr)_loader dump_filecsdiS)Nnamer r rr rr+szdump_file..cs Sr ) get_sourcer rrr rr.s )Z metadata_fnZ payload_fn)raddZtorch.fx.graph_modulerrrrrrr rrr!s     rtbcCsdd|D}|S)NcSs&g|]}|j|jt|j|jdqS))linerrloc)linenorrrr!.0framer r r 5sz"from_traceback..r )r rr r rfrom_traceback2s r( num_framesc Csddlm}ddlm}|}|rt|d|dS|}tj t t tj j}tt|D]\}}||jvrT|t|||t||}t|Sq5t|d|dS)Nr)TracingContextCapturedTracebackr )Z torch._guardsr*torch.utils._tracebackr, extract_stackr(extractsummaryospathdirnameinspectgetfilersep enumeratereversedrr) r)r*r,Zuser_tbr Ztorch_filepathir&Z filtered_tbr r rget_user_stackAs      r:FcppcsNddlmddlm}|j|d}fdd|D}t|d|dS) zJ Returns the traceback for the user stack and the framework stack runinteresting_filesr+)r<cs<g|]}|jdr|jvsd|jvsd|jvr|qS)z.pyzat::ztorch::)rendswithrr$r=r rr'`s  z'get_framework_stack..r N)Z%torch.fx.experimental.symbolic_shapesr>r-r,r/r0r()r)r<r,r r r=rget_framework_stackVs    r@)r;F)__doc__r4r1 tracebackcollections.abcrtypingrrZtorch._logging._internalrrdictr int__annotations__setrrr FrameSummarylistr(r:boolr@r r r rs* $