split out varlen batch search into utils#2556

jayhshah merged 2 commits into

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

reubenconducts:rstern/search-utilreubenconducts/flash-attention:rstern/search-utilCopy head branch name to clipboard