Yh(ddlZddlZddlZddlZddlZddlmZddlm Z ej dZ dddZ dddd Z d ZGd d Zd ZdS)N) TracebackType)Optionalc#K dVdS#t$r}|j}g}|w|jjj}|jjd}|dkr!|tjddd5}| |dddn #1swxYwY|j}td|j d}| |jj }t|jd r+| |jj|jj }t#||ji|jd t&ji}t+d||j|j} || n|||j}|wd} t5|D] }| |_|} || d}~wwxYw) N__compile_source__zwFz.py)modedeletesuffixz__inspect_currentframe()eval)co_name co_linetable)r co_firstlineno__inspect_currentframe) Exception __traceback__tb_framef_code co_filename f_globalsgettempfileNamedTemporaryFilewritecompilenamereplacer hasattrr rr f_localsinspect currentframertb_lasti tb_linenoappendtb_nextreversedwith_traceback) exctbstackfilenamesourcefframecode fake_framefake_tbr$s h/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/utils/_traceback.pyreport_compile_source_on_errorr21s<P*  N*N*N*  n{)5H[*../CDDF:%%&*<20U5$GGFOOO$$$$$$$$$$$$$$$  9166JJ||EL,@|AA5<88 <<%*\%>',|'B(D"OVu~V'?AUVV (j"+r|TT W%%%% R   BAnF5//  B BJGG  )))]N*s: GAG 2B G B G B D.G  Gbasec$|Atjtjt} tj||g}|t |dzdS#t $r|cYSwxYw)ziShorten a source filepath, with the assumption that torch/ subdirectories don't need to be shown to user.N)ospathdirname__file__ commonpathlen ValueError)fnr4prefixs r1shorten_filenamer@s |wrwx8899%##RJ//#f++/##$$  s!B BBF)r4linecrd}|r |jd}|t|j|d|jd|jS)z Format a FrameSummary in a short way, without printing full absolute path or code. The idea is the result fits on a single line. z # r3:z in )rAr@r*linenor)r-r4rA extra_lines r1 format_framerGsX J ) (((  f*5>EEE f f f fZ_Zd f ffcPttj|dS)zJFormat a TracebackType in a short way, printing only the inner-most frame.)rG traceback extract_tbr(s r1format_traceback_shortrNs  ,R004 5 55rHcleZdZddgZddZdZdZdZedddd d Z d Z ed Z d S)CapturedTracebackr(skiprc"||_||_dSNr(rQ)selfr(rQs r1__init__zCapturedTraceback.__init__s rHcd|_dSrSrMrUs r1cleanupzCapturedTraceback.cleanups rHcddl}|jtjSt |jj|jgd|jS)Nr) torch._C._profilerr(rK StackSummary_extract_symbolized_tb_C _profilersymbolize_tracebacksrQ)rUtorchs r1summaryzCapturedTraceback.summarysY!!!! 7?)++ +% H  3 3TWI > >q A49   rHcdd|jdfS)NrT)rQrXs r1 __getstate__zCapturedTraceback.__getstate__s!      rHF)scriptcpprQcddl}|s|r|dks Jdt|jjd|||s|rdn|dzS)a Like traceback.extract_stack(), but faster (approximately 20x faster); it is fast enough that you can unconditionally log stacks this way as part of normal execution. It returns a torch._C._profiler.CapturedTraceback object that must be formatted specially with format_captured_tb. By default, this only reports Python backtraces (like extract_stack). You can set the script/cpp kwargs to also turn on TorchScript/C++ trace reporting. rNzskip with script/cpp NYIT)pythonrerfr6)r[rPr^r_gather_traceback)rerfrQras r1extractzCapturedTraceback.extractsz "!!!  9S 919998999 H  / /tFPS / T T ,3 ,AAD1H    rHcNtj|S)al Formats a single torch._C._profiler.CapturedTraceback into a list of strings equivalent to the output of traceback.format_list. Note that if pass it CapturedTraceback with C++ traces, it is better not to use this function and use the batch formatting API format_captured_tbs to amortize the cost of symbolization )rK format_listrbrXs r1formatzCapturedTraceback.formats$T\\^^444rHcddl}g}g}tD]L\}}|j|g"|d||M|jjfd|D|D]1}tj| ||<2|S)z_ Bulk version of CapturedTraceback.format. Returns a list of list of strings. rNc*g|]}|jSrM).0itbss r1 z0CapturedTraceback.format_all..s0Q0Q0QqQ0Q0Q0QrH) r[ enumerater(r#r^r_r`rKrlrb)rsrars delayed_idxsrrr(s` r1 format_allzCapturedTraceback.format_alls "!!!)+ s^^ ' 'EAru} "  $##A&&&& //0Q0Q0Q0QL0Q0Q0QRRR < |S)z Given a symbolized traceback from symbolize_tracebacks, return a StackSummary object of pre-processed stack trace entries. Nr*rAr)rKr\r%r# FrameSummary)r(rQr)r,s r1r]r]sh  " $ $E bi RR Y+AjM1V9aiPPQQQQ LrH) contextlibros.pathr7rrKtypesrtypingrcontextmanagerr2r@rGrNrPr]rprHr1rsR Q*Q*Q*h"& % % % % %!%5 g g g g g666 UUUUUUUUprH