o
    ٰi                     @  sD  d Z ddlmZ ddlmZ ddlmZ er8d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r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dd0d1Z dd2d3Z!dd5d6Z"dd8d9Z#dd;d<Z$dd>d?Z%dd@dAZ&ddCdDZ'ddFdGZ(ddIdJZ)ddLdMZ*ddOdPZ+ddRdSZ,ddUdVZ-ddWdXZ.ddZd[Z/dd]d^Z0dd`daZ1ddcddZ2ddedfZ3ddhdiZ4ddkdlZ5ddmdnZ6ddpdqZ7ddrdsZ8ddudvZ9ddwdxZ:ddydzZ;dd{d|Z<dd}d~Z=dddZ>dddZ?dddZ@g dZAdS )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                C     d S N r   r   r   L/home/ubuntu/.local/lib/python3.10/site-packages/tvm_ffi/testing/_ffi_api.pyTestIntPairSum&       r   c                C  r   r   r   r   r   r   r   add_one'   r   r   argsr   c                  G  r   r   r   r   r   r   r   apply(   r   r   c                  G  r   r   r   r   r   r   r   echo)   r   r   c                   C  r   r   r   r   r   r   r   get_add_one_c_symbol*   r   r   c                   C  r   r   r   r   r   r   r   get_mlir_add_one_c_symbol+   r   r    r	   c                   C  r   r   r   r   r   r   r   make_unregistered_object,   r   r!   c                  G  r   r   r   r   r   r   r   nop-   r   r"   c                C  r   r   r   r   r   r   r   object_use_count.   r   r#   Tensor | Noneboolc                C  r   r   r   r   r   r   r   optional_tensor_view_has_value/   r   r&   Nonec                C  r   r   r   r   r   r   r   run_check_signal0   r   r(   Sequence[int | None]_1Mapping[str, Sequence[int]]_2
str | Nonec                C  r   r   r   )r   r*   r,   r   r   r   schema_arr_map_opt1   r   r.   c                C  r   r   r   r   r   r   r   schema_id_any2   r   r/   Sequence[Any]c                C  r   r   r   r   r   r   r   schema_id_arr3   r   r1   Sequence[int]c                C  r   r   r   r   r   r   r   schema_id_arr_int4   r   r3   Sequence[Object]c                C  r   r   r   r   r   r   r   schema_id_arr_obj5   r   r5   Sequence[str]c                C  r   r   r   r   r   r   r   schema_id_arr_str6   r   r7   c                C  r   r   r   r   r   r   r   schema_id_bool7   r   r8   bytesc                C  r   r   r   r   r   r   r   schema_id_bytes8   r   r:   r   c                C  r   r   r   r   r   r   r   schema_id_device9   r   r;   r
   c                C  r   r   r   r   r   r   r   schema_id_dltensor:   r   r<   r   c                C  r   r   r   r   r   r   r   schema_id_dtype;   r   r=   floatc                C  r   r   r   r   r   r   r   schema_id_float<   r   r?   Callable[..., Any]c                C  r   r   r   r   r   r   r   schema_id_func=   r   rA   0Callable[[int, float, Callable[..., Any]], None]c                C  r   r   r   r   r   r   r   schema_id_func_typed>   r   rC   c                C  r   r   r   r   r   r   r   schema_id_int?   r   rD   Mapping[Any, Any]c                C  r   r   r   r   r   r   r   schema_id_map@   r   rF   Mapping[str, int]c                C  r   r   r   r   r   r   r   schema_id_map_str_intA   r   rH   Mapping[str, Object]c                C  r   r   r   r   r   r   r   schema_id_map_str_objB   r   rJ   Mapping[str, str]c                C  r   r   r   r   r   r   r   schema_id_map_str_strC   r   rL   c                C  r   r   r   r   r   r   r   schema_id_objectD   r   rM   
int | Nonec                C  r   r   r   r   r   r   r   schema_id_opt_intE   r   rO   Object | Nonec                C  r   r   r   r   r   r   r   schema_id_opt_objF   r   rQ   c                C  r   r   r   r   r   r   r   schema_id_opt_strG   r   rR   strc                C  r   r   r   r   r   r   r   schema_id_stringH   r   rT   c                C  r   r   r   r   r   r   r   schema_id_tensorI   r   rU   	int | strc                C  r   r   r   r   r   r   r   schema_id_variant_int_strJ   r   rW   c                   C  r   r   r   r   r   r   r   schema_no_argsK   r   rX   c                   C  r   r   r   r   r   r   r   schema_no_args_no_returnL   r   rY   c                C  r   r   r   r   r   r   r   schema_no_returnM   r   rZ   c                  G  r   r   r   r   r   r   r   schema_packedN   r   r[   c                C  r   r   r   r   r   r   r   schema_tensor_view_inputO   r   r\   int | str | Sequence[int]c                C  r   r   r   r   r   r   r   schema_variant_mixP   r   r^   c                C  r   r   r   )r   r*   r   r   r   test_raise_errorQ   r   r_   ),r   r   r   r   r   r    r!   r"   r#   r&   r(   r.   r/   r1   r3   r5   r7   r8   r:   r;   r<   r=   r?   rA   rC   rD   rF   rH   rJ   rL   rM   rO   rQ   rR   rT   rU   rW   rX   rY   rZ   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   r'   )r   r)   r*   r+   r,   r-   r   r+   )r   r   r   r   )r   r0   r   r0   )r   r2   r   r2   )r   r4   r   r4   )r   r6   r   r6   )r   r%   r   r%   )r   r9   r   r9   )r   r   r   r   )r   r
   r   r
   )r   r   r   r   )r   r>   r   r>   )r   r@   r   r@   )r   rB   r   rB   )r   rE   r   rE   )r   rG   r   rG   )r   rI   r   rI   )r   rK   r   rK   )r   r	   r   r	   )r   rN   r   rN   )r   rP   r   rP   )r   r-   r   r-   )r   rS   r   rS   )r   rV   r   rV   )r   r'   )r   r
   r   r'   )r   r]   r   r]   )r   rS   r*   rS   r   r'   )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#   r&   r(   r.   r/   r1   r3   r5   r7   r8   r:   r;   r<   r=   r?   rA   rC   rD   rF   rH   rJ   rL   rM   rO   rQ   rR   rT   rU   rW   rX   rY   rZ   r[   r\   r^   r_   __all__r   r   r   r   <module>   sp   












































