
    ,h                      0    S SK JrJrJrJrJrJr  S SKJr  g)   )_enter_autocast_exit_autocastautocast
custom_bwd
custom_fwdis_autocast_available)
GradScalerN)	autocast_moder   r   r   r   r   r   grad_scalerr	        J/var/www/html/shao/venv/lib/python3.13/site-packages/torch/amp/__init__.py<module>r      s     $r   