mirror of
https://github.com/sgjzfzzf/triton-tvm-ffi.git
synced 2026-07-01 08:51:56 +08:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
# 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)
|
|
]
|