enable cjit launcher

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
2026-07-01 08:51:56 +08:00 · 2026-01-31 10:36:41 +08:00
parent f0739b9dca
commit ac7497b2c8
7 changed files with 180 additions and 38 deletions
@@ -20,6 +20,7 @@ LIB = _FFI_LOAD_LIB("triton_tvm_ffi", "triton_tvm_ffi")
 _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)
@@ -46,6 +47,7 @@ __all__ = [
    "LIB",
    "TVMFFILauncherImpl",
    "string_to_type",
+    "type_to_ctype",
    "type_to_string",
    # tvm-ffi-stubgen(end)
 ]