---
id: pr-vllm-17110
repo: vllm-project/vllm
pr: 17110
title: '[FEAT] [ROCm]: Add AITER CK 2 Stages MoE support'
author: tjtanaa
date: '2025-05-14'
url: https://github.com/vllm-project/vllm/pull/17110
source_category: upstream-code
architectures:
- sm100
tags:
- fp8
- kernel-fusion
- moe
- quantization
techniques:
- kernel-fusion
hardware_features:
- fp8
kernel_types:
- moe
- quantization
languages:
- python
captured_at: '2026-04-27'
status: merged
merge_sha: 612c2edb4f940b26550ca64576995957e4118a12
inclusion_reason: kernel file changes
changed_paths:
- tests/kernels/moe/test_moe.py
- tests/model_executor/test_enabled_custom_ops.py
- vllm/model_executor/layers/fused_moe/fused_moe.py
- vllm/model_executor/layers/fused_moe/layer.py
- vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py
- vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe.py
- vllm/model_executor/layers/quantization/fp8.py
head_ref_oid: fca957b7243bad9df88ed159fcd27b1036b6cabb
artifact_dir: evidence/pull-bundles/vllm/gh-17110
---
# [FEAT] [ROCm]: Add AITER CK 2 Stages MoE support

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

## Changed Paths

- `tests/kernels/moe/test_moe.py`
- `tests/model_executor/test_enabled_custom_ops.py`
- `vllm/model_executor/layers/fused_moe/fused_moe.py`
- `vllm/model_executor/layers/fused_moe/layer.py`
- `vllm/model_executor/layers/fused_moe/rocm_aiter_fused_moe.py`
- `vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors_moe.py`
- `vllm/model_executor/layers/quantization/fp8.py`

## Evidence Bundle

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