---
id: pr-vllm-29642
repo: vllm-project/vllm
pr: 29642
title: '[Kernel][MoE] optimize `moe_align_block_size`'
author: jinzhen-lin
date: '2025-12-07'
url: https://github.com/vllm-project/vllm/pull/29642
source_category: upstream-code
architectures:
- sm100
tags:
- kernel-fusion
- moe
techniques:
- kernel-fusion
hardware_features: []
kernel_types:
- moe
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 879ddb09c3b7b5a21f75e46ded148ce70f1c486e
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/kernels/benchmark_moe_align_block_size.py
- csrc/moe/moe_align_sum_kernels.cu
- csrc/moe/moe_ops.h
- csrc/moe/torch_bindings.cpp
- tests/kernels/moe/test_moe.py
- tests/kernels/moe/test_moe_align_block_size.py
- vllm/_custom_ops.py
- vllm/model_executor/layers/fused_moe/fused_marlin_moe.py
- vllm/model_executor/layers/fused_moe/fused_moe.py
- vllm/model_executor/layers/fused_moe/moe_align_block_size.py
head_ref_oid: f7e3a0879e633f9924d477b85a3775cb1ae2a2e5
artifact_dir: evidence/pull-bundles/vllm/gh-29642
---
# [Kernel][MoE] optimize `moe_align_block_size`

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

## Changed Paths

- `benchmarks/kernels/benchmark_moe_align_block_size.py`
- `csrc/moe/moe_align_sum_kernels.cu`
- `csrc/moe/moe_ops.h`
- `csrc/moe/torch_bindings.cpp`
- `tests/kernels/moe/test_moe.py`
- `tests/kernels/moe/test_moe_align_block_size.py`
- `vllm/_custom_ops.py`
- `vllm/model_executor/layers/fused_moe/fused_marlin_moe.py`
- `vllm/model_executor/layers/fused_moe/fused_moe.py`
- `vllm/model_executor/layers/fused_moe/moe_align_block_size.py`

## Evidence Bundle

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