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 | type | None, default: None) – The name of the error class.

  • cls (type | None, default: None) – The class to register.

Return type:

Any

Returns:

fregister – Register function if f is not specified.

Examples

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

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