---
id: pr-flashinfer-1831
repo: flashinfer-ai/flashinfer
pr: 1831
title: Update the routing for TRTLLMGEN to support kimi k2 and qwen
author: ChristinaZ
date: '2025-10-22'
url: https://github.com/flashinfer-ai/flashinfer/pull/1831
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: 979b7590585a1cdfbfa62df1dbd852d8d81002d2
inclusion_reason: kernel file changes
changed_paths:
- csrc/trtllm_fused_moe_kernel_launcher.cu
- csrc/trtllm_fused_moe_routing_deepseek.cu
- csrc/trtllm_fused_moe_routing_llama4.cu
- csrc/trtllm_fused_moe_routing_renormalize.cu
- csrc/trtllm_fused_moe_runner.cu
- flashinfer/fused_moe/core.py
- include/flashinfer/trtllm/fused_moe/DevKernel.h
- include/flashinfer/trtllm/fused_moe/RoutingKernel.cuh
- include/flashinfer/trtllm/fused_moe/RoutingKernel.h
- include/flashinfer/trtllm/fused_moe/RoutingKernelTopK.cuh
- tests/moe/test_trtllm_gen_fused_moe.py
head_ref_oid: 4b3845247d423e3646fc66eb9d4fbb71316e5538
artifact_dir: evidence/pull-bundles/flashinfer/gh-1831
---
# Update the routing for TRTLLMGEN to support kimi k2 and qwen

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

## Changed Paths

- `csrc/trtllm_fused_moe_kernel_launcher.cu`
- `csrc/trtllm_fused_moe_routing_deepseek.cu`
- `csrc/trtllm_fused_moe_routing_llama4.cu`
- `csrc/trtllm_fused_moe_routing_renormalize.cu`
- `csrc/trtllm_fused_moe_runner.cu`
- `flashinfer/fused_moe/core.py`
- `include/flashinfer/trtllm/fused_moe/DevKernel.h`
- `include/flashinfer/trtllm/fused_moe/RoutingKernel.cuh`
- `include/flashinfer/trtllm/fused_moe/RoutingKernel.h`
- `include/flashinfer/trtllm/fused_moe/RoutingKernelTopK.cuh`
- `tests/moe/test_trtllm_gen_fused_moe.py`

## Evidence Bundle

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