[Cute][Flex] Allow q_offset 1 and add block-sizes to disambiguate edge cases#2187

drisspg merged 1 commit into

mainDao-AILab/flash-attention:mainfrom

drisspg/stack/3Dao-AILab/flash-attention:drisspg/stack/3Copy head branch name to clipboard