---
id: pr-flashinfer-2114
repo: flashinfer-ai/flashinfer
pr: 2114
title: 'feature: make the LSE returned by MLA support base 2 or e  #2113'
author: staugust
date: '2025-11-20'
url: https://github.com/flashinfer-ai/flashinfer/pull/2114
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 0753095742679c0f4ec9649b0dd3a611b4ee22d2
inclusion_reason: kernel file changes
changed_paths:
- csrc/batch_mla_binding.cu
- csrc/batch_mla_run.cu
- csrc/batch_mla_sm90_binding.cu
- csrc/batch_mla_sm90_run.cu
- flashinfer/mla.py
- include/flashinfer/attention/mla.cuh
- include/flashinfer/attention/mla_hopper.cuh
- include/flashinfer/attention/mla_params.cuh
head_ref_oid: f42019e22a2d2b0ea32060f3387cab8f0275268c
artifact_dir: evidence/pull-bundles/flashinfer/gh-2114
---
# feature: make the LSE returned by MLA support base 2 or e  #2113

- Repository: `flashinfer-ai/flashinfer`
- PR: [2114](https://github.com/flashinfer-ai/flashinfer/pull/2114)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `csrc/batch_mla_binding.cu`
- `csrc/batch_mla_run.cu`
- `csrc/batch_mla_sm90_binding.cu`
- `csrc/batch_mla_sm90_run.cu`
- `flashinfer/mla.py`
- `include/flashinfer/attention/mla.cuh`
- `include/flashinfer/attention/mla_hopper.cuh`
- `include/flashinfer/attention/mla_params.cuh`

## Evidence Bundle

`evidence/pull-bundles/flashinfer/gh-2114`
