o
    Zh1                     @   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                       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.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      @       @      ?  T
init_scalegrowth_factorbackoff_factorgrowth_intervalenabledreturnNc                    s   t  jd|||||d d S )Ncuda)r
   r   r   r   r   )super__init__)selfr
   r   r   r   r   	__class__ I/var/www/auris/lib/python3.10/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      s0    )Ztyping_extensionsr   ZtorchZtorch.amp.grad_scalerr   __all__ampr   r   r   r   r   <module>   s
    