
    sg                     |    d dl mZ d dlmZ d dlZdgZ G d dej                  j                  j                        Zy)    )Any)
deprecatedNautocastc                        e Zd ZdZ ede      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.
    zi`torch.cpu.amp.autocast(args...)` is deprecated. Please use `torch.amp.autocast('cpu', args...)` instead.)categoryTenableddtypecache_enabledc                     t         j                  j                         r|| _        d| _        || _        y t        |   d|||       y )Ncpu)r   r	   r
   )torch_jit_internalis_scripting_enableddevice
fast_dtypesuper__init__)selfr   r	   r
   	__class__s       N/var/www/html/venv/lib/python3.12/site-packages/torch/cpu/amp/autocast_mode.pyr   zautocast.__init__   sJ     ++-#DMDK#DO7%} 	 	
    c                 `    t         j                  j                         r| S t        |          S N)r   r   r   r   	__enter__)r   r   s    r   r   zautocast.__enter__%   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__0   s*    ++-Kw%%r   )__name__
__module____qualname____doc__r   FutureWarningr   bfloat16boolr	   r   r   r   r    r"   __classcell__)r   s   @r   r   r      s    
 	C "^^"	

 {{
 	


#; ;s ;C ;
& &r   )	typingr   typing_extensionsr   r   __all__ampautocast_moder    r   r   <module>r2      s3     (  ,(&uyy&&// (&r   