---
id: pr-vllm-16756
repo: vllm-project/vllm
pr: 16756
title: '[torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass'
author: ProExpertProg
date: '2025-06-12'
url: https://github.com/vllm-project/vllm/pull/16756
source_category: upstream-code
architectures:
- sm100
tags:
- attention
- kernel-fusion
- mla
- pipeline-stages
- quantization
techniques:
- kernel-fusion
- pipeline-stages
hardware_features: []
kernel_types:
- attention
- mla
- quantization
languages:
- triton
captured_at: '2026-04-27'
status: merged
merge_sha: f98548b9da39bce8aaff64974e1e68208aeada5e
inclusion_reason: kernel file changes
changed_paths:
- .buildkite/test-pipeline.yaml
- tests/compile/backend.py
- tests/compile/test_async_tp.py
- tests/compile/test_fusion.py
- tests/compile/test_fusion_attn.py
- vllm/_custom_ops.py
- vllm/attention/backends/abstract.py
- vllm/attention/backends/blocksparse_attn.py
- vllm/attention/backends/dual_chunk_flash_attn.py
- vllm/attention/backends/flash_attn.py
- vllm/attention/backends/flashinfer.py
- vllm/attention/backends/hpu_attn.py
- vllm/attention/backends/ipex_attn.py
- vllm/attention/backends/mla/common.py
- vllm/attention/backends/pallas.py
- vllm/attention/backends/rocm_flash_attn.py
- vllm/attention/backends/torch_sdpa.py
- vllm/attention/backends/xformers.py
- vllm/attention/layer.py
- vllm/compilation/fusion.py
- vllm/compilation/fusion_attn.py
- vllm/compilation/fx_utils.py
- vllm/compilation/noop_elimination.py
- vllm/compilation/pass_manager.py
- vllm/compilation/vllm_inductor_pass.py
- vllm/config.py
- vllm/envs.py
- vllm/v1/attention/backends/flash_attn.py
- vllm/v1/attention/backends/flashinfer.py
- vllm/v1/attention/backends/flex_attention.py
- vllm/v1/attention/backends/mla/common.py
- vllm/v1/attention/backends/pallas.py
- vllm/v1/attention/backends/triton_attn.py
head_ref_oid: 1a8a79460e54632ca90c973e2a259ed698d139dc
artifact_dir: evidence/pull-bundles/vllm/gh-16756
---
# [torch.compile][ROCm] Fuse quantization onto attention using a torch.compile pass

- Repository: `vllm-project/vllm`
- PR: [16756](https://github.com/vllm-project/vllm/pull/16756)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `.buildkite/test-pipeline.yaml`
- `tests/compile/backend.py`
- `tests/compile/test_async_tp.py`
- `tests/compile/test_fusion.py`
- `tests/compile/test_fusion_attn.py`
- `vllm/_custom_ops.py`
- `vllm/attention/backends/abstract.py`
- `vllm/attention/backends/blocksparse_attn.py`
- `vllm/attention/backends/dual_chunk_flash_attn.py`
- `vllm/attention/backends/flash_attn.py`
- `vllm/attention/backends/flashinfer.py`
- `vllm/attention/backends/hpu_attn.py`
- `vllm/attention/backends/ipex_attn.py`
- `vllm/attention/backends/mla/common.py`
- `vllm/attention/backends/pallas.py`
- `vllm/attention/backends/rocm_flash_attn.py`
- `vllm/attention/backends/torch_sdpa.py`
- `vllm/attention/backends/xformers.py`
- `vllm/attention/layer.py`
- `vllm/compilation/fusion.py`
- `vllm/compilation/fusion_attn.py`
- `vllm/compilation/fx_utils.py`
- `vllm/compilation/noop_elimination.py`
- `vllm/compilation/pass_manager.py`
- `vllm/compilation/vllm_inductor_pass.py`
- `vllm/config.py`
- `vllm/envs.py`
- `vllm/v1/attention/backends/flash_attn.py`
- `vllm/v1/attention/backends/flashinfer.py`
- `vllm/v1/attention/backends/flex_attention.py`
- `vllm/v1/attention/backends/mla/common.py`
- `vllm/v1/attention/backends/pallas.py`
- `vllm/v1/attention/backends/triton_attn.py`

## Evidence Bundle

`evidence/pull-bundles/vllm/gh-16756`
