---
id: pr-flashinfer-3039
repo: flashinfer-ai/flashinfer
pr: 3039
title: 'feat: Integrate CuTe DSL FMHA prefill kernels by loading cubin'
author: limin2021
date: '2026-04-24'
url: https://github.com/flashinfer-ai/flashinfer/pull/3039
source_category: upstream-code
architectures:
- sm100
tags:
- TMA
- attention_kv
- benchmark
- benchmark_test
- blackwell
- compiler_runtime
- cuda kernel
- cute dsl
- fp8
- mla
- mxfp4
techniques:
- async-copy
hardware_features:
- blackwell
- fp4
- fp8
- mxfp4
- tma
kernel_types:
- attention
- compiler
- mla
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: bb873d207c8d6ce5c224c90b3b714a68714472f5
inclusion_reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.
changed_paths:
- benchmarks/routines/attention.py
- benchmarks/routines/flashinfer_benchmark_utils.py
- flashinfer/artifacts.py
- flashinfer/attention/__init__.py
- flashinfer/attention/_core.py
- flashinfer/attention/cute_dsl/__init__.py
- flashinfer/attention/cute_dsl/fmha.py
- flashinfer/prefill.py
- tests/attention/test_trtllm_gen_attention.py
- tests/test_artifacts.py
artifact_dir: evidence/pull-bundles/flashinfer/gh-3039
head_ref_oid: 9d79f42b631b71b94b89020f631e5ed199e18c6b
---
# feat: Integrate CuTe DSL FMHA prefill kernels by loading cubin

- Repository: `flashinfer-ai/flashinfer`
- PR: [3039](https://github.com/flashinfer-ai/flashinfer/pull/3039)
- Decision: `Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.`
- Reason: Preserve page/KV/layout and plan-run contracts; profile memory traffic separately from math utilization.

## Changed Paths

- `benchmarks/routines/attention.py`
- `benchmarks/routines/flashinfer_benchmark_utils.py`
- `flashinfer/artifacts.py`
- `flashinfer/attention/__init__.py`
- `flashinfer/attention/_core.py`
- `flashinfer/attention/cute_dsl/__init__.py`
- `flashinfer/attention/cute_dsl/fmha.py`
- `flashinfer/prefill.py`
- `tests/attention/test_trtllm_gen_attention.py`
- `tests/test_artifacts.py`

## Evidence Bundle

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