---
id: pr-flashinfer-952
repo: flashinfer-ai/flashinfer
pr: 952
title: 'perf: Use 2WG pipeline design for MLA implementation on Hopper'
author: yzh119
date: '2025-03-29'
url: https://github.com/flashinfer-ai/flashinfer/pull/952
source_category: upstream-code
architectures:
- sm100
- sm90
tags:
- attention
- mla
- pipeline-stages
techniques:
- pipeline-stages
hardware_features: []
kernel_types:
- attention
- mla
languages:
- cuda-cpp
captured_at: '2026-04-17'
status: merged
merge_sha: 60d37b741f88f11159e7eec35e2261aa668dc2a4
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/bench_deepseek_mla.py
- include/flashinfer/attention/mla_hopper.cuh
head_ref_oid: ab1d5d96720f10e7bafd6654c8786715352d3e42
artifact_dir: evidence/pull-bundles/flashinfer/gh-952
---
# perf: Use 2WG pipeline design for MLA implementation on Hopper

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

## Changed Paths

- `benchmarks/bench_deepseek_mla.py`
- `include/flashinfer/attention/mla_hopper.cuh`

## Evidence Bundle

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