# tvm-ffi-stubgen(begin): import-section # fmt: off # isort: off from __future__ import annotations from tvm_ffi import init_ffi_api as _FFI_INIT_FUNC from tvm_ffi.libinfo import load_lib_module as _FFI_LOAD_LIB from typing import TYPE_CHECKING if TYPE_CHECKING: from collections.abc import Mapping from typing import Any # isort: on # fmt: on # tvm-ffi-stubgen(end) # tvm-ffi-stubgen(import-object): tvm_ffi.libinfo.load_lib_module;False;_FFI_LOAD_LIB LIB = _FFI_LOAD_LIB("triton_tvm_ffi", "utils") # tvm-ffi-stubgen(begin): global/triton_tvm_ffi.utils # fmt: off _FFI_INIT_FUNC("triton_tvm_ffi.utils", __name__) if TYPE_CHECKING: def get_device_properties(_0: int, /) -> Mapping[str, int]: ... def launch(*args: Any) -> Any: ... def load_binary(_0: str, _1: bytes, _2: int, _3: int, /) -> tuple[int, int, int, int, int]: ... # fmt: on # tvm-ffi-stubgen(end) __all__ = [ # tvm-ffi-stubgen(begin): __all__ "LIB", "get_device_properties", "launch", "load_binary", # tvm-ffi-stubgen(end) ]