U ʹjL(@sddlmZddlmZmZddlZddlZddlZddlZddl Z ej ddZ ddddZ dd d d d Zd dZGdddZddZdS)) TracebackType)ListOptionalNc csVz dVWnDtk rP}z"|j}g}|dk r|jjj}|jjd}|dkr|dk rtjdddd}| |W5QRX|j}t d|j d}|j |jj d }t|jd r|j |jj|jjd }t||j|jd tji}td||j|j} || n |||j}q,d} t|D]}| |_|} q$|| W5d}~XYnXdS) NZ__compile_source__zwFz.py)modedeletesuffixz__inspect_currentframe()eval)co_name co_linetable)r co_firstlinenoZ__inspect_currentframe) Exception __traceback__tb_framef_code co_filename f_globalsgettempfileNamedTemporaryFilewritecompilenamereplacer hasattrr r r f_localsinspect currentframertb_lasti tb_linenoappendtb_nextreversedwith_traceback) exctbstackfilenamesourcefframecodeZ fake_frameZfake_tbr!r,U/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/utils/_traceback.pyreport_compile_source_on_error.sR       r.basecCsb|dkrtjtjt}ztj||g}Wntk rH|YSX|t|ddSdS)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)fnr0prefixr,r,r-shorten_filenames r;F)r0linecCs: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 # r/:z in )r<r;r'linenor)r*r0r<Z extra_liner,r,r- format_frames r@cCstt|dS)zT Format a TracebackType in a short way, printing only the inner-most frame. )r@ traceback extract_tbr%r,r,r-format_traceback_shortsrEc@s`eZdZddgZdddZddZdd Zd d Zed d dd ddZ ddZ eddZ dS)CapturedTracebackr%skiprcCs||_||_dSNr%rG)selfr%rGr,r,r-__init__szCapturedTraceback.__init__cCs d|_dSrHrDrJr,r,r-cleanupszCapturedTraceback.cleanupcCs8ddl}|jdkrtSt|jj|jgd|jS)Nr) torch._C._profilerr%rB StackSummary_extract_symbolized_tb_C _profilersymbolize_tracebacksrG)rJtorchr,r,r-summarys zCapturedTraceback.summarycCsdd|jdfS)NrI)rGrLr,r,r- __getstate__szCapturedTraceback.__getstate__F)scriptcpprGcCsJddl}|s|r |dks tdt|jjjd||d|s<|r@dn|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)pythonrWrXr1)rNAssertionErrorrFrQrRZgather_traceback)rWrXrGrTr,r,r-extracts 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 )rB format_listrUrLr,r,r-formatszCapturedTraceback.formatcsddl}g}g}tD]2\}}|jdkr6|gq|d||q|jjfdd|D}t||D]\}}t | ||<qr|S)z_ Bulk version of CapturedTraceback.format. Returns a list of list of strings. rNcsg|]}|jqSr,rD).0itbsr,r- sz0CapturedTraceback.format_all..) rN enumerater%r rQrRrSziprBr\rU)rarTrsZ delayed_idxsr_r%ZstbsZstbr,r`r- format_alls    zCapturedTraceback.format_allN)r) __name__ __module__ __qualname__ __slots__rKrMrUrV staticmethodr[r]rfr,r,r,r-rFs   rFc 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)rBrOr"r FrameSummary)r%rGr&r)r,r,r-rPs"rP)typesrtypingrrrrB contextlibros.pathr2contextmanagerr.r;r@rErFrPr,r,r,r-s ' V V