# tvm-ffi-stubgen(begin): import-section # fmt: off # isort: off from __future__ import annotations from tvm_ffi import Object as _ffi_Object, init_ffi_api as _FFI_INIT_FUNC, register_object as _FFI_REG_OBJ from tvm_ffi.libinfo import load_lib_module as _FFI_LOAD_LIB from typing import TYPE_CHECKING if TYPE_CHECKING: from collections.abc import Sequence from tvm_ffi import Object 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", "triton_tvm_ffi") # tvm-ffi-stubgen(begin): global/triton_tvm_ffi # fmt: off _FFI_INIT_FUNC("triton_tvm_ffi", __name__) if TYPE_CHECKING: def string_to_type(_0: str, /) -> int | None: ... def type_to_ctype(_0: int, /) -> str: ... def type_to_string(_0: int, /) -> str: ... # fmt: on # tvm-ffi-stubgen(end) # tvm-ffi-stubgen(import-object): tvm_ffi.register_object;False;_FFI_REG_OBJ # tvm-ffi-stubgen(import-object): ffi.Object;False;_ffi_Object @_FFI_REG_OBJ("triton_tvm_ffi.TVMFFILauncherImpl") class TVMFFILauncherImpl(_ffi_Object): """FFI binding for `triton_tvm_ffi.TVMFFILauncherImpl`.""" # tvm-ffi-stubgen(begin): object/triton_tvm_ffi.TVMFFILauncherImpl # fmt: off if TYPE_CHECKING: @staticmethod def __c_ffi_init__(_0: Sequence[int], _1: bool, _2: bool, /) -> Object: ... def launch(self, _1: int, _2: int, _3: int, _4: int, _5: int, _6: int, _7: int, _8: int, _9: int, _10: int, _11: Sequence[Any], /) -> None: ... # fmt: on # tvm-ffi-stubgen(end) __all__ = [ # tvm-ffi-stubgen(begin): __all__ "LIB", "TVMFFILauncherImpl", "string_to_type", "type_to_ctype", "type_to_string", # tvm-ffi-stubgen(end) ]