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

remove CudaUtils dependencies

Browse Source 
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
This commit is contained in:
Jinjie Liu
2026-01-28 14:30:43 +08:00
parent 15b4b23ec8
commit b79d880adf
3 changed files with 65 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/triton_tvm_ffi/utils/_ffi_api.py
+2
View File
@@ -17,6 +17,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 load_binary(_0: str, _1: bytes, _2: int, _3: int, /) -> tuple[int, int, int, int, int]: ...
# fmt: on
# tvm-ffi-stubgen(end)
@@ -24,5 +25,6 @@ __all__ = [
# tvm-ffi-stubgen(begin): __all__
"LIB",
"get_device_properties",
"load_binary",
# tvm-ffi-stubgen(end)
]