[Custom Fwd/Bwd] Fix deprecation warning for newer torch versions#1565

tridao merged 1 commit into

Dao-AILab:mainDao-AILab/flash-attention:mainfrom

vasqu:torch-amp-deprecationvasqu/flash-attention:torch-amp-deprecationCopy head branch name to clipboard