o Y۷i%@sdZddlmZddlZddlZddlZddlZddlmZddl m Z d&d d Z Gd ddZ e Z d'ddZd(ddZee _ee _  d)d*ddZedeedeed eed!eed"eed#eed$eed%edS)+zError handling.) annotationsN)Any)core backtracestrreturnlist[tuple[str, int, str]]c Cszd}g}|dD]1}t||}|r:z|d}t|d}|d}||||fWq ty9Yq wq |S)zParse the backtrace string into a list of (filename, lineno, func). Parameters ---------- backtrace The backtrace string. Returns ------- result The list of (filename, lineno, func) z!File "(.+?)", line (\d+), in (.+) r)splitrematchstripgroupintappend ValueError)rpatternresultlinerfilenamelinenofuncrC/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/error.py_parse_backtraces   rc@s8eZdZdZdddZdd d ZdddZdddZdS)TracebackManagerz&Helper to manage traceback generation.rNonecCs i|_dS)z/Initialize the traceback manager and its cache.N) _code_cache)selfrrr__init__?s zTracebackManager.__init__rrrrrtypes.CodeTypecCs|||f}||jvr|j|Stjd|dd}t|D]}t|dr&d|_t|dr.d|_qt||d}|j||d}||j|<|S)Nz _getframe()eval)rmode col_offsetrend_col_offset)co_nameco_firstlineno) r astparsewalkhasattrr&r'compilereplace)r!rrrkeytreenode code_objectrrr_get_cached_code_objectCs       z(TracebackManager._get_cached_code_objecttypes.FrameTypecCs$||||}dtji}t|||S)z:Create a frame object from the filename, lineno, and func. _getframe)r4sysr6r$)r!rrrr3contextrrr _create_frameYs  zTracebackManager._create_frametbtypes.TracebackType | Nonetypes.TracebackTypecCs d d d }|||||||S) aAppend a traceback to the given traceback. Parameters ---------- tb The traceback to append to. filename The filename of the traceback lineno The line number of the traceback func The function name of the traceback Returns ------- new_tb The new traceback with the appended frame. r:r;framer5rrrr<cSst|||j|S)N)types TracebackTypef_lasti)r:r=rrrrcreatesz1TracebackManager.append_traceback..createN)r:r;r=r5rrrr<)r9)r!r:rrrrArrrappend_tracebackbs z!TracebackManager.append_tracebackN)rr)rrrrrrrr#)rrrrrrrr5) r:r;rrrrrrrr<)__name__ __module__ __qualname____doc__r"r4r9rBrrrrr<s     rpy_error BaseExceptioncCsB|j}zt|D] \}}}t||||}q||W~~S~~w)z3Append the backtrace to the py_error and return it.) __traceback__r_TRACEBACK_MANAGERrBwith_traceback)rGrr:rrrrrr_with_append_backtraces   rLr:r;c Csbg}|dur*|j}|j}|jj}|jj}|d|d|d|d|j}|dusdt|S)z"Convert the traceback to a string.Nz File "z", line z, in r ) tb_frame tb_linenof_code co_filenamer(rtb_nextjoinreversed)r:linesr=rrfuncnamerrr_traceback_to_backtrace_strs rW name_or_clsstr | type | Nonecls type | Nonercs6ttr }|jdfdd }|dur|S||S)aRegister an error class so it can be recognized by the ffi error handler. Parameters ---------- name_or_cls The name of the error class. cls The class to register. Returns ------- fregister Register function if f is not specified. Examples -------- .. code-block:: python import tvm_ffi # Register a custom Python exception so tvm_ffi.Error maps to it @tvm_ffi.error.register_error class MyError(RuntimeError): pass # Convert a Python exception to an FFI Error and back ffi_err = tvm_ffi.convert(MyError("boom")) py_err = ffi_err.py_error() assert isinstance(py_err, MyError) myclstypercs,ttrn|j}|tj|<|tj|<|S)z0Register the error class name with the FFI core.) isinstancerrCrERROR_NAME_TO_TYPEERROR_TYPE_TO_NAME)r\err_namerXrrregisters  z register_error..registerN)r\r]rr])r^r]rC)rXrZrcrrbrregister_errors &rd RuntimeErrorr TypeErrorAttributeErrorKeyError IndexErrorAssertionError MemoryError)rrrr )rGrHrrrrH)r:r;rr)NN)rXrYrZr[rr)rF __future__rr*rr7r>typingrrMrrrrJrLrW_WITH_APPEND_BACKTRACE_TRACEBACK_TO_BACKTRACE_STRrdrerrfrgrhrirjrkrrrrs4    N  - 6