tvm_tirx.support.rocm.callback_rocm_link

tvm_tirx.support.rocm.callback_rocm_link = ffi.Function

Callable wrapper around a TVM FFI function.

Instances are obtained by converting Python callables with tvm_ffi.convert(), or by looking up globally-registered FFI functions using tvm_ffi.get_global_func().

Examples

@tvm_ffi.register_global_func("my.add")
def add(a, b):
    return a + b

f = tvm_ffi.get_global_func("my.add")
assert isinstance(f, tvm_ffi.Function)
assert f(1, 2) == 3

See also

tvm_ffi.register_global_func()

Register a Python callable as a global FFI function.

tvm_ffi.get_global_func()

Look up a previously registered global FFI function by name.