tvm_ffi.register_global_func

tvm_ffi.register_global_func#

tvm_ffi.register_global_func(func_name, f=None, override=False)[source]#

Register global function.

Parameters:
  • func_name (str or function) – The function name

  • f (function, optional) – The function to be registered.

  • override (boolean optional) – Whether override existing entry.

Returns:

fregister – Register function if f is not specified.

Return type:

function

Examples

import tvm_ffi

# we can use decorator to register a function
@tvm_ffi.register_global_func("mytest.echo")
def echo(x):
    return x
# After registering, we can get the function by its name
f = tvm_ffi.get_global_func("mytest.echo")
assert f(1) == 1

# we can also directly register a function
tvm_ffi.register_global_func("mytest.add_one", lambda x: x + 1)
f = tvm_ffi.get_global_func("mytest.add_one")
assert f(1) == 2

See also

tvm_ffi.get_global_func(), tvm_ffi.remove_global_func()