# 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 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(begin): global/triton_tvm_ffi.utils # fmt: off _FFI_INIT_FUNC("triton_tvm_ffi.utils", __name__) if TYPE_CHECKING: def build_signature_metadata(*args: Any) -> Any: ... def cuOccupancyMaxActiveClusters(*args: Any) -> Any: ... def fill_tma_descriptor(*args: Any) -> Any: ... 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]: ... def set_printf_fifo_size(*args: Any) -> Any: ... # fmt: on # tvm-ffi-stubgen(end) __all__ = [ # tvm-ffi-stubgen(begin): __all__ "build_signature_metadata", "cuOccupancyMaxActiveClusters", "fill_tma_descriptor", "get_device_properties", "launch", "load_binary", "set_printf_fifo_size", # tvm-ffi-stubgen(end) ]