
    `i                       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
 ddlmZmZmZmZ ddlmZ dd	lmZmZ  ed
e           erd\dZd]dZd^dZd^dZd_dZd_dZd`dZd^dZdadZdbdZdcdZddd%Z ded&Z!dfd(Z"dgd*Z#dhd,Z$did.Z%djd/Z&dkd1Z'dld3Z(dmd5Z)dnd7Z*dod9Z+dpd;Z,dqd=Z-d]d>Z.drd@Z/dsdBZ0dtdDZ1dudFZ2dvdGZ3dwdIZ4dxdKZ5dydLZ6dzdNZ7dmdOZ8d{dQZ9d_dRZ:d|dSZ;dcdTZ<d^dUZ=d}dVZ>d~dXZ?ddYZ@g dZZAd[S )z FFI API for namespace `testing`.    )annotations   )init_ffi_api)TYPE_CHECKING)MappingSequence)DeviceObjectTensordtype)TestIntPair)AnyCallabletesting_0r   returnintc                   d S N r   s    l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/testing/_ffi_api.pyTestIntPairSumr   &             c                   d S r   r   r   s    r   add_oner   '   r   r   argsr   c                     d S r   r   r   s    r   applyr!   (   r   r   c                     d S r   r   r    s    r   echor#   )   r   r   c                     d S r   r   r   r   r   get_add_one_c_symbolr%   *   r   r   c                     d S r   r   r   r   r   get_mlir_add_one_c_symbolr'   +   r   r   r
   c                     d S r   r   r   r   r   make_unregistered_objectr)   ,   r   r   c                     d S r   r   r    s    r   nopr+   -   r   r   c                   d S r   r   r   s    r   object_use_countr-   .   r   r   Tensor | Noneboolc                   d S r   r   r   s    r   optional_tensor_view_has_valuer1   /   r   r   Nonec                   d S r   r   r   s    r   run_check_signalr4   0   r   r   Sequence[int | None]_1Mapping[str, Sequence[int]]_2
str | Nonec                   d S r   r   )r   r6   r8   s      r   schema_arr_map_optr;   1   r   r   c                   d S r   r   r   s    r   schema_id_anyr=   2   r   r   Sequence[Any]c                   d S r   r   r   s    r   schema_id_arrr@   3   r   r   Sequence[int]c                   d S r   r   r   s    r   schema_id_arr_intrC   4   r   r   Sequence[Object]c                   d S r   r   r   s    r   schema_id_arr_objrF   5   r   r   Sequence[str]c                   d S r   r   r   s    r   schema_id_arr_strrI   6   r   r   c                   d S r   r   r   s    r   schema_id_boolrK   7   r   r   bytesc                   d S r   r   r   s    r   schema_id_bytesrN   8   r   r   r	   c                   d S r   r   r   s    r   schema_id_devicerP   9   r   r   r   c                   d S r   r   r   s    r   schema_id_dltensorrR   :   r   r   r   c                   d S r   r   r   s    r   schema_id_dtyperT   ;   r   r   floatc                   d S r   r   r   s    r   schema_id_floatrW   <   r   r   Callable[..., Any]c                   d S r   r   r   s    r   schema_id_funcrZ   =   r   r   0Callable[[int, float, Callable[..., Any]], None]c                   d S r   r   r   s    r   schema_id_func_typedr]   >   r   r   c                   d S r   r   r   s    r   schema_id_intr_   ?   r   r   Mapping[Any, Any]c                   d S r   r   r   s    r   schema_id_maprb   @   r   r   Mapping[str, int]c                   d S r   r   r   s    r   schema_id_map_str_intre   A   r   r   Mapping[str, Object]c                   d S r   r   r   s    r   schema_id_map_str_objrh   B   r   r   Mapping[str, str]c                   d S r   r   r   s    r   schema_id_map_str_strrk   C   r   r   c                   d S r   r   r   s    r   schema_id_objectrm   D   r   r   
int | Nonec                   d S r   r   r   s    r   schema_id_opt_intrp   E   r   r   Object | Nonec                   d S r   r   r   s    r   schema_id_opt_objrs   F   r   r   c                   d S r   r   r   s    r   schema_id_opt_strru   G   r   r   strc                   d S r   r   r   s    r   schema_id_stringrx   H   r   r   c                   d S r   r   r   s    r   schema_id_tensorrz   I   r   r   	int | strc                   d S r   r   r   s    r   schema_id_variant_int_strr}   J   r   r   c                     d S r   r   r   r   r   schema_no_argsr   K   r   r   c                     d S r   r   r   r   r   schema_no_args_no_returnr   L   r   r   c                   d S r   r   r   s    r   schema_no_returnr   M   r   r   c                     d S r   r   r    s    r   schema_packedr   N   r   r   c                   d S r   r   r   s    r   schema_tensor_view_inputr   O   r   r   int | str | Sequence[int]c                   d S r   r   r   s    r   schema_variant_mixr   P   r   r   c                   d S r   r   )r   r6   s     r   test_raise_errorr   Q   r   r   ),r   r   r!   r#   r%   r'   r)   r+   r-   r1   r4   r;   r=   r@   rC   rF   rI   rK   rN   rP   rR   rT   rW   rZ   r]   r_   rb   re   rh   rk   rm   rp   rs   ru   rx   rz   r}   r   r   r   r   r   r   r   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   r2   )r   r5   r6   r7   r8   r9   r   r7   )r   r   r   r   )r   r>   r   r>   )r   rA   r   rA   )r   rD   r   rD   )r   rG   r   rG   )r   r/   r   r/   )r   rL   r   rL   )r   r	   r   r	   )r   r   r   r   )r   r   r   r   )r   rU   r   rU   )r   rX   r   rX   )r   r[   r   r[   )r   r`   r   r`   )r   rc   r   rc   )r   rf   r   rf   )r   ri   r   ri   )r   r
   r   r
   )r   rn   r   rn   )r   rq   r   rq   )r   r9   r   r9   )r   rv   r   rv   )r   r{   r   r{   )r   r2   )r   r   r   r2   )r   r   r   r   )r   rv   r6   rv   r   r2   )B__doc__
__future__r   registryr   _FFI_INIT_FUNCtypingr   collections.abcr   r   tvm_ffir	   r
   r   r   tvm_ffi.testingr   r   r   __name__r   r   r!   r#   r%   r'   r)   r+   r-   r1   r4   r;   r=   r@   rC   rF   rI   rK   rN   rP   rR   rT   rW   rZ   r]   r_   rb   re   rh   rk   rm   rp   rs   ru   rx   rz   r}   r   r   r   r   r   r   r   __all__r   r   r   <module>r      s  " ' &
 # " " " " " 5 5 5 5 5 5             %11111111555555555555++++++$$$$$$$$ y( # # # ,;6666''''%%%%$$$$****////1111####3333IIII1111 M  M  M  M----AAAAEEEEKKKKEEEE000033336666888833333333LLLL O  O  O  O----IIIIQQQQWWWWQQQQ6666????EEEE????00006666EEEE$$$$////1111----<<<<^^^^::::/ / /r   