[Feature Request] Return LSE from all TRT LLM attention kernels #2169
Description
Currently only some APIs return LSEs, while underlying kernels are capable of returning LSEs in all cases. Extend API to always return LSEs if asked. Example implementation: fw-ai/flashinfer#7