codegen.py 453 Bytes
Newer Older
1
"""Code generation related functions."""
2
from ._ffi.function import _init_api
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

def build_module(lowered_func, target):
    """Build lowered_func into Module.

    Parameters
    ----------
    lowered_func : LoweredFunc
        The lowered function

    target : str
        The target module type.

    Returns
    -------
    module : Module
        The corressponding module.
    """
    return _Build(lowered_func, target)

_init_api("tvm.codegen")