[CUTE] Allow grads to be preallocated#2065

tridao merged 1 commit into

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

drisspg:allow-for-out-of-placedrisspg/flash-attention:allow-for-out-of-placeCopy head branch name to clipboard