---
id: pr-vllm-31246
repo: vllm-project/vllm
pr: 31246
title: '[Kernel] Add topk_sigmoid kernel'
author: xyang16
date: '2026-01-21'
url: https://github.com/vllm-project/vllm/pull/31246
source_category: upstream-code
architectures:
- sm100
tags:
- kernel-fusion
- moe
- tma
techniques:
- kernel-fusion
hardware_features:
- tma
kernel_types:
- moe
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 63227accf5af941215e5a7fd48c305a15e2c8814
inclusion_reason: kernel file changes
changed_paths:
- benchmarks/kernels/benchmark_fused_topk.py
- csrc/moe/moe_ops.h
- csrc/moe/topk_softmax_kernels.cu
- csrc/moe/torch_bindings.cpp
- tests/kernels/moe/test_fused_topk.py
- tests/model_executor/test_enabled_custom_ops.py
- vllm/_aiter_ops.py
- vllm/_custom_ops.py
- vllm/model_executor/layers/fused_moe/config.py
- vllm/model_executor/layers/fused_moe/router/fused_topk_bias_router.py
- vllm/model_executor/layers/fused_moe/router/fused_topk_router.py
- vllm/model_executor/layers/fused_moe/router/router_factory.py
- vllm/model_executor/models/minimax_m2.py
head_ref_oid: b33befc8615d2d5a7a08df997afbc16129df0dc0
artifact_dir: evidence/pull-bundles/vllm/gh-31246
---
# [Kernel] Add topk_sigmoid kernel

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

## Changed Paths

- `benchmarks/kernels/benchmark_fused_topk.py`
- `csrc/moe/moe_ops.h`
- `csrc/moe/topk_softmax_kernels.cu`
- `csrc/moe/torch_bindings.cpp`
- `tests/kernels/moe/test_fused_topk.py`
- `tests/model_executor/test_enabled_custom_ops.py`
- `vllm/_aiter_ops.py`
- `vllm/_custom_ops.py`
- `vllm/model_executor/layers/fused_moe/config.py`
- `vllm/model_executor/layers/fused_moe/router/fused_topk_bias_router.py`
- `vllm/model_executor/layers/fused_moe/router/fused_topk_router.py`
- `vllm/model_executor/layers/fused_moe/router/router_factory.py`
- `vllm/model_executor/models/minimax_m2.py`

## Evidence Bundle

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