o
    Hñh  ã                   @   s8   d dl mZmZmZ d dlmZ d dlmZ g d¢ZdS )é   )ÚautocastÚ
custom_bwdÚ
custom_fwd)Úamp_definitely_not_available)Ú
GradScaler)r   r   r   r   r   N)	Úautocast_moder   r   r   Úcommonr   Úgrad_scalerr   Ú__all__© r   r   úK/var/www/vscode/kcb/lib/python3.10/site-packages/torch/cuda/amp/__init__.pyÚ<module>   s    