JinjieLiu/triton-tvm-ffi
1
0
Fork 0
mirror of https://github.com/sgjzfzzf/triton-tvm-ffi.git synced 2026-07-01 08:51:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

implement launch

Browse Source 
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
Jinjie Liu
2026-01-29 01:54:16 +08:00
parent b79d880adf
commit fecf48e403
3 changed files with 193 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/triton_tvm_ffi/utils/_ffi_api.py
+3
View File
@@ -7,6 +7,7 @@ from tvm_ffi.libinfo import load_lib_module as _FFI_LOAD_LIB
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)
@@ -17,6 +18,7 @@ LIB = _FFI_LOAD_LIB("triton_tvm_ffi", "utils")
_FFI_INIT_FUNC("triton_tvm_ffi.utils", __name__)
if TYPE_CHECKING:
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]: ...
# fmt: on
# tvm-ffi-stubgen(end)
@@ -25,6 +27,7 @@ __all__ = [
# tvm-ffi-stubgen(begin): __all__
"LIB",
"get_device_properties",
"launch",
"load_binary",
# tvm-ffi-stubgen(end)
]