o hL(@sddlmZddlmZmZddlZddlZddlZddlZddl Z ej ddZ ddddZ dd d d d Zd dZGdddZddZdS)) TracebackType)ListOptionalNc cs\zdVWdSty}z|j}g}|dur|jjj}|jjd}|dkr|durtjdddd }| |Wdn1sDwY|j}t d|j d}|j |jj d }t|jd rl|j |jj|jjd }t||ji|jd tji}td||j|j} || n|||j}|dusd} t|D]}| |_|} q|| d}~ww) 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 r r f_localsinspect currentframertb_lasti tb_linenoappendtb_nextreversedwith_traceback) exctbstackfilenamesourcefframecode fake_framefake_tbr#r0L/var/www/html/ai/venv/lib/python3.10/site-packages/torch/utils/_traceback.pyreport_compile_source_on_error.sX        F  r2basecCs\|durtjtjt}z tj||g}Wn ty#|YSw|t|ddS)z Shorten a source filepath, under the assumption that anything under torch/ directory is "obvious" and doesn't need to be shown to user. N)ospathdirname__file__ commonpath ValueErrorlen)fnr4prefixr0r0r1shorten_filenames r?F)r4linecCs:d}|r |jd}|t|j|dd|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 )r@r?r)linenor)r,r4r@ extra_liner0r0r1 format_frames &rEcCstt|dS)zT Format a TracebackType in a short way, printing only the inner-most frame. )rE traceback extract_tbr'r0r0r1format_traceback_shortsrJc@s`eZdZddgZdddZddZdd Zd d Zed d dd ddZ ddZ eddZ dS)CapturedTracebackr'skiprcCs||_||_dSNr'rL)selfr'rLr0r0r1__init__s zCapturedTraceback.__init__cCs d|_dSrMrIrOr0r0r1cleanups zCapturedTraceback.cleanupcCs8ddl}|jdur tSt|jj|jgd|jS)Nr) torch._C._profilerr'rG StackSummary_extract_symbolized_tb_C _profilersymbolize_tracebacksrL)rOtorchr0r0r1summarys zCapturedTraceback.summarycCsdd|jdfS)NrN)rLrQr0r0r1 __getstate__szCapturedTraceback.__getstate__F)scriptcpprLcCsLddl}|s|r|dksJdt|jjjd||d|s|r!dS|dS)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)pythonr\r]r5)rSrKrVrWgather_traceback)r\r]rLrYr0r0r1extracts  zCapturedTraceback.extractcCst|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 )rG format_listrZrQr0r0r1formatszCapturedTraceback.formatcsddl}g}g}tD]\}}|jdur|gq |d||q |jjfdd|D}t||D]\}}t | ||<q9|S)z_ Bulk version of CapturedTraceback.format. Returns a list of list of strings. rNcsg|]}|jqSr0rI).0itbsr0r1 sz0CapturedTraceback.format_all..) rS enumerater'r"rVrWrXziprGrarZ)rfrYrs delayed_idxsrdr'stbsstbr0rer1 format_alls    zCapturedTraceback.format_allN)r) __name__ __module__ __qualname__ __slots__rPrRrZr[ staticmethodr`rbrnr0r0r0r1rKs   rKc CsBt}t||dD]}|t|d|d|dq |S)z Given a symbolized traceback from symbolize_tracebacks, return a StackSummary object of pre-processed stack trace entries. Nr)r@r)rGrTr$r" FrameSummary)r'rLr(r+r0r0r1rUs"rU)typesrtypingrrrrG contextlibros.pathr6contextmanagerr2r?rErJrKrUr0r0r0r1s ' V  V