---
id: pr-flashinfer-3021
repo: flashinfer-ai/flashinfer
pr: 3021
title: 'fix: extend moe alltoall top-k specializations'
author: bobboli
date: '2026-04-13'
url: https://github.com/flashinfer-ai/flashinfer/pull/3021
source_category: upstream-code
architectures:
- sm100
tags:
- moe
techniques: []
hardware_features: []
kernel_types:
- moe
languages:
- cuda-cpp
captured_at: '2026-04-27'
status: merged
merge_sha: 19055a650750eb3998bbfcdc1e19d910ef3bf8a2
inclusion_reason: kernel file changes
changed_paths:
- csrc/nv_internal/tensorrt_llm/kernels/communicationKernels/moeAlltoAllKernels.cu
- csrc/nv_internal/tensorrt_llm/kernels/communicationKernels/moeAlltoAllKernels.h
- tests/comm/test_trtllm_moe_alltoall.py
head_ref_oid: 553f64094b6345cde44780019c21cedc1b759f77
artifact_dir: evidence/pull-bundles/flashinfer/gh-3021
---
# fix: extend moe alltoall top-k specializations

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

## Changed Paths

- `csrc/nv_internal/tensorrt_llm/kernels/communicationKernels/moeAlltoAllKernels.cu`
- `csrc/nv_internal/tensorrt_llm/kernels/communicationKernels/moeAlltoAllKernels.h`
- `tests/comm/test_trtllm_moe_alltoall.py`

## Evidence Bundle

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