Tvm how to get func from static lib?

1

Hi, I can get function from dynamic exported lib, but I want get from static, how to do that?

def test_add(lib_path):
    if lib_path.endswith('.a'):
        mod = tvm.runtime.registry.get('runtime.SystemLib')
    else:
        mod = tvm.runtime.load_module(lib_path)

    fadd = mod.get_function('vector_add')

    dev = tvm.cpu()
    dtype = np.float16

this seems not work for static lib.

what’s the python API same as c++:

tvm::runtime::Module mod_tvmlib =
      (*tvm::runtime::Registry::Get("runtime.SystemLib"))();

?

1 Like