TODO

• Add flashattention v2 cuda version
• Add flashattention v2 metal version

Reference

flashattention v1 cuda version https://github.com/tspeterkim/flash-attention-minimal/blob/main/flash.cu video: https://www.bilibili.com/video/BV1zM4m1S7gg/?share_source=copy_web&vd_source=bb7496f78e4d303270b7c97ae8f69402