
    [Thq
                         S SK Jr  S SKrS SKJr  SSKJrJr  SSKJ	r	  S r
\R                  S 5       rS	 r " S
 S5      rg)    )partialN)Future   )	functions	rpc_async)UNSET_RPC_TIMEOUTc                 B    [        U R                  5       U5      " U0 UD6$ Ngetattrlocal_valuerref	func_nameargskwargss       X/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/rpc/rref_proxy.py_local_invoker      s"    4##%y14B6BB    c                 B    [        U R                  5       U5      " U0 UD6$ r
   r   r   s       r   _local_invoke_async_executionr      s"    4##%y14B6BBr   c                    ^ ^^^^^^^	^
 UUUUU U4S jm	T R                  TSS9nT[        :w  a  UR                  5         T	" U5      $ [        5       m
UU	U
4S jnU
4S jmUR	                  U5        T
$ )Nc                 X  > U R                  5       n[        n[        U[        R                  R
                  5      =(       d$    [        U[        R                  R
                  5      nU(       d#  [        UT5      n[        US5      (       a  [        nT" T	R                  5       UT	TTT4T
S9$ )N_wrapped_async_rpc_function)r   timeout)valuer   
issubclasstorchjitScriptModule_Cr   hasattrr   owner)rref_fut	rref_type_invoke_funcbypass_typefuncr   r   r   rpc_apir   r   s        r   _rref_type_cont$_invoke_rpc.<locals>._rref_type_cont   s    NN$	$ EII,B,BC 
zuxx,,H
 9i0Dt:;;<JJL	40	
 	
r   F)r   blockingc                    >  T" U 5      R                  T5        g ! [         a  nTR                  U5         S nAg S nAff = fr
   )thenBaseExceptionset_exception)futex_complete_opr*   results     r   _wrap_rref_type_cont)_invoke_rpc.<locals>._wrap_rref_type_cont5   s<    )$)),7  )$$R(()s    
A ;A c                    >  TR                  U R                  5       5        g ! [         a  nTR                  U5         S nAg S nAff = fr
   )
set_resultr   r/   r0   )r1   r2   r4   s     r   r3   !_invoke_rpc.<locals>._complete_op;   s=    )!!#))+.  )$$R(()s   # 
AAA)	_get_typer   waitr   r.   )r   r)   r   r   r   r   r$   r5   r3   r*   r4   s   ``````  @@@r   _invoke_rpcr<      se    
 
( ~~g~>H)x((  	)	) 	*+r   c                   &    \ rS rSr\4S jrS rSrg)	RRefProxyG   c                 (    Xl         X l        X0l        g r
   )r   r)   rpc_timeout)selfr   r)   r   s       r   __init__RRefProxy.__init__H   s    	"r   c                 b    [        [        U R                  U R                  XR                  5      $ r
   )r   r<   r   r)   rA   )rB   r   s     r   __getattr__RRefProxy.__getattr__M   s&    DLL)=M=M
 	
r   )r)   rA   r   N)__name__
__module____qualname____firstlineno__r   rC   rF   __static_attributes__ r   r   r>   r>   G   s    .? #

r   r>   )	functoolsr   r   torch.futuresr    r   r   	constantsr   r   async_executionr   r<   r>   rM   r   r   <module>rS      sF        " (C C C.f	
 	
r   