---
id: pr-tensorrt-llm-13433
repo: NVIDIA/TensorRT-LLM
pr: 13433
title: '[None][perf] Extend customMoeRouting kernel to support Qwen3.5'
author: nv-guomingz
date: '2026-04-29'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/13433
source_category: upstream-code
architectures:
- sm100
tags:
- benchmark_test
- cuda kernel
- moe_routing
- routing
techniques:
- async-copy
- kernel-fusion
hardware_features:
- fp4
- nvfp4
- tma
kernel_types:
- moe
- norm
- softmax
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 338e94ed65f9d568ee91c8c1d61477bed7bf94e4
inclusion_reason: Separate routing, permutation, top-k, and expert GEMM costs; keep tail-token and expert-layout tests attached to the idea.
changed_paths:
- cpp/tensorrt_llm/kernels/customMoeRoutingKernels.cu
- cpp/tensorrt_llm/kernels/moeTopKFuncs.cuh
- cpp/tensorrt_llm/thop/customMoeRoutingOp.cpp
- tensorrt_llm/_torch/modules/fused_moe/routing.py
- tests/unittest/_torch/modules/test_moe_routing.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-13433
head_ref_oid: bc693ed5a3161d9ba050ae74f0940ffb581eee1c
---
# [None][perf] Extend customMoeRouting kernel to support Qwen3.5

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [13433](https://github.com/NVIDIA/TensorRT-LLM/pull/13433)
- Decision: `Separate routing, permutation, top-k, and expert GEMM costs; keep tail-token and expert-layout tests attached to the idea.`
- Reason: Separate routing, permutation, top-k, and expert GEMM costs; keep tail-token and expert-layout tests attached to the idea.

## Changed Paths

- `cpp/tensorrt_llm/kernels/customMoeRoutingKernels.cu`
- `cpp/tensorrt_llm/kernels/moeTopKFuncs.cuh`
- `cpp/tensorrt_llm/thop/customMoeRoutingOp.cpp`
- `tensorrt_llm/_torch/modules/fused_moe/routing.py`
- `tests/unittest/_torch/modules/test_moe_routing.py`

## Evidence Bundle

`evidence/pull-bundles/tensorrt-llm/gh-13433`
