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()