+
    –°³j  ã                   ód   a € R t0 t ^ RIt^ RIHu Ht ^ RIHt ^ RIHt . t	] ^ k  ! R R4      t
R# )é    N)ÚTensor)Ú2_scripted_functional_optimizer_deprecation_warningc                   óH   a € ] tR t^t o RV 3R lR lltV 3R lR ltRtV tR# )Ú_FunctionalRMSpropc                óf   <€ V ^8„  d   QhRS[ S[,          RS[RS[RS[RS[RS[RS[RS[R	S[R
S[/
# )é   ÚparamsÚlrÚalphaÚepsÚweight_decayÚmomentumÚcenteredÚforeachÚmaximizeÚ_allow_empty_param_list)Úlistr   ÚfloatÚbool)ÚformatÚ__classdict__s   "€Úk/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/optim/functional_rmsprop.pyÚ__annotate__Ú_FunctionalRMSprop.__annotate__   s   ø€ ÷  Yñ  Yá‘V•ð Yñ ð Yñ ð	 Yñ
 ð Yñ ð Yñ ð Yñ ð Yñ ð Yñ ð Yñ "&ñ Yó    c           
     óz  € \        ^R7       RVRVRVRVRV/V n        Wpn        W€n        W�n        \        V4      ^ 8X  d   V
'       g   \        R4      hRV/V n        \        P                  P                  \        \        P                  \        \        \        P                  3,          3,          / 4      V n        R	# )
r   )Ú
stacklevelr
   r   r   r   r   z%optimizer got an empty parameter listr	   N)r   Údefaultsr   r   r   ÚlenÚ
ValueErrorÚparam_groupÚtorchÚjitÚannotateÚdictr   ÚstrÚstate)Úselfr	   r
   r   r   r   r   r   r   r   r   s   &&&&&&&&&&&r   Ú__init__Ú_FunctionalRMSprop.__init__   sž   € ô 	;ÀaÕHà�"Ø�UØ�3Ø˜LØ˜ð
ˆŒð !ŒØŒØ Œäˆv‹;˜!Ô×$;ÜÐDÓEÐEð % fÐ-ˆÔä—Y‘Y×'Ñ'¬¬U¯\©\¼4ÄÄUÇ\Á\Ð@QÕ;RÐ-RÕ(SÐUWÓXˆŽ
r   c                ó>   <€ V ^8„  d   QhRS[ S[R,          ,          /# )r   Ú	gradientsN)r   r   )r   r   s   "€r   r   r   :   s    ø€ ÷ Fñ F™d¡6¨D¥=Õ1ñ Fr   c                ó$  € V P                   R ,          p. p. p. p. p. p. pV P                  R,          p	V P                  R,          p
V P                  R,          pV P                  R,          pV P                  R,          p\        V4      \        V4      8w  d3   \        RR\        V4       R2,           R	\        V4       2,           4      hR
p\	        W!4       EF·  w  ppVf   K  V\
        P                  ! V4      ,          pVP                  V4       VP                  V4       WðP                  9  dÐ   / V P                  V&   V P                  V,          p\
        P                  ! R4      VR&   \
        P                  ! V\
        P                  R7      VR&   V^ 8”  d*   \
        P                  ! V\
        P                  R7      VR&   V P                  '       d*   \
        P                  ! V\
        P                  R7      VR&   V P                  V,          pVP                  VR,          4       V^ 8”  d   VP                  VR,          4       V P                  '       d   VP                  VR,          4       VP                  VR,          4       EKº  	  \
        P                  ! 4       ;_uu_ 4        \        P                  ! VVVVVVV	V
VVVV P                  V P                   V P"                  VR7       RRR4       R#   + '       g   i     R# ; i)r	   r
   r   r   r   r   zEthe gradients passed in does not equal to the size of the parameters!zParams length: z. zGradients length: FNç        Ústep)Úmemory_formatÚ
square_avgÚmomentum_bufferÚgrad_avg)	r
   r   r   r   r   r   r   r   Úhas_complex)r!   r   r   r    Úzipr"   Ú
is_complexÚappendr'   ÚtensorÚ
zeros_likeÚpreserve_formatr   Úno_gradÚFÚrmspropr   r   )r(   r,   r	   Úparams_with_gradÚgradsÚsquare_avgsÚ	grad_avgsÚmomentum_buffer_listÚstate_stepsr
   r   r   r   r   r4   ÚparamÚgradientr'   s   &&                r   r/   Ú_FunctionalRMSprop.step:   s‡  € Ø×!Ñ! (Õ+ˆØÐØˆØˆØˆ	Ø!ÐØˆØ�]‰]˜4Õ ˆØ—‘˜gÕ&ˆØ�m‰m˜EÕ"ˆØ—=‘= Õ,ˆØ—}‘} ^Õ4ˆäˆv‹;œ#˜i›.Ô(ÜØWØ#¤C¨£K =°Ð3õ4à&¤s¨9£~Ð&6Ð7õ8óð ð ˆÜ" 6×5‰OˆE�8ØÔ#Øœu×/Ò/°Ó6Õ6�Ø ×'Ñ'¨Ô.Ø—‘˜XÔ&à§
¡
Ô*Ø(*�D—J‘J˜uÑ%Ø ŸJ™J uÕ-�EÜ$)§L¢L°Ó$5�E˜&‘MÜ*/×*:Ò*:Ø¬U×-BÑ-Bô+�E˜,Ñ'ð   !”|Ü38×3CÒ3CØ!´×1FÑ1Fô4˜Ð/Ñ0ð —}—}�}Ü,1×,<Ò,<Ø!´×1FÑ1Fô-˜˜jÑ)ð Ÿ
™
 5Õ)�Ø×"Ñ" 5¨Õ#6Ô7Ø˜a”<Ø(×/Ñ/°Ð6GÕ0HÔIØ—=—=�=Ø×$Ñ$ U¨:Õ%6Ô7à×"Ñ" 5¨¥=×1ñ;  6ô> �]Š]�_�_Ü�IŠIØ ØØØØ$ØØØØØ)Ø!ØŸ™ØŸ™ØŸ™Ø'õ÷ �_�_Š_ús   Ê0AK>Ë>L	)r   r   r   r   r!   r'   N)	g{®Gáz„?g®Gáz®ï?g:Œ0âŽyE>r.   r.   FFFF)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r)   r/   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s   ø‡ € ÷ Yò  Y÷DFö Fr   r   c                óV   € V ^8„  d   Qh/ ^ \         9   d   \        \        ,          ;R&   # )r   Ú__all__)Ú__conditional_annotations__r   r&   )r   s   "r   r   r      s   € ‡�÷ Ò ŒŒc�Ñ ò r   )rO   r"   Útorch.optim._functionalÚoptimÚ_functionalr<   r   Ú,torch.distributed.optim._deprecation_warningr   rN   r   r   )rO   s   @r   Ú<module>rT      s1   øð÷ €ß #Ð #Ý õð
 €Ó ÷ió ir   