
    i                         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       U/var/www/html/engine/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     ++-#DMDK#DO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&uyy&&// 6&
6&r   