+
    –°³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# )Ú_FunctionalRpropc                ó€   <€ V ^8„  d   QhRS[ S[,          RS[RS[S[S[3,          RS[S[S[3,          RS[RS[RS[/# )é   ÚparamsÚlrÚetasÚ
step_sizesÚforeachÚmaximizeÚ_allow_empty_param_list)Úlistr   ÚfloatÚtupleÚbool)ÚformatÚ__classdict__s   "€Úi/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributed/optim/functional_rprop.pyÚ__annotate__Ú_FunctionalRprop.__annotate__   sp   ø€ ÷ Yñ Yá‘V•ðYñ ðYñ ‘E™5�LÕ!ð	Yñ
 ™%¡˜,Õ'ðYñ ðYñ ðYñ "&ñYó    c                óv  € \        ^R7       RV/V n        W0n        W@n        WPn        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
   z%optimizer got an empty parameter listr	   N)r   Údefaultsr   r   r   r   ÚlenÚ
ValueErrorÚparam_groupÚtorchÚjitÚannotateÚdictr   ÚstrÚstate)Úselfr	   r
   r   r   r   r   r   s   &&&&&&&&r   Ú__init__Ú_FunctionalRprop.__init__   s�   € ô 	;ÀaÕHà�"ð
ˆŒð Œ	Ø$ŒØŒØ Œäˆ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   4   s   ø€ ÷ 5ñ 5™d¡6¨D¥=Õ1ñ 5r   c                óž  € V P                   R ,          p. p. p. p. p. pV P                  R,          pV P                  w  ršV P                  w  r¼\	        V4      \	        V4      8w  d3   \        RR\	        V4       R2,           R\	        V4       2,           4      hRp\        W!4       EF3  w  rï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&   \        P                  ! Wø4      VR&   V P                  V,          pVP                  VR,          4       VP                  VR,          4       VP                  VR	,          4       EK6  	  \        P                  ! 4       ;_uu_ 4        \         P"                  ! VVVVVVVV	V
V P$                  V P&                  VR7       RRR4       R#   + '       g   i     R# ; i)r	   r
   zEthe gradients passed in does not equal to the size of the parameters!zParams length: z. zGradients length: FNg        Ústep)Úmemory_formatÚprevÚ	step_size)Ústep_size_minÚstep_size_maxÚetaminusÚetaplusr   r   Úhas_complex)r   r   r   r   r   r   Úzipr    Ú
is_complexÚappendr%   ÚtensorÚ
zeros_likeÚpreserve_formatÚ	full_likeÚno_gradÚFÚrpropr   r   )r&   r*   r	   Úparams_with_gradÚgradsÚprevsr   Ústate_stepsr
   r2   r3   r0   r1   r4   ÚparamÚgradientr%   s   &&               r   r,   Ú_FunctionalRprop.step4   sè  € Ø×!Ñ! (Õ+ˆØÐØˆØˆØˆ
ØˆØ�]‰]˜4Õ ˆØ ŸI™IÑˆØ'+§¡Ñ$ˆäˆv‹;œ#˜i›.Ô(ÜØWØ#¤C¨£K =°Ð3õ4à&¤s¨9£~Ð&6Ð7õ8óð ð ˆÜ" 6×5‰OˆEØÔ#Øœu×/Ò/°Ó6Õ6�Ø ×'Ñ'¨Ô.Ø—‘˜XÔ&à§
¡
Ô*Ø(*�D—J‘J˜uÑ%Ø ŸJ™J uÕ-�EÜ$)§L¢L°Ó$5�E˜&‘MÜ$)×$4Ò$4Ø¬U×-BÑ-Bô%�E˜&‘Mô */¯ª¸Ó)F�E˜+Ñ&àŸ
™
 5Õ)�Ø—‘˜U 6�]Ô+Ø×!Ñ! %¨Õ"4Ô5Ø×"Ñ" 5¨¥=×1ñ%  6ô( �]Š]�_�_Ü�GŠGØ ØØØØØ+Ø+Ø!ØØŸ™ØŸ™Ø'õ÷ �_�_Š_ús   Ç:7H;È;I	)r   r   r   r   r   r%   r   N)g{®Gáz„?)g      à?g333333ó?)g�íµ ÷Æ°>é2   FFF)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r'   r,   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s   ø‡ € ÷Yò Y÷85ö 5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   øð÷ €ß #Ð #Ý õð
 €Ó ÷Ró Rr   