Function RunCodegen

python/tvm/relax/transform/transform.py:702–727

Produce the runtime::Module with an annotated codegen and global symbol. Parameters ---------- target_options: Optional[dict] Pairs of a target name and compilation options entry_functions: Optional[List[str]] The set of entry functions to start from. Returns

(
    target_options: dict | None = None,
    entry_functions: list[str] | None = None,
)

def RunCodegen(
    target_options: dict | None = None,
    entry_functions: list[str] | None = None,
) -> tvm.ir.transform.Pass:
    """Produce the runtime::Module with an annotated codegen and global symbol.

    Parameters
    ----------
    target_options: Optional[dict]
        Pairs of a target name and compilation options
    entry_functions: Optional[List[str]]
        The set of entry functions to start from.

    Returns
    -------
    ret : tvm.transform.Pass
        The registered pass to remove unused functions.
    """
    if entry_functions is None:
        entry_functions = []

    # enable cutlass byoc registries
    # pylint: disable=unused-import,import-outside-toplevel
    from tvm.contrib import cutlass as _cutlass

    return _ffi_api.RunCodegen(target_options, entry_functions)  # type: ignore


def FoldConstant() -> tvm.ir.transform.Pass: