o iT@sfddlZddlmZddlmZddlmZeedddZdd Zd d Z eed ee d dS)N)_Await)_register_builtin) set_modulez torch.jitcOstjj|g|Ri|S)zpCreate Await object that will call specified functioni with specified args, when it is requested for the result.)torch_C _awaitable)funcargskwargsr M/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/jit/_await.pyr srcC tj|S)zjRequest await the result of execution, if Await is not completed yet, the func will be called immediately.)rr_awaitable_wait)awr r r r rcCr )z-Create completed Await with specified result.)rr_awaitable_nowait)or r r rrrzprim::awaitable_waitzprim::awaitable_nowait) rtorch._jit_internalrtorch.jit._builtinsr torch.utilsrrrrr r r r s