o
    ÜÜÓhb  ã                   @   sf   d dl Z d dlmZ d dlmZ d dlmZ eedƒ dd„ Zdd	„ Zd
d„ Z	eedƒ ee	dƒ dS )é    N)Ú_Await)Ú_register_builtin)Ú
set_modulez	torch.jitc                 O   s   t jj| g|¢R i |¤ŽS )z
    Creates Await object that will call specified functioni with specified args,
    when it is requested for the result.
    )ÚtorchÚ_CÚ
_awaitable)ÚfuncÚargsÚkwargs© r   úF/var/www/html/ai/venv/lib/python3.10/site-packages/torch/jit/_await.pyr   
   s   r   c                 C   ó   t j | ¡S )zy
    Requests await the result of execution, if Await is not completed yet,
    the func will be called immediately.
    )r   r   Ú_awaitable_wait)Úawr   r   r   r      s   r   c                 C   r   )z8
    Creates completed Await with specified result.
    )r   r   Ú_awaitable_nowait)Úor   r   r   r      s   r   zprim::awaitable_waitzprim::awaitable_nowait)
r   Útorch._jit_internalr   Útorch.jit._builtinsr   Útorch.utilsr   r   r   r   r   r   r   r   Ú<module>   s    

