tvm_ffi.register_error

Contents

tvm_ffi.register_error#

tvm_ffi.register_error(name_or_cls=None, cls=None)[source]#

Register an error class so it can be recognized by the ffi error handler.

Parameters:
  • name_or_cls (str or class) – The name of the error class.

  • cls (class) – The class to register.

Returns:

fregister – Register function if f is not specified.

Return type:

function

Examples

@tvm.error.register_error
class MyError(RuntimeError):
    pass

err_inst = tvm.error.create_ffi_error("MyError: xyz")
assert isinstance(err_inst, MyError)