o
    Y۷iC                     @  s.  d Z ddlmZ ddlmZ ddlmZ er:ddlm	Z	m
Z
mZmZ ddlmZmZmZ ddlmZ dd	lmZmZ ed
e erdddZdddZdddZdddZdddZdddZdd d!Zdd$d%Zdd&d'Zdd(d)Z dd+d,Z!dd-d.Z"dd/d0Z#dd2d3Z$dd4d5Z%dd6d7Z&dd9d:Z'dd;d<Z(dd=d>Z)ddAdBZ*ddCdDZ+ddFdGZ,ddIdJZ-ddKdLZ.ddNdOZ/ddPdQZ0ddRdSZ1ddTdUZ2ddVdWZ3ddXdYZ4ddZd[Z5dd\d]Z6dd^d_Z7dd`daZ8ddbdcZ9ddddeZ:ddfdgZ;ddhdiZ<ddkdlZ=ddmdnZ>ddodpZ?ddqdrZ@ddsdtZAddvdwZBddydzZCdd|d}ZDdd~dZEdddZFdddZGdddZHdddZIdddZJdddZKdddZLdddZMdddZNdddZOdddZPdddZQdddZRdddZSdddZTdddZdddZUdddZVdddZWdddZXg dZYdS )zFFI API.    )annotations   )init_ffi_api)TYPE_CHECKING)MappingMutableMappingMutableSequenceSequence)ModuleObjectStructuralKey)
AccessPath)AnyCallableffiargsr   returnc                  G     d S N r   r   r   F/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/_ffi_api.pyArray&       r   _0Sequence[Any]_1boolc                C  r   r   r   r   r   r   r   r   ArrayContains'   r   r   intc                C  r   r   r   r   r   r   r   ArrayGetItem(   r   r!   c                C  r   r   r   r   r   r   r   	ArraySize)   r   r#   bytesc                C  r   r   r   r"   r   r   r   Bytes*   r   r%   c                C  r   r   r   r"   r   r   r   DeepCopy+   r   r&   c                  G  r   r   r   r   r   r   r   Dict,   r   r'   MutableMapping[Any, Any]Nonec                C  r   r   r   r"   r   r   r   	DictClear-   r   r*   c                C  r   r   r   r   r   r   r   	DictCount.   r   r+   c                C  r   r   r   r   r   r   r   	DictErase/   r   r,   Callable[..., Any]c                C  r   r   r   r"   r   r   r   DictForwardIterFunctor0   r   r.   c                C  r   r   r   r   r   r   r   DictGetItem1   r   r/   c                C  r   r   r   r   r   r   r   DictGetItemOrMissing2   r   r0   _2c                C  r   r   r   r   r   r1   r   r   r   DictSetItem3   r   r3   c                C  r   r   r   r"   r   r   r   DictSize4   r   r4   c                C  r   r   r   r"   r   r   r   FromJSONGraph5   r   r5   strc                C  r   r   r   r"   r   r   r   FromJSONGraphString6   r   r7   c                   C  r   r   r   r   r   r   r   FunctionListGlobalNamesFunctor7   r   r8   c                C  r   r   r   r"   r   r   r   FunctionRemoveGlobal8   r   r9   _3$tuple[AccessPath, AccessPath] | Nonec                C  r   r   r   r   r   r1   r:   r   r   r   GetFirstStructuralMismatch9   r   r=   c                C  r   r   r   r"   r   r   r   GetGlobalFuncMetadata:   r   r>   r   c                   C  r   r   r   r   r   r   r   GetInvalidObject;   r   r?   Sequence[str]c                   C  r   r   r   r   r   r   r   GetRegisteredTypeKeys<   r   rA   c                  G  r   r   r   r   r   r   r   List=   r   rB   MutableSequence[Any]c                C  r   r   r   r   r   r   r   
ListAppend>   r   rD   c                C  r   r   r   r"   r   r   r   	ListClear?   r   rE   c                C  r   r   r   r   r   r   r   ListContains@   r   rF   c                C  r   r   r   r   r   r   r   	ListEraseA   r   rG   c                C  r   r   r   r2   r   r   r   ListEraseRangeB   r   rH   c                C  r   r   r   r   r   r   r   ListGetItemC   r   rI   c                C  r   r   r   r2   r   r   r   
ListInsertD   r   rJ   c                C  r   r   r   r   r   r   r   ListPopE   r   rK   c                C  r   r   r   r<   r   r   r   ListReplaceSliceF   r   rL   c                C  r   r   r   r"   r   r   r   ListReverseG   r   rM   c                C  r   r   r   r2   r   r   r   ListSetItemH   r   rN   c                C  r   r   r   r"   r   r   r   ListSizeI   r   rO   c                  G  r   r   r   r   r   r   r   MakeObjectFromPackedArgsJ   r   rP   c                  G  r   r   r   r   r   r   r   MapK   r   rQ   Mapping[Any, Any]c                C  r   r   r   r   r   r   r   MapCountL   r   rS   c                C  r   r   r   r"   r   r   r   MapForwardIterFunctorM   r   rT   c                C  r   r   r   r   r   r   r   
MapGetItemN   r   rU   c                C  r   r   r   r   r   r   r   MapGetItemOrMissingO   r   rV   c                C  r   r   r   r"   r   r   r   MapSizeP   r   rW   r
   c                C  r   r   r   r"   r   r   r   ModuleClearImportsQ   r   rX   Callable[..., Any] | Nonec                C  r   r   r   r2   r   r   r   ModuleGetFunctionR   r   rZ   
str | Nonec                C  r   r   r   r2   r   r   r   ModuleGetFunctionDocS   r   r\   c                C  r   r   r   r2   r   r   r   ModuleGetFunctionMetadataT   r   r]   c                C  r   r   r   r"   r   r   r   ModuleGetKindU   r   r^   c                C  r   r   r   r"   r   r   r   ModuleGetPropertyMaskV   r   r_   c                C  r   r   r   r"   r   r   r   ModuleGetWriteFormatsW   r   r`   c                C  r   r   r   r"   r   r   r   ModuleGlobalsAddX   r   ra   c                C  r   r   r   r"   r   r   r   ModuleGlobalsRemoveY   r   rb   c                C  r   r   r   r2   r   r   r   ModuleImplementsFunctionZ   r   rc   c                C  r   r   r   r   r   r   r   ModuleImportModule[   r   rd   c                C  r   r   r   r   r   r   r   ModuleInspectSource\   r   re   c                C  r   r   r   r"   r   r   r   ModuleLoadFromFile]   r   rf   c                C  r   r   r   r2   r   r   r   ModuleWriteToFile^   r   rg   c                C  r   r   r   r"   r   r   r   	ReprPrint_   r   rh   c                  G  r   r   r   r   r   r   r   Shape`   r   ri   c                C  r   r   r   r"   r   r   r   Stringa   r   rj   c                C  r   r   r   r<   r   r   r   StructuralEqualb   r   rk   c                C  r   r   r   r2   r   r   r   StructuralHashc   r   rl   _StructuralKeyc                C  r   r   r   r"   r   r   r   r   d   r   r   c                C  r   r   r   r   r   r   r   StructuralKeyEquale   r   rn   c                  G  r   r   r   r   r   r   r   	SystemLibf   r   ro   c                C  r   r   r   r   r   r   r   ToJSONGraphg   r   rp   c                C  r   r   r   r   r   r   r   ToJSONGraphStringh   r   rq   )Cr   r   r!   r#   r%   r&   r'   r*   r+   r,   r.   r/   r0   r3   r4   r5   r7   r8   r9   r=   r>   r?   rA   rB   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rS   rT   rU   rV   rW   rX   rZ   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   r   rn   ro   rp   rq   N)r   r   r   r   )r   r   r   r   r   r   )r   r   r   r    r   r   )r   r   r   r    )r   r$   r   r$   )r   r   r   r   )r   r(   r   r)   )r   r(   r   r   r   r    )r   r(   r   r   r   r)   )r   r(   r   r-   )r   r(   r   r   r   r   )r   r(   r   r   r1   r   r   r)   )r   r(   r   r    )r   r6   r   r   )r   r-   )r   r6   r   r   )
r   r   r   r   r1   r   r:   r   r   r;   )r   r6   r   r6   )r   r   )r   r@   )r   rC   r   r   r   r)   )r   rC   r   r)   )r   rC   r   r   r   r   )r   rC   r   r    r   r)   )r   rC   r   r    r1   r    r   r)   )r   rC   r   r    r   r   )r   rC   r   r    r1   r   r   r)   )
r   rC   r   r    r1   r    r:   rC   r   r)   )r   rC   r   r    )r   rR   r   r   r   r    )r   rR   r   r-   )r   rR   r   r   r   r   )r   rR   r   r    )r   r
   r   r)   )r   r
   r   r6   r1   r   r   rY   )r   r
   r   r6   r1   r   r   r[   )r   r
   r   r6   )r   r
   r   r    )r   r
   r   r@   )r   r
   r   r6   r1   r   r   r   )r   r
   r   r
   r   r)   )r   r
   r   r6   r   r6   )r   r6   r   r
   )r   r
   r   r6   r1   r6   r   r)   )r   r   r   r6   )
r   r   r   r   r1   r   r:   r   r   r   )r   r   r   r   r1   r   r   r    )r   r   r   rm   )r   r   r   r   r   r   )r   r   r   r   r   r   )r   r   r   r   r   r6   )Z__doc__
__future__r   registryr   _FFI_INIT_FUNCtypingr   collections.abcr   r   r   r	   tvm_ffir
   r   r   rm   tvm_ffi.access_pathr   r   r   __name__r   r   r!   r#   r%   r&   r'   r*   r+   r,   r.   r/   r0   r3   r4   r5   r7   r8   r9   r=   r>   r?   rA   rB   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rS   rT   rU   rV   rW   rX   rZ   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rn   ro   rp   rq   __all__r   r   r   r   <module>   s   



































































