# 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, Sequence from tvm_ffi import Object 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(_0: int, _1: int, _2: int, _3: int, _4: int, _5: tuple[int, int, int], _6: Object, _7: Object, _8: Object, _9: bool, _10: bool, _11: Object, _12: Object, _13: Sequence[Any], /) -> None: ... 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) ]