o
    ei                     @   sT   d dl Z d dlmZ d dlmZ d dlZdgZededG d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                       s   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dZ
d	d
 Zdejdfdedej	def fddZ fddZdededef fddZ fddZ  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                    s   t  | S N)super__new__)clsr   r	   r
   	__class__ e/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/cpu/amp/autocast_mode.pyr      s   zautocast.__new__c                 C   s   d S r   r   )r   r   r   r   __init_subclass__'   s   zautocast.__init_subclass__c                    s8   t j r|| _d| _|| _d S t jd|||d d S )Ncpu)r   r	   r
   )torch_jit_internalis_scripting_enableddevice
fast_dtyper   __init__)selfr   r	   r
   r   r   r   r   *   s   

zautocast.__init__c                    s   t j r| S t  S r   )r   r   r   r   	__enter__)r   r   r   r   r   9   s   

zautocast.__enter__exc_typeexc_valexc_tbc                    s   t j rd S t |||S r   )r   r   r   r   __exit__)r   r   r   r    r   r   r   r!   ?   s   
zautocast.__exit__c                    s   t j r|S t |S r   )r   r   r   r   __call__)r   funcr   r   r   r"   D   s   
zautocast.__call__)__name__
__module____qualname____doc__sysversion_infor   bfloat16boolr	   r   r   r   r   r   r!   r"   __classcell__r   r   r   r   r      s6    
)r(   typingr   typing_extensionsr   r   __all__FutureWarningampautocast_moder   r   r   r   r   <module>   s   