+
    –°³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# )Ú_FunctionalAdagradc                óx   <€ 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[RS[RS[RS[/# )é   ÚparamsÚlrÚlr_decayÚweight_decayÚinitial_accumulator_valueÚwarmup_lr_multiplierÚwarmup_num_itersÚepsÚcoalesce_gradÚforeachÚfusedÚ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_adagrad.pyÚ__annotate__Ú_FunctionalAdagrad.__annotate__   s‘   ø€ ÷ -ñ -á‘V•ð-ñ ð-ñ ð	-ñ
 ð-ñ $)ð-ñ $ð-ñ  ð-ñ ð-ñ ð-ñ ð-ñ ð-ñ ð-ñ "&ñ-ó    c                óN  € \        ^R7       RVRVRVRVRVRVRV/V n        W�n        W n        W°n        WÀn        \        P                  P                  \        \        P                  \        \        \        P                  3,          3,          / 4      V n        \        V4      ^ 8X  d   V'       g   \        R	4      hR
V/V n        V P                  R
,           FI  pR\        P                   ! VP"                  V4      R\        P$                  ! R4      /V P                  V&   KK  	  R# )r   )Ú
stacklevelr
   r   r   r   r   r   r   z%optimizer got an empty parameter listr	   ÚsumÚstepç        N)r   Údefaultsr   r   r   r   ÚtorchÚjitÚannotateÚdictr   ÚstrÚstateÚlenÚ
ValueErrorÚparam_groupÚ	full_likeÚdataÚtensor)Úselfr	   r
   r   r   r   r   r   r   r   r   r   r   r   Úps   &&&&&&&&&&&&&& r   Ú__init__Ú_FunctionalAdagrad.__init__   sý   € ô  	;ÀaÕHà�"Ø˜Ø�3Ø˜LØ'Ð)BØ"Ð$8ØÐ 0ð
ˆŒð +ÔØŒØŒ
Ø ŒÜ—Y‘Y×'Ñ'¬¬U¯\©\¼4ÄÄUÇ\Á\Ð@QÕ;RÐ-RÕ(SÐUWÓXˆŒ
äˆv‹;˜!Ô×$;ÜÐDÓEÐEð % fÐ-ˆÔð ×!Ñ! (×+Ð+ˆAà”u—’ q§v¡vÐ/HÓIØœŸš SÓ)ðˆD�J‰J�q‹Mó ,r   c                ó>   <€ V ^8„  d   QhRS[ S[R,          ,          /# )r   Ú	gradientsN)r   r   )r   r   s   "€r   r   r   G   s   ø€ ÷ *ñ *™d¡6¨D¥=Õ1ñ *r   c                óÚ  € V P                   R ,          p. p. p. p. p\        V4      \        V4      8w  d3   \        RR\        V4       R2,           R\        V4       2,           4      hRRr‡\        V P                   R ,          V4       FŸ  w  ršV
f   K  WzP                  ,          pV\
        P                  ! V	4      ,          pVP                  V	4       VP                  V
4       V P                  V	,          pVP                  VR,          4       VP                  VR,          4       K¡  	  \
        P                  ! 4       ;_uu_ 4        \        P                  ! VVVVV P                  R	,          V P                  R
,          V P                  R,          V P                  R,          VV P                  V P                  VV P                  RRR7       RRR4       R#   + '       g   i     R# ; i)r	   zEthe gradients passed in does not equal to the size of the parameters!zParams length: z. zGradients length: FNr!   r"   r
   r   r   r   )r
   r   r   r   Úhas_sparse_gradr   r   Úhas_complexr   Ú
grad_scaleÚ	found_inf)r-   r+   r,   ÚzipÚ	is_sparser%   Ú
is_complexÚappendr*   Úno_gradÚFÚadagradr$   r   r   r   )r1   r6   r	   Úparams_with_gradÚgradsÚ
state_sumsÚstate_stepsr8   r9   ÚparamÚgradientr*   s   &&          r   r"   Ú_FunctionalAdagrad.stepG   s�  € Ø×!Ñ! (Õ+ˆØÐØˆØˆ
Ø$&ˆäˆv‹;œ#˜i›.Ô(ÜØWØ#¤C¨£K =°Ð3õ4à&¤s¨9£~Ð&6Ð7õ8óð ð (-¨e˜Ü" 4×#3Ñ#3°HÕ#=¸yÖI‰OˆEØÔ#Ø×#5Ñ#5Õ5�Øœu×/Ò/°Ó6Õ6�Ø ×'Ñ'¨Ô.Ø—‘˜XÔ&ØŸ
™
 5Õ)�Ø×!Ñ! %¨¥,Ô/Ø×"Ñ" 5¨¥=Ö1ñ  Jô �]Š]�_�_Ü�IŠIØØØØØ—=‘= Õ&Ø!Ÿ]™]¨>Õ:ØŸ™ zÕ2Ø—M‘M %Õ(Ø /ØŸ™ØŸ™Ø'Ø—j‘jØØõ÷ �_�_Š_ús   ÅBGÇG*	)r   r$   r   r   r   r-   r*   N)g{®Gáz„?r#   r#   r#   g      ð?r#   g»½×Ùß|Û=TFFFF)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r3   r"   Ú__static_attributes__Ú__classdictcell__)r   s   @r   r   r      s   ø‡ € ÷-ò -÷^*ö *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   )rR   r%   Útorch.optim._functionalÚoptimÚ_functionalrA   r   Ú,torch.distributed.optim._deprecation_warningr   rQ   r   r   )rR   s   @r   Ú<module>rW      s1   øð÷ €ß #Ð #Ý õð
 €Ó ÷Zó Zr   