§
    ™/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
dS )é    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                 óF   •— t          ¦   «                              | ¦  «        S ©N)ÚsuperÚ__new__)Úclsr
   r   r   Ú	__class__s       €ú`/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torch/cpu/amp/autocast_mode.pyr   zautocast.__new__   s   ø€ õ ‘7”7—?’? 3Ñ'Ô'Ð'ó    c                 ó   — d S r   © )r   s    r   Ú__init_subclass__zautocast.__init_subclass__'   s   € ØˆDr   c                 ó¼   •— t           j                             ¦   «         r|| _        d| _        || _        d S t          ¦   «                              d|||¬¦  «         d S )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ØˆFÝ‰Œ×ÒØ˜7¨%¸}ð 	ñ 	
ô 	
ð 	
ð 	
ð 	
r   c                 ó„   •— t           j                             ¦   «         r| S t          ¦   «                              ¦   «         S r   )r   r   r   r   Ú	__enter__)r!   r   s    €r   r#   zautocast.__enter__9   s5   ø€ ÝÔ×+Ò+Ñ-Ô-ð 	ØˆKÝ‰wŒw× Ò Ñ"Ô"Ð"r   Úexc_typeÚexc_valÚexc_tbc                 óŠ   •— t           j                             ¦   «         rd S t          ¦   «                              |||¦  «        S r   )r   r   r   r   Ú__exit__)r!   r$   r%   r&   r   s       €r   r(   zautocast.__exit__?   s;   ø€ ÝÔ×+Ò+Ñ-Ô-ð 	ØˆFÝ‰wŒw×Ò ¨'°6Ñ:Ô:Ð:r   c                 ó†   •— t           j                             ¦   «         r|S t          ¦   «                              |¦  «        S r   )r   r   r   r   Ú__call__)r!   Úfuncr   s     €r   r*   zautocast.__call__D   s7   ø€ ÝÔ×+Ò+Ñ-Ô-ð 	ØˆKÝ‰wŒw×Ò Ñ%Ô%Ð%r   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚsysÚversion_infor   Úbfloat16Úboolr   r   r   r    r#   r   r(   r*   Ú__classcell__)r   s   @r   r   r      sI  ø€ € € € € ðð ð Ô˜'Ò!Ð!ð !Ø!&¤Ø"&ð		(ð 	(àð	(ð ”;ð	(ð  ð		(ð 	(ð 	(ð 	(ð 	(ð 	(ð	ð 	ð 	ð
 Ø"œ^Ø"ð	
ð 
àð
ð Œ{ð
ð ð	
ð 
ð 
ð 
ð 
ð 
ð#ð #ð #ð #ð #ð; ð ;¨sð ;¸Cð ;ð ;ð ;ð ;ð ;ð ;ð
&ð &ð &ð &ð &ð &ð &ð &ð &r   )r0   Útypingr   Útyping_extensionsr   r   Ú__all__ÚFutureWarningÚampÚautocast_moder   r   r   r   ú<module>r;      s¨   ðà 
€
€
€
Ø Ð Ð Ð Ð Ð Ø (Ð (Ð (Ð (Ð (Ð (à €€€ð ˆ,€ð €ð?àðñ ô ð
6&ð 6&ð 6&ð 6&ð 6&ˆuŒyÔ&Ô/ñ 6&ô 6&ñô ð
6&ð 6&ð 6&r   