---
id: pr-flashinfer-1924
repo: flashinfer-ai/flashinfer
pr: 1924
title: 'MLA RoPE + quantization fused kernel: shape generalization for MHA / GQA'
author: kahyunnam
date: '2025-10-18'
url: https://github.com/flashinfer-ai/flashinfer/pull/1924
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- fp8
- kernel-fusion
- mla
- quantization
techniques:
- kernel-fusion
hardware_features:
- fp8
kernel_types:
- attention
- mla
- quantization
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 8a88b0e31a51eb1afa1b7e23b385c5e4b9f84d48
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/bench_mla_rope_quantize_fp8.py
- benchmarks/bench_rope_quantize_fp8.py
- csrc/flashinfer_rope_binding.cu
- csrc/rope.cu
- flashinfer/rope.py
- include/flashinfer/pos_enc.cuh
- tests/attention/test_rope.py
head_ref_oid: 19e3d04c8ffda56529538affbcd13ff17331b67c
artifact_dir: evidence/pull-bundles/flashinfer/gh-1924
---
# MLA RoPE + quantization fused kernel: shape generalization for MHA / GQA

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

## Changed Paths

- `benchmarks/bench_mla_rope_quantize_fp8.py`
- `benchmarks/bench_rope_quantize_fp8.py`
- `csrc/flashinfer_rope_binding.cu`
- `csrc/rope.cu`
- `flashinfer/rope.py`
- `include/flashinfer/pos_enc.cuh`
- `tests/attention/test_rope.py`

## Evidence Bundle

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