| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| """FFI API.""" |
| |
| # tvm-ffi-stubgen(begin): import-section |
| # fmt: off |
| # isort: off |
| from __future__ import annotations |
| from .registry import init_ffi_api as _FFI_INIT_FUNC |
| from typing import TYPE_CHECKING |
| if TYPE_CHECKING: |
| from collections.abc import Mapping, MutableMapping, MutableSequence, Sequence |
| from ctypes import c_void_p |
| from tvm_ffi import Device, Module, Object, StructuralKey as _StructuralKey |
| from tvm_ffi.access_path import AccessPath |
| from typing import Any, Callable |
| # isort: on |
| # fmt: on |
| # tvm-ffi-stubgen(end) |
| |
| # tvm-ffi-stubgen(begin): global/ffi@.registry |
| # fmt: off |
| _FFI_INIT_FUNC("ffi", __name__) |
| if TYPE_CHECKING: |
| def Array(*args: Any) -> Any: ... |
| def ArrayContains(_0: Sequence[Any], _1: Any, /) -> bool: ... |
| def ArrayGetItem(_0: Sequence[Any], _1: int, /) -> Any: ... |
| def ArraySize(_0: Sequence[Any], /) -> int: ... |
| def Bytes(_0: bytes, /) -> bytes: ... |
| def ContainerFindFirstNonCPUDevice(_0: Any, /) -> Device: ... |
| def DeepCopy(_0: Any, /) -> Any: ... |
| def Dict(*args: Any) -> Any: ... |
| def DictClear(_0: MutableMapping[Any, Any], /) -> None: ... |
| def DictCount(_0: MutableMapping[Any, Any], _1: Any, /) -> int: ... |
| def DictErase(_0: MutableMapping[Any, Any], _1: Any, /) -> None: ... |
| def DictForwardIterFunctor(_0: MutableMapping[Any, Any], /) -> Callable[..., Any]: ... |
| def DictGetItem(_0: MutableMapping[Any, Any], _1: Any, /) -> Any: ... |
| def DictGetItemOrMissing(_0: MutableMapping[Any, Any], _1: Any, /) -> Any: ... |
| def DictSetItem(_0: MutableMapping[Any, Any], _1: Any, _2: Any, /) -> None: ... |
| def DictSize(_0: MutableMapping[Any, Any], /) -> int: ... |
| def FromJSONGraph(_0: Any, /) -> Any: ... |
| def FromJSONGraphString(_0: str, /) -> Any: ... |
| def FunctionFromExternC(_0: c_void_p, _1: c_void_p, _2: c_void_p, /) -> Callable[..., Any]: ... |
| def FunctionListGlobalNamesFunctor() -> Callable[..., Any]: ... |
| def FunctionRemoveGlobal(_0: str, /) -> bool: ... |
| def GetFirstStructuralMismatch(_0: Any, _1: Any, _2: bool, _3: bool, /) -> tuple[AccessPath, AccessPath] | None: ... |
| def GetGlobalFuncMetadata(_0: str, /) -> str: ... |
| def GetInvalidObject() -> Object: ... |
| def GetKwargsObject() -> Object: ... |
| def GetRegisteredTypeKeys() -> Sequence[str]: ... |
| def List(*args: Any) -> Any: ... |
| def ListAppend(_0: MutableSequence[Any], _1: Any, /) -> None: ... |
| def ListClear(_0: MutableSequence[Any], /) -> None: ... |
| def ListContains(_0: MutableSequence[Any], _1: Any, /) -> bool: ... |
| def ListErase(_0: MutableSequence[Any], _1: int, /) -> None: ... |
| def ListEraseRange(_0: MutableSequence[Any], _1: int, _2: int, /) -> None: ... |
| def ListGetItem(_0: MutableSequence[Any], _1: int, /) -> Any: ... |
| def ListInsert(_0: MutableSequence[Any], _1: int, _2: Any, /) -> None: ... |
| def ListPop(_0: MutableSequence[Any], _1: int, /) -> Any: ... |
| def ListReplaceSlice(_0: MutableSequence[Any], _1: int, _2: int, _3: MutableSequence[Any], /) -> None: ... |
| def ListReverse(_0: MutableSequence[Any], /) -> None: ... |
| def ListSetItem(_0: MutableSequence[Any], _1: int, _2: Any, /) -> None: ... |
| def ListSize(_0: MutableSequence[Any], /) -> int: ... |
| def MakeFieldGetter(_0: int, /) -> int: ... |
| def MakeFieldSetter(_0: int, _1: int, _2: int, /) -> Callable[..., Any]: ... |
| def MakeObjectFromPackedArgs(*args: Any) -> Any: ... |
| def Map(*args: Any) -> Any: ... |
| def MapCount(_0: Mapping[Any, Any], _1: Any, /) -> int: ... |
| def MapForwardIterFunctor(_0: Mapping[Any, Any], /) -> Callable[..., Any]: ... |
| def MapGetItem(_0: Mapping[Any, Any], _1: Any, /) -> Any: ... |
| def MapGetItemOrMissing(_0: Mapping[Any, Any], _1: Any, /) -> Any: ... |
| def MapSize(_0: Mapping[Any, Any], /) -> int: ... |
| def ModuleClearImports(_0: Module, /) -> None: ... |
| def ModuleGetFunction(_0: Module, _1: str, _2: bool, /) -> Callable[..., Any] | None: ... |
| def ModuleGetFunctionDoc(_0: Module, _1: str, _2: bool, /) -> str | None: ... |
| def ModuleGetFunctionMetadata(_0: Module, _1: str, _2: bool, /) -> str | None: ... |
| def ModuleGetKind(_0: Module, /) -> str: ... |
| def ModuleGetPropertyMask(_0: Module, /) -> int: ... |
| def ModuleGetWriteFormats(_0: Module, /) -> Sequence[str]: ... |
| def ModuleGlobalsAdd(_0: Module, /) -> None: ... |
| def ModuleGlobalsRemove(_0: Module, /) -> None: ... |
| def ModuleImplementsFunction(_0: Module, _1: str, _2: bool, /) -> bool: ... |
| def ModuleImportModule(_0: Module, _1: Module, /) -> None: ... |
| def ModuleInspectSource(_0: Module, _1: str, /) -> str: ... |
| def ModuleLoadFromFile(_0: str, /) -> Module: ... |
| def ModuleWriteToFile(_0: Module, _1: str, _2: str, /) -> None: ... |
| def RecursiveEq(_0: Any, _1: Any, /) -> bool: ... |
| def RecursiveGe(_0: Any, _1: Any, /) -> bool: ... |
| def RecursiveGt(_0: Any, _1: Any, /) -> bool: ... |
| def RecursiveHash(_0: Any, /) -> int: ... |
| def RecursiveLe(_0: Any, _1: Any, /) -> bool: ... |
| def RecursiveLt(_0: Any, _1: Any, /) -> bool: ... |
| def ReprPrint(_0: Any, /) -> str: ... |
| def Shape(*args: Any) -> Any: ... |
| def String(_0: str, /) -> str: ... |
| def StructuralEqual(_0: Any, _1: Any, _2: bool, _3: bool, /) -> bool: ... |
| def StructuralHash(_0: Any, _1: bool, _2: bool, /) -> int: ... |
| def StructuralKey(_0: Any, /) -> _StructuralKey: ... |
| def StructuralKeyEqual(_0: Any, _1: Any, /) -> bool: ... |
| def SystemLib(*args: Any) -> Any: ... |
| def ToJSONGraph(_0: Any, _1: Any, /) -> Any: ... |
| def ToJSONGraphString(_0: Any, _1: Any, /) -> str: ... |
| def _PyClassRegisterTypeAttrColumns(_0: int, _1: int, /) -> None: ... |
| def _RegisterFFIInit(_0: int, /) -> None: ... |
| # fmt: on |
| # tvm-ffi-stubgen(end) |
| |
| |
| __all__ = [ |
| # tvm-ffi-stubgen(begin): __all__ |
| "Array", |
| "ArrayContains", |
| "ArrayGetItem", |
| "ArraySize", |
| "Bytes", |
| "ContainerFindFirstNonCPUDevice", |
| "DeepCopy", |
| "Dict", |
| "DictClear", |
| "DictCount", |
| "DictErase", |
| "DictForwardIterFunctor", |
| "DictGetItem", |
| "DictGetItemOrMissing", |
| "DictSetItem", |
| "DictSize", |
| "FromJSONGraph", |
| "FromJSONGraphString", |
| "FunctionFromExternC", |
| "FunctionListGlobalNamesFunctor", |
| "FunctionRemoveGlobal", |
| "GetFirstStructuralMismatch", |
| "GetGlobalFuncMetadata", |
| "GetInvalidObject", |
| "GetKwargsObject", |
| "GetRegisteredTypeKeys", |
| "List", |
| "ListAppend", |
| "ListClear", |
| "ListContains", |
| "ListErase", |
| "ListEraseRange", |
| "ListGetItem", |
| "ListInsert", |
| "ListPop", |
| "ListReplaceSlice", |
| "ListReverse", |
| "ListSetItem", |
| "ListSize", |
| "MakeFieldGetter", |
| "MakeFieldSetter", |
| "MakeObjectFromPackedArgs", |
| "Map", |
| "MapCount", |
| "MapForwardIterFunctor", |
| "MapGetItem", |
| "MapGetItemOrMissing", |
| "MapSize", |
| "ModuleClearImports", |
| "ModuleGetFunction", |
| "ModuleGetFunctionDoc", |
| "ModuleGetFunctionMetadata", |
| "ModuleGetKind", |
| "ModuleGetPropertyMask", |
| "ModuleGetWriteFormats", |
| "ModuleGlobalsAdd", |
| "ModuleGlobalsRemove", |
| "ModuleImplementsFunction", |
| "ModuleImportModule", |
| "ModuleInspectSource", |
| "ModuleLoadFromFile", |
| "ModuleWriteToFile", |
| "RecursiveEq", |
| "RecursiveGe", |
| "RecursiveGt", |
| "RecursiveHash", |
| "RecursiveLe", |
| "RecursiveLt", |
| "ReprPrint", |
| "Shape", |
| "String", |
| "StructuralEqual", |
| "StructuralHash", |
| "StructuralKey", |
| "StructuralKeyEqual", |
| "SystemLib", |
| "ToJSONGraph", |
| "ToJSONGraphString", |
| "_PyClassRegisterTypeAttrColumns", |
| "_RegisterFFIInit", |
| # tvm-ffi-stubgen(end) |
| ] |
