Ë
    M²j„  ã                   óž   — d dl Z d dlmZ d dlmZ d dlZdgZ ede¬«       G d„ dej                  j                  j                  «      «       Z
y)é    N)ÚAny)Ú
deprecatedÚautocastzi`torch.cpu.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cpu', args...)` instead.)Úcategoryc                   óø   ‡ — e Zd ZdZej
                  dk  r'dej                  dfdedej                  defˆ fd„Z
d„ Zdej                  dfdedej                  defˆ fd	„Zˆ fd
„Zdededefˆ fd„Zˆ fd„Zˆ xZS )r   z˜
    See :class:`torch.autocast`.
    ``torch.cpu.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cpu", args...)`` instead.
    )é   é   TÚenabledÚdtypeÚcache_enabledc                 ó"   •— t         ‰| �  | «      S ©N)ÚsuperÚ__new__)Úclsr
   r   r   Ú	__class__s       €úZ/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/cpu/amp/autocast_mode.pyr   zautocast.__new__   s   ø€ ô ‘7‘? 3Ó'Ð'ó    c                  ó   — y r   © )r   s    r   Ú__init_subclass__zautocast.__init_subclass__'   s   € Ør   c                 ó”   •— t         j                  j                  «       r|| _        d| _        || _        y t        ‰| �  d|||¬«       y )NÚcpu)r
   r   r   )ÚtorchÚ_jit_internalÚis_scriptingÚ_enabledÚdeviceÚ
fast_dtyper   Ú__init__)Úselfr
   r   r   r   s       €r   r    zautocast.__init__*   sJ   ø€ ô ×Ñ×+Ñ+Ô-Ø#ˆDŒMØˆDŒKØ#ˆDŒOØÜ‰ÑØ˜7¨%¸}ð 	õ 	
r   c                 ó`   •— t         j                  j                  «       r| S t        ‰| �  «       S r   )r   r   r   r   Ú	__enter__)r!   r   s    €r   r#   zautocast.__enter__9   s(   ø€ Ü×Ñ×+Ñ+Ô-ØˆKÜ‰wÑ Ó"Ð"r   Úexc_typeÚexc_valÚexc_tbc                 ód   •— t         j                  j                  «       ry t        ‰| �  |||«      S r   )r   r   r   r   Ú__exit__)r!   r$   r%   r&   r   s       €r   r(   zautocast.__exit__?   s,   ø€ Ü×Ñ×+Ñ+Ô-ØÜ‰wÑ ¨'°6Ó:Ð:r   c                 ób   •— t         j                  j                  «       r|S t        ‰| �  |«      S r   )r   r   r   r   Ú__call__)r!   Úfuncr   s     €r   r*   zautocast.__call__D   s*   ø€ Ü×Ñ×+Ñ+Ô-ØˆKÜ‰wÑ Ó%Ð%r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚsysÚversion_infor   Úbfloat16Úboolr   r   r   r    r#   r   r(   r*   Ú__classcell__)r   s   @r   r   r      s®   ø„ ñð ×Ñ˜'Ò!ð !Ø!&§¡Ø"&ñ		(àð	(ð —;‘;ð	(ð  õ		(ò	ð
 Ø"Ÿ^™^Ø"ñ	
àð
ð �{‰{ð
ð õ	
ô#ð; ð ;¨sð ;¸Cõ ;÷
&ð &r   )r0   Útypingr   Útyping_extensionsr   r   Ú__all__ÚFutureWarningÚampÚautocast_moder   r   r   r   ú<module>r;      sR   ðã 
Ý Ý (ã ð ˆ,€ñ ð?àôô
6&ˆu�y‰y×&Ñ&×/Ñ/ó 6&óñ
6&r   