
    9i                         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       N/var/www/icac/venv/lib/python3.11/site-packages/torch/cpu/amp/autocast_mode.pyr   zautocast.__new__   s     77??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     ++-- 	#DMDK#DO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ww  """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ww'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ww%%%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&uy&/ 6& 6& 
6& 6& 6&r   