
    sh                     8    S SK JrJrJr  S SKJr  S SKJr  / SQrg)   )autocast
custom_bwd
custom_fwd)amp_definitely_not_available)
GradScaler)r   r   r   r   r   N)	autocast_moder   r   r   commonr   grad_scalerr   __all__     k/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/cuda/amp/__init__.py<module>r      s    ; ; 0 #r   