o j9:j] @sUdZddlZddlZddlZddlmZddlmZddlZ iZ e e e fed<eZee ed<de dBde fd d Zd e ddfd d Zdeejdee e effddZde dee e effddZ dde dedee e effddZdS)zG Utilities for converting data types into structured JSON for dumping. N)Sequence)Any INTERN_TABLE DUMPED_FILESsreturncsPdurdStdur&ttt<tjjjdfddddS)NstrcsfSNr rrr `/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/torch/_logging/structured.pyszintern_string..T)suppress_context)rgetlentorch_logging _internaltrace_structured)rr r r intern_strings rfilenamecsXdvrdStvr dStddlmtjjjdfddfddddS) N eval_with_keyr)_loader dump_filecsdiS)Nnamer r rr rr,szdump_file..cs Sr ) get_sourcer rrr rr/s ) metadata_fn payload_fn)raddtorch.fx.graph_modulerrrrrrr rrr"s     rtbcCsdd|D}|S)NcSs&g|]}|j|jt|j|jdqS))linerrloc)linenorrrr%.0framer r r 6sz"from_traceback..r )r$r r r rfrom_traceback3s r, num_framesc Csddlm}ddlm}|}|rt|d|dS|}tj t t tj j}tt|D]\}}||jvrT|t|||t||}t|Sq5t|d|dS)Nr)TracingContextCapturedTracebackr) torch._guardsr.torch.utils._tracebackr0 extract_stackr,extractsummaryospathdirnameinspectgetfilersep enumeratereversedrr) r-r.r0user_tbr$torch_filepathir* filtered_tbr r rget_user_stackBs      rBFcppcsNddlmddlm}|j|d}fdd|D}t|d|dS) zJ Returns the traceback for the user stack and the framework stack runinteresting_filesr/)rDcs<g|]}|jdr|jvsd|jvsd|jvr|qS)z.pyzat::ztorch::)rendswithrr(rEr rr+as  z'get_framework_stack..rN)%torch.fx.experimental.symbolic_shapesrFr2r0r4r5r,)r-rDr0r$r rErget_framework_stackWs    rI)rCF)__doc__r9r6 tracebackcollections.abcrtypingrtorch._logging._internalrrdictr int__annotations__setrrr FrameSummarylistr,rBboolrIr r r rs*  $