ó
    Eñi•  ã                   óü   • S SK r S SKrS SKJr  S SKJr  S SKr/ SQr\" S\S9 " S S\R                  R                  R                  5      5       r\" S	\S9S
 5       r\" S\S9SSS.S jj5       r\" S\S9S 5       rg)é    N)ÚAny)Ú
deprecated)ÚautocastÚ
custom_fwdÚ
custom_bwdzk`torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', args...)` instead.)Úcategoryc                   ó  ^ • \ rS rSrSr\R                  S:  a*  S\R                  S4S\	S\R                  S\	4U 4S jjjrS	 rS\R                  S4S\	S\R                  S\	4U 4S
 jjjrU 4S jrS\S\S\4U 4S jjrU 4S jrSrU =r$ )r   é   zŽSee :class:`torch.autocast`.

``torch.cuda.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cuda", args...)`` instead.
)é   r
   TÚenabledÚdtypeÚcache_enabledc                 ó"   >• [         TU ]  U 5      $ ©N)ÚsuperÚ__new__)Úclsr   r   r   Ú	__class__s       €ÚY/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/cuda/amp/autocast_mode.pyr   Úautocast.__new__    s   ø€ ô ‘7‘? 3Ó'Ð'ó    c                 ó   • g r   © )r   s    r   Ú__init_subclass__Úautocast.__init_subclass__(   s   € Ør   c                 ó’   >• [         R                  R                  5       (       a  Xl        SU l        X l        g [        TU ]  SXUS9  g )NÚcuda)r   r   r   )ÚtorchÚ_jit_internalÚis_scriptingÚ_enabledÚdeviceÚ
fast_dtyper   Ú__init__)Úselfr   r   r   r   s       €r   r$   Úautocast.__init__+   sG   ø€ ô ×Ñ×+Ñ+×-Ñ-Ø#ŒMØ ˆDŒKØ#ŒOØÜ‰ÑØ˜GÀð 	ò 	
r   c                 ój   >• [         R                  R                  5       (       a  U $ [        TU ]  5       $ r   )r   r   r    r   Ú	__enter__)r%   r   s    €r   r(   Úautocast.__enter__:   s+   ø€ Ü×Ñ×+Ñ+×-Ñ-ØˆKÜ‰wÑ Ó"Ð"r   Úexc_typeÚexc_valÚexc_tbc                 ól   >• [         R                  R                  5       (       a  g [        TU ]  XU5      $ r   )r   r   r    r   Ú__exit__)r%   r*   r+   r,   r   s       €r   r.   Úautocast.__exit__@   s-   ø€ Ü×Ñ×+Ñ+×-Ñ-ØÜ‰wÑ °6Ó:Ð:r   c                 ól   >• [         R                  R                  5       (       a  U$ [        TU ]  U5      $ r   )r   r   r    r   Ú__call__)r%   Úfuncr   s     €r   r1   Úautocast.__call__E   s-   ø€ Ü×Ñ×+Ñ+×-Ñ-ØˆKÜ‰wÑ Ó%Ð%r   )r!   r"   r#   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__ÚsysÚversion_infor   Úfloat16Úboolr   r   r   r$   r(   r   r.   r1   Ú__static_attributes__Ú__classcell__)r   s   @r   r   r      s¸   ø† ñð ×Ñ˜'Ó!ð !Ø!&§¡Ø"&ñ		(àð	(ð —;‘;ð	(ð  ÷		(ð 	(ò	ð
 Ø"Ÿ]™]Ø"ñ	
àð
ð �{‰{ð
ð ÷	
ð 
õ#ð; ð ;¨sð ;¸C÷ ;÷
&ó &r   r   z‹`torch.cuda.amp.autocast_mode._cast(value, dtype)` is deprecated. Please use `torch.amp.autocast_mode._cast(value, 'cuda', dtype)` instead.c                 óX   • [         R                  R                  R                  U SU5      $ )Nr   )r   ÚampÚautocast_modeÚ_cast)Úvaluer   s     r   rB   rB   L   s#   € ô �9‰9×"Ñ"×(Ñ(¨°¸Ó>Ð>r   z{`torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead.)Úcast_inputsc                ód   • [         R                  " [        R                  R                  SS9" XS9$ )z�
``torch.cuda.amp.custom_fwd(args...)`` is deprecated. Please use
``torch.amp.custom_fwd(args..., device_type='cuda')`` instead.
r   ©Údevice_type©ÚfwdrD   )Ú	functoolsÚpartialr   r@   r   rH   s     r   r   r   U   s+   € ô ×ÒœUŸY™Y×1Ñ1¸vÒFØñð r   z{`torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead.c                 óh   • [         R                  " [        R                  R                  SS9" U 5      $ )z�
``torch.cuda.amp.custom_bwd(args...)`` is deprecated. Please use
``torch.amp.custom_bwd(args..., device_type='cuda')`` instead.
r   rF   )rJ   rK   r   r@   r   )Úbwds    r   r   r   d   s&   € ô ×ÒœUŸY™Y×1Ñ1¸vÒFÀsÓKÐKr   r   )rJ   r9   Útypingr   Útyping_extensionsr   r   Ú__all__ÚFutureWarningr@   rA   r   rB   r   r   r   r   r   Ú<module>rR      sÄ   ðã Û 
Ý Ý (ã ò 3€ñ ð@àñô
6&ˆu�y‰y×&Ñ&×/Ñ/ó 6&óð
6&ñt ðPàññ
?óð
?ñ ðNàñð
¨õ óð
ñ ðNàññ
Lóñ
Lr   