o
    Zh                     @   s2   d dl mZ d dlZdgZG dd dejjZdS )    )
deprecatedN
GradScalerc                       sR   e Zd ZdZeded					dd	ed
ededededdf fddZ	  Z
S )r   z
    See :class:`torch.amp.GradScaler`.
    ``torch.cpu.amp.GradScaler(args...)`` is deprecated. Please use ``torch.amp.GradScaler("cpu", args...)`` instead.
    zm`torch.cpu.amp.GradScaler(args...)` is deprecated. Please use `torch.amp.GradScaler('cpu', args...)` instead.)category      @       @      ?  T
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnNc                    s   t  jd|||||d d S )Ncpu)r	   r
   r   r   r   )super__init__)selfr	   r
   r   r   r   	__class__ H/var/www/auris/lib/python3.10/site-packages/torch/cpu/amp/grad_scaler.pyr      s   
zGradScaler.__init__)r   r   r   r   T)__name__
__module____qualname____doc__r   FutureWarningfloatintboolr   __classcell__r   r   r   r   r   	   s0    )Ztyping_extensionsr   Ztorch__all__ampr   r   r   r   r   <module>   s    