Refactor attention mask and bias handling for efficiency#177

LoserCheems merged 2 commits intomainflash-algo/flash-sparse-attention:mainfrom

update-exampleflash-algo/flash-sparse-attention:update-exampleCopy head branch name to clipboard