o
    Iñhµ   ã                   @   s0   d dl mZmZmZmZmZmZ d dlmZ dS )é   )Ú_enter_autocastÚ_exit_autocastÚautocastÚ
custom_bwdÚ
custom_fwdÚis_autocast_available)Ú
GradScalerN)	Úautocast_moder   r   r   r   r   r   Úgrad_scalerr   © r   r   úF/var/www/vscode/kcb/lib/python3.10/site-packages/torch/amp/__init__.pyÚ<module>   s     