
    9i                      2    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	        E/var/www/icac/venv/lib/python3.11/site-packages/torch/amp/__init__.py<module>r      so                   $ # # # # # # #r   