a
    h1                     @   s>   d dl mZ d dlZd dlmZ dgZG dd dejjZdS )    )
deprecatedN)OptState
GradScalerc                	       s>   e Zd ZdZededdeeeeed	d
 fddZ	  Z
S )r   z
    See :class:`torch.amp.GradScaler`.
    ``torch.cuda.amp.GradScaler(args...)`` is deprecated. Please use ``torch.amp.GradScaler("cuda", args...)`` instead.
    zo`torch.cuda.amp.GradScaler(args...)` is deprecated. Please use `torch.amp.GradScaler('cuda', args...)` instead.)category      @       @      ?  TN)
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnc                    s   t  jd|||||d d S )NZcuda)r
   r   r   r   r   )super__init__)selfr
   r   r   r   r   	__class__ H/var/www/auris/lib/python3.9/site-packages/torch/cuda/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      s$        )Ztyping_extensionsr   ZtorchZtorch.amp.grad_scalerr   __all__ampr   r   r   r   r   <module>   s   