mirror of
https://github.com/sgjzfzzf/triton-tvm-ffi.git
synced 2026-07-01 08:51:56 +08:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
# 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_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: tuple[int, int, int], _7: Object, _8: Object, _9: Object, _10: Object, _11: Object, _12: Sequence[Any], /) -> None: ...
|
|
# fmt: on
|
|
# tvm-ffi-stubgen(end)
|
|
|
|
|
|
__all__ = [
|
|
# tvm-ffi-stubgen(begin): __all__
|
|
"LIB",
|
|
"TVMFFILauncherImpl",
|
|
"string_to_type",
|
|
"type_to_string",
|
|
# tvm-ffi-stubgen(end)
|
|
]
|