support typedvalue

Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>

python/triton_tvm_ffi/_ffi_api.py

@@ -0,0 +1,54 @@
+# 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 Mapping
+    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 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]: ...
+    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.TypedValue")
+class TypedValue(_ffi_Object):
+    # tvm-ffi-stubgen(begin): object/triton_tvm_ffi.TypedValue
+    # fmt: off
+    if TYPE_CHECKING:
+        @staticmethod
+        def __c_ffi_init__(_0: int, _1: Any, /) -> Object: ...
+    # fmt: on
+    # tvm-ffi-stubgen(end)
+
+
+__all__ = [
+    # tvm-ffi-stubgen(begin): __all__
+    "LIB",
+    "TypedValue",
+    "get_device_properties",
+    "launch",
+    "load_binary",
+    "string_to_type",
+    "type_to_string",
+    # tvm-ffi-stubgen(end)
+]