+
    &j[                        ^ RI Ht ^ RIHtHt ^ RIHtHtHt ^ RI	H
t
 ^ RIt^ RIHt ^ RIHt ^ RIHt ]'       d   ^ R	IHt ^ R
IHt ^ RIHt R.tR R ltR R ltR R ltR R ltRR R lltRR R llt]! R]R7      tRR R llt R# )    )OrderedDict)IteratorSequence)castTYPE_CHECKINGTypeVar)TypeIsN)_get_device_index)Module)comm)ScriptMethod)ScriptModule)EnabledProxy	replicatec                >    V ^8  d   QhR\         R\        R,          /# )   modulereturnr   )r   r	   )formats   "s/Users/jameslopez/projects/CWCArchive/cwc-podcast/.venv/lib/python3.14/site-packages/torch/nn/parallel/replicate.py__annotate__r      s     6 6f 6)? 6    c                 H    ^ RI p\        WP                  P                  4      # r   N)	torch.jit
isinstancejitr   r   torchs   & r   _is_script_moduler       s    fii4455r   c                >    V ^8  d   QhR\         R\        R,          /# )r   r   r   r   )objectr	   )r   s   "r   r   r      s     5 5f 5)? 5r   c                 H    ^ RI p\        WP                  P                  4      # r   )r   r   _Cr   r   s   & r   _is_script_methodr%      s    fhh3344r   c                    V ^8  d   QhRR/# )r   r   r    )r   s   "r   r   r   !   s     $ $^ $r   c                  >    ^ RI p V P                  P                  4       # r   )r   r   r   r   s    r   _init_script_moduler*   !   s    99!!##r   c                    V ^8  d   QhRR/# )r   r   r   r'   )r   s   "r   r   r   '   s     % % %r   c                  J    ^ RI p V P                  P                  P                  # r   )torch.jit._stater   _state_enabledr)   s    r   _is_jit_enabledr0   '   s    99$$$r   c                `    V ^8  d   QhR\         R\        \         ,          R,          R\        /# )r   r   memoNr   )r   setbool)r   s   "r   r   r   4   s*       s6{T/A T r   c                    R  R lp\        4       '       g   R# Vf   \        4       pVP                  V 4       \        V 4      '       dX   VP	                  V! V 4      4       \
        ;QJ d"    R V! V 4       4       F  '       d   K   R# 	  R# ! R V! V 4       4       4      # V P                  4        F  pW19   d   K  \        W14      '       d   K   R# 	  R# )c                F    V ^8  d   QhR\         R\        \         ,          /# )r   r   r   )r   r   )r   s   "r   r   *_replicatable_module.<locals>.__annotate__6   s      6 hv.> r   c                 <    V P                  4       p\        V4       V# N)modulesnext)r   gens   & r   descendant_modules0_replicatable_module.<locals>.descendant_modules6   s    nnS	
r   Tc              3   8   "   T F  p\        V4      x  K  	  R # 5ir9   )r    ).0
descendants   & r   	<genexpr>'_replicatable_module.<locals>.<genexpr>D   s      
<Vjj))<Vs   F)r0   r3   addr    updateallchildren_replicatable_module)r   r2   r=   childs   &&  r   rH   rH   4   s    
 |u 	HHV  &v./s 
<Nv<V
ss 	
s 	
s 
<Nv<V
 
 	
 " =#E00 # r   c          
          V ^8  d   QhR\         \        P                  ,          R\         \        \        P                  ,          ,          R\
        R\        \        \        P                  ,          ,          /# )r   tensorsdevicesdetachr   )r   r   Tensorintdevicer4   list)r   s   "r   r   r   S   sU     
 
ell#
cELL()
 
 
$u||
	
r   c                 l   ^ RI Hp \        V 4      ^ 8X  d   . # V'       d   V  Uu. uFD  p\        V\        P
                  P                  4      '       * ;'       d    VP                  4       NKF  	  pp\        P                  ! W4      pV F?  p\        V4       F-  w  rV	'       g   K  \        P                  ! Wx,          4      Wx&   K/  	  KA  	  V# VP                  ! V.V O5!  p
\        ^ \        V
4      \        V 4      4       Uu. uF   p\        WV\        V 4      ,            4      NK"  	  up# u upi u upi )r   )	Broadcast)torch.nn.parallel._functionsrS   lenr   r   nnUninitializedParameter
is_complexr   broadcast_coalesced	enumerateview_as_complexapplyrangerQ   )rK   rL   rM   rS   tcomplex_maskoutputsdevice_outputsirX   tensor_copiess   &&&        r   _broadcast_coalesced_reshaperd   S   s   
 7
7|q	 
 1ehh==>>QQ1<<>Q 	 

 **7<%N!*<!8:(-(=(=n>O(PN% "9 &
 !:': 1c-0#g,?
? 1s7|#345?
 	


s   4D,D,&D1T)boundc          	          V ^8  d   QhR\         R\        \        \        P                  ,          ,          R\
        R\        \         ,          /# )r   networkrL   rM   r   )re   r   rO   r   rP   r4   rQ   )r   s   "r   r   r   v   sN     `G `G`GcELL()`G `G 
!W	`Gr   c           	         \        V 4      '       g   \        R 4      hV'       g   . # V Uu. uF  p\        VR4      NK  	  pp\        V4      p\	        V P                  4       4      p\        V4       UUu/ uF  w  rgWvbK	  	  ppp\        WQV4      p	\	        V P                  4       4      p
. p. pV
 FA  pVP                  '       d   V'       g   VP                  V4       K0  VP                  V4       KC  	  \        V4       UUu/ uF  w  rmWbK	  	  ppp\        V4       UUu/ uF  w  rmWbK	  	  ppp\        WVR7      p\        WRR7      p\	        V P                  4       4      pV Uu. uF  p. NK  	  pp/ p\        V4       FT  w  ppVVV&   \        V4       F:  pVP                  4       p\        4       Vn        VV,          P                  V4       K<  	  KV  	  \        V4       EF   w  ppVP                   P#                  4        F  w  ppVf4   \        V4       F"  pVV,          V,          pRVP                   V&   K$  	  K=  VV,          p\        V4       F.  pVV,          V,          p\%        VVVV,          V,          4       K0  	  K  	  VP&                  P#                  4        F  w  ppVf4   \        V4       F"  pVV,          V,          pRVP&                  V&   K$  	  K=  W,          p\        V4       F?  pVV,          V,          pV	V,          V,          p\%        VVV4       VVP                  V&   KA  	  K  	  VP(                  P#                  4        F  w  ppVf4   \        V4       F"  pVV,          V,          pRVP(                  V&   K$  	  K=  VP                  '       d   V'       g   TpW,          p M
TpW,          p \        V4       F.  pVV,          V,          p\%        VVVV,          V ,          4       K0  	  K  	  EK#  	  \        V4       Uu. uF!  p\+        \,        VV,          ^ ,          4      NK#  	  up# u upi u uppi u uppi u uppi u upi u upi )zJCannot replicate network where python modules are children of ScriptModuleT)rM   N)rH   RuntimeErrorr
   rU   rQ   
parametersrZ   rd   buffersrequires_gradappendr:   r]   _replicate_for_data_parallelr   _former_parameters_modulesitemssetattr_parameters_buffersr   re   )!rh   rL   rM   xnum_replicasparamsidxparamparam_indicesparam_copiesrl   
buffers_rgbuffers_not_rgbufbuffer_indices_rgbuffer_indices_not_rgbuffer_copies_rgbuffer_copies_not_rgr:   _module_copiesmodule_indicesrb   r   jreplicakeyrI   
module_idx	param_idx
param_copybuffer_copies
buffer_idxs!   &&&                              r   r   r   v   s   
  ((X
 	
 	3:;7a D)7G;w<L'$$&'F2;F2CD2CJCUZ2CMD/HL7??$%G%'J)+NVc"!!#&	  3<J2GH2Ghc2GH6?6OP6O(#SX6OP3JPVW7 7??$%G5<(=WWM(=(*Nw'	6!"v|$A99;G *5G&!##G, % ( w'	6 ////1JC}|,A+A.q1G,0G$$S) - ,E2
|,A+A.q1GGS-*::*FG - 2 !,,224JC}|,A+A.q1G/3G'', - *0	|,A+A.q1G!-a!;J GS*56@G..s3 - 5 --/HC{|,A+A.q1G,0G$$S) - $$$V$4M!2!7J$8M!6!;J|,A+A.q1GGS-*::*FG - 05 (T 382EF2EQDM!$Q'(2EFFg < E IP )>t Gs#   Q 4QQ Q'Q'Qr9   )F)!collectionsr   collections.abcr   r   typingr   r   r   typing_extensionsr	   r   torch._utilsr
   torch.nn.modulesr   torch.nn.parallelr   torch._Cr   r   r   r-   r   __all__r    r%   r*   r0   rH   rd   re   r   r'   r   r   <module>r      sr    # . / / $  * # " %&- -65$%>
@ Cv`G `Gr   