a ho @sUdZddlZddlZddlZddlmZddlmZmZddl Z iZ e e efed<eZee ed<ee eddd Ze dd d d Zeejee e efd ddZeee e efdddZdeeee e efdddZdS)zG Utilities for converting data types into structured JSON for dumping. N)Sequence)AnyOptional INTERN_TABLE DUMPED_FILES)sreturncsRdur dStddurNttt<tjjjdfddddS)NstrcsfSNr rrr G/var/www/auris/lib/python3.9/site-packages/torch/_logging/structured.pyzintern_string..T)Zsuppress_context)rgetlentorch_logging _internaltrace_structured)rr r r intern_strings r)filenamercsXdvr dStvrdStddlmtjjjdfddfddddS) NZ eval_with_keyr)_loader dump_filecsdiS)Nnamer r rr rr+szdump_file..cs Sr ) get_sourcer rrr rr.r)Z metadata_fnZ payload_fn)raddZtorch.fx.graph_modulerrrrrrr rrr!s    r)tbrcCsdd|D}|S)NcSs&g|]}|j|jt|j|jdqS))linerrloc)linenorrrr".0framer r r 5s z"from_traceback..r )r!rr r rfrom_traceback2s r)) num_framesrc Csddlm}ddlm}|}|r8t|d|dS|}tj t t tj j}tt|D]>\}}||jvrj|t|||t||}t|Sqjt|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;F)r*cpprcsNddlmddlm}|j|d}fdd|D}t|d|dS) zJ Returns the traceback for the user stack and the framework stack runinteresting_filesr,)r=cs<g|]4}|jdr |jvs4d|jvs4d|jvr|qS)z.pyzat::ztorch::)rendswithrr%r>r rr(`s   z'get_framework_stack..r N)Z%torch.fx.experimental.symbolic_shapesr?r.r-r0r1r))r*r=r-r!r r>rget_framework_stackVs    rA)r<F)__doc__r5r2 tracebackcollections.abcrtypingrrZtorch._logging._internalrrdictr int__annotations__setrrr FrameSummarylistr)r;boolrAr r r rs" "