feat!: return flash attention KV cache softmax LSE#852

voltjia merged 1 commit into

masterInfiniTensor/InfiniOps:masterfrom

feat/align-flash-attn-kvcache-returnsInfiniTensor/InfiniOps:feat/align-flash-attn-kvcache-returnsCopy head branch name to clipboard