Flash Attention

通过优化内存访问模式加速注意力计算并降低显存占用的算法