
    {i3	                         S SK 7  S SKrS SKJs  Jr  S SKJr  S SKJr  S SKJr  S SK J	r	  Sr
S rS r\\/rS	\\
   S
\S\4S jrg)    )*N)Tensor)Future)RRef)Tuplec                     U R                   U R                  U R                  /UQ7n0 UEn[        R                  " U R                   R                  5       [        UU5      $ N)module_rrefdeviceis_device_map_setrpc	rpc_asyncowner_remote_forward)selfargskwargss      9/mnt/rpi/tmp/tmp01f2ivx4/_remote_module_non_scriptable.pyforward_asyncr      sX    dkk4+A+AIDIDZF== 	     c                     U R                   U R                  U R                  /UQ7n0 UEn[        R                  " U R                   R                  5       [        UU5      nUR                  5       $ r	   )r
   r   r   r   r   r   r   wait)r   r   r   ret_futs       r   forwardr      sc    dkk4+A+AIDIDZFmm 	G <<>r   r
   r   r   c                 f   U R                  5       n[        R                  " U5      nUR                  S:w  a  UR                  " U0 UD6$ / UQ7nSnU H1  n[        U[        5      (       a  UR                  U5      4OU4nXg-   nM3     0 UEnUR                  5        H1  u  p[        U	[        5      (       d  M  XH   R                  U5      XH'   M3     U(       a  UR                  " U0 UD6$ Sn
UR                  " U0 UD6 H0  n[        U[        5      (       a  UR                  5       4OU4nX-   n
M2     U
$ )Ncuda )
local_valuetorchr   typer   
isinstancer   toitemscpu)r
   r   r   r   r   moduleout_argsargkvretis               r   r   r   -   s   $$&F\\&!F{{f~~t.v.. T8DH#-c6#:#:svvf~>  ZFa  	V,FI  ~~x2622 C^^X00$Q//QUUWJaTg 1 Jr   )typingr   torch.distributed.rpcdistributedr   r   torch._jit_internalr   r   r   module_interface_clsr   r   _generated_methodsstrboolr   r   r   r   <module>r4      sb      # #  & &   	  $*+$58$MQ$r   