
    ^j(                     8    d dl mZmZmZmZmZmZ d dlmZ g dZ	y)   )_enter_autocast_exit_autocastautocast
custom_bwd
custom_fwdis_autocast_available)
GradScaler)r   r   r   r   r	   N)
autocast_moder   r   r   r   r   r   grad_scalerr	   __all__     ]/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/amp/__init__.py<module>r      s     $r   