+
    ˜°³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   "Ú\/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/nn/parallel/replicate.pyÚ__annotate__r      s   € ÷ 6ñ 6œfð 6¬°Õ)?ñ 6ó    c                 óH   € ^ RI p\        WP                  P                  4      # ©r   N)Ú	torch.jitÚ
isinstanceÚjitr   ©r   Útorchs   & r   Ú_is_script_moduler       s   € Ûä�fŸi™i×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ñ 5œfð 5¬°Õ)?ñ 5r   c                 óH   € ^ RI p\        WP                  P                  4      # r   )r   r   Ú_Cr   r   s   & r   Ú_is_script_methodr%      s   € Ûä�fŸh™h×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   € Ûà�9‰9×!Ñ!Ó#Ð#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   € Ûà�9‰9×Ñ×$Ñ$Ð$r   c                ó`   € V ^8„  d   QhR\         R\        \         ,          R,          R\        /# )r   r   ÚmemoNr   )r   ÚsetÚbool)r   s   "r   r   r   4   s*   € ÷ ñ ¤ð ¬s´6­{¸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ð ¬h´vÕ.>ñ r   c                 ó<   € V P                  4       p\        V4       V# ©N)ÚmodulesÚnext)r   Úgens   & r   Údescendant_modulesÚ0_replicatable_module.<locals>.descendant_modules6   s   € Ø�n‰nÓˆÜˆ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   é € ð 
Ù<V¨jÔ˜j×)Ð)Ó<Vùs   ‚F)r0   r3   Úaddr    ÚupdateÚallÚchildrenÚ_replicatable_module)r   r2   r=   Úchilds   &&  r   rH   rH   4   sÁ   € õô
 ×ÒÙØ‚|Ü‹uˆð 	‡H�HˆVÔÜ˜× Ò Ø�‰Ñ& vÓ.Ô/ß‹sñ 
Ù<NÈvÔ<Vó
�sŒsð 	
Šsð 	
ˆsñ 
Ù<NÈvÔ<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   € ÷ 
ñ 
Ü”e—l‘lÕ#ð
ä”cœEŸL™LÕ(Õ)ð
ô ð
ô 
Œ$Œ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  r‰V	'       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\        W¨V\        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ÔØˆ	çñ ó
á�ô ˜1œeŸh™h×=Ñ=Ó>Ô>×QÐQÀ1Ç<Á<Ã>ÒQÙð 	ð 
ô
 ×*Ò*¨7Ó<ˆã%ˆNÜ!*¨<Ö!8‘�ß‘:Ü(-×(=Ò(=¸nÕ>OÓ(P�NÓ%ó "9ñ &ð
 ˆà!Ÿš¨Ð:°'Ó:ˆô ˜1œc -Ó0´#°g³,Ô?ó
á?�ô � 1¤s¨7£|Õ#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ä”cœEŸL™LÕ(Õ)ð`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  rmWÖbK	  	  ppp\        V4       UUu/ uF  w  rmWÖbK	  	  ppp\        W±VR7      p\        WÁR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:Ó;±7¨aÔ   DÖ)±7€GÐ;Ü�w“<€Lä�'×$Ñ$Ó&Ó'€FÜ2;¸FÔ2CÔDÑ2C¡J C�U’ZÑ2C€MÑDÜ/°ÀÓH€Lä�7—?‘?Ó$Ó%€GØ%'€JØ)+€NÛˆØ××Ð§VØ×Ñ˜cÖ"à×!Ñ! #Ö&ñ	 ô 3<¸JÔ2GÔHÑ2G¡h c˜šÑ2GÐÑHÜ6?ÀÔ6OÔPÑ6O©(¨#˜SšXÑ6OÐÑPä3°JÐPVÔWÐÜ7Ø¨ôÐô �7—?‘?Ó$Ó%€GÙ5<Ó(=±W°«±W€MÐ(=Ø(*€Nä˜wÖ'‰	ˆˆ6Ø!"ˆ�vÑÜ�|Ö$ˆAØ×9Ñ9Ó;ˆGô *5«ˆGÔ&à˜!Õ×#Ñ# GÖ,ó %ñ (ô ˜w×'‰	ˆˆ6Ø Ÿ/™/×/Ñ/Ö1‰JˆC�ØŠ}Ü˜|Ö,�AØ+¨AÕ.¨qÕ1�GØ,0�G×$Ñ$ SÓ)ó -ð ,¨EÕ2�
Ü˜|Ö,�AØ+¨AÕ.¨qÕ1�GÜ˜G S¨-¸Õ*:¸:Õ*FÖGó -ñ 2ð !×,Ñ,×2Ñ2Ö4‰JˆC�ØŠ}Ü˜|Ö,�AØ+¨AÕ.¨qÕ1�GØ/3�G×'Ñ'¨Ó,ó -ð *Õ0�	Ü˜|Ö,�AØ+¨AÕ.¨qÕ1�GØ!-¨a¥°Õ!;�Jô ˜G S¨*Ô5à6@�G×.Ñ.¨sÓ3ó -ñ 5ð Ÿ™×-Ñ-Ö/‰HˆC�ØŠ{Ü˜|Ö,�AØ+¨AÕ.¨qÕ1�GØ,0�G×$Ñ$ SÓ)ó -ð ×$×$Ð$¯VØ$4�MØ!2Õ!7‘Jà$8�MØ!6Õ!;�JÜ˜|Ö,�AØ+¨AÕ.¨qÕ1�GÜ˜G S¨-¸Õ*:¸:Õ*FÖGó -ô 0ñ5 (ôT 38¸Ô2EÓFÑ2E¨QŒD”�M !Õ$ QÕ'Ö(Ñ2EÑFÐFùòg <ùó 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õ$õ%÷÷>
ñ@ ˆC�vÔ€÷`Gñ `Gr   