
    9i                     *   d dl Z d dlZd dlmZ d dlmZ d dlZg dZ ede           G d dej	        j
        j                              Z ed	e          d
             Z ede          dddd            Z ede          d             ZdS )    N)Any)
deprecated)autocast
custom_fwd
custom_bwdzk`torch.cuda.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cuda', 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   zSee :class:`torch.autocast`.

    ``torch.cuda.amp.autocast(args...)`` is deprecated. Please use ``torch.amp.autocast("cuda", args...)`` instead.
    )      Tenableddtypecache_enabledc                 F    t                                          |           S N)super__new__)clsr   r   r   	__class__s       O/var/www/icac/venv/lib/python3.11/site-packages/torch/cuda/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cuda)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G5 	 	
 	
 	
 	
 	
r   c                     t           j                                        r| S t                                                      S r   )r   r   r   r   	__enter__)r#   r   s    r   r%   zautocast.__enter__:   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__E   s7    ++-- 	Kww%%%r   )__name__
__module____qualname____doc__sysversion_infor   float16boolr   r   r   r"   r%   r   r*   r,   __classcell__)r   s   @r   r   r      sI         '!! !!&"&		( 	(	( ;	(  		( 	( 	( 	( 	( 	(	 	 	
 "]"	
 

 {
 	
 
 
 
 
 
# # # # #; ;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                 N    t           j        j                            | d|          S )Nr   )r   ampautocast_mode_cast)valuer   s     r   r:   r:   L   s!     9"((>>>r   z{`torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead.)cast_inputsc                b     t          j        t          j        j        d          | |          S )z
    ``torch.cuda.amp.custom_fwd(args...)`` is deprecated. Please use
    ``torch.amp.custom_fwd(args..., device_type='cuda')`` instead.
    r   device_typefwdr<   )	functoolspartialr   r8   r   r@   s     r   r   r   U   s8     G9UY1vFFF[   r   z{`torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead.c                 ^     t          j        t          j        j        d          |           S )z
    ``torch.cuda.amp.custom_bwd(args...)`` is deprecated. Please use
    ``torch.amp.custom_bwd(args..., device_type='cuda')`` instead.
    r   r>   )rB   rC   r   r8   r   )bwds    r   r   r   d   s*     G9UY1vFFFsKKKr   r   )rB   r2   typingr   typing_extensionsr   r   __all__FutureWarningr8   r9   r   r:   r   r   r   r   r   <module>rJ      sx       



       ( ( ( ( ( (  3
2
2 @  
6& 6& 6& 6& 6&uy&/ 6& 6& 
6&t P  
? ? 
? N  
     
 N  
L L 
L L Lr   