---
id: pr-flashinfer-898
repo: flashinfer-ai/flashinfer
pr: 898
title: 'perf: fix MLA split-k performance bug'
author: yzh119
date: '2025-02-26'
url: https://github.com/flashinfer-ai/flashinfer/pull/898
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- mla
techniques: []
hardware_features: []
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-04-17'
status: merged
merge_sha: 1e330b712c23315d76d9f1692a0a225623666d1a
inclusion_reason: kernel file changes
changed_paths:
- csrc/batch_mla_run.cu
- csrc/batch_mla_sm90_run.cu
- include/flashinfer/attention/mla.cuh
- include/flashinfer/attention/mla_hopper.cuh
- include/flashinfer/attention/mla_params.cuh
- include/flashinfer/attention/scheduler.cuh
head_ref_oid: 19cb7193fe217a3ae5a6d1a54e80c533b2dad0c3
artifact_dir: evidence/pull-bundles/flashinfer/gh-898
---
# perf: fix MLA split-k performance bug

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

## Changed Paths

- `csrc/batch_mla_run.cu`
- `csrc/batch_mla_sm90_run.cu`
- `include/flashinfer/attention/mla.cuh`
- `include/flashinfer/attention/mla_hopper.cuh`
- `include/flashinfer/attention/mla_params.cuh`
- `include/flashinfer/attention/scheduler.cuh`

## Evidence Bundle

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