tvm_ffi.load_module

Contents

tvm_ffi.load_module#

tvm_ffi.load_module(path)[source]#

Load module from file.

Parameters:

path (str | PathLike) – The path to the module file.

Return type:

Module

Returns:

The loaded module

Examples

import tvm_ffi
from pathlib import Path

# Works with string paths
mod = tvm_ffi.load_module("path/to/module.so")
# Also works with pathlib.Path objects
mod = tvm_ffi.load_module(Path("path/to/module.so"))

mod.func_name(*args)

See also

tvm_ffi.Module

Contents