
    sg                     8    d dl mZmZmZ d dlmZ d dlmZ g dZy)   )autocast
custom_bwd
custom_fwd)amp_definitely_not_available)
GradScaler)r   r   r   r   r   N)	autocast_moder   r   r   commonr   grad_scalerr   __all__     J/var/www/html/venv/lib/python3.12/site-packages/torch/cuda/amp/__init__.py<module>r      s    ; ; 0 #r   