flexloopy.base 源代码

"""Utilities to locate and load the flexloopy shared library."""
from typing import Optional
import tvm_ffi
from ._ffi_api import LIB


# Maintain compatibility with existing code that uses _LIB from base
[文档] _LIB = LIB
[文档] def get_lib() -> tvm_ffi.Module: """Get the loaded flexloopy library module. Returns: The loaded library module. """ return _LIB
__all__ = ["get_lib", "_LIB"]