---
id: pr-tensorrt-llm-12884
repo: NVIDIA/TensorRT-LLM
pr: 12884
title: '[TRTLLM-11585][feat] Add CUTEDSL moe backend for nemotron-h'
author: Wanli-Jiang
date: '2026-05-10'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12884
source_category: upstream-code
architectures:
- sm100
tags:
- arch_pipeline
- benchmark kernel
- blackwell
- compiler_runtime
- cute dsl
- gemm_quant
- memory_primitives
- moe_routing
techniques:
- fine-grained-quantization
- kernel-fusion
- pipeline-stages
hardware_features:
- blackwell
kernel_types:
- compiler
- gemm
- memory
- moe
- runtime
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 7ce209b4619e6dc4cc2b0778c63b03cf355c5b38
inclusion_reason: 'Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.'
changed_paths:
- tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/blockscaled_contiguous_gather_grouped_gemm_act_fusion.py
- tensorrt_llm/_torch/modules/fused_moe/create_moe.py
- tensorrt_llm/_torch/modules/fused_moe/fused_moe_cute_dsl.py
- tensorrt_llm/_torch/modules/fused_moe/quantization.py
- tests/integration/defs/accuracy/test_llm_api_pytorch.py
- tests/integration/test_lists/qa/llm_function_core.txt
- tests/integration/test_lists/test-db/l0_dgx_b200.yml
- tests/integration/test_lists/test-db/l0_gb200_multi_gpus.yml
- tests/scripts/cute_dsl_kernels/moe_as_dense_gemm/run_moe_as_dense_gemm_fc1.py
- tests/scripts/cute_dsl_kernels/run_blockscaled_contiguous_gather_grouped_gemm_act_fusion.py
- tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12884
head_ref_oid: ac6d56aa92a9052056db080bb7f88d10f116c28c
---
# [TRTLLM-11585][feat] Add CUTEDSL moe backend for nemotron-h

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12884](https://github.com/NVIDIA/TensorRT-LLM/pull/12884)
- Decision: `Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.`
- Reason: Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.

## Changed Paths

- `tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/blockscaled_contiguous_gather_grouped_gemm_act_fusion.py`
- `tensorrt_llm/_torch/modules/fused_moe/create_moe.py`
- `tensorrt_llm/_torch/modules/fused_moe/fused_moe_cute_dsl.py`
- `tensorrt_llm/_torch/modules/fused_moe/quantization.py`
- `tests/integration/defs/accuracy/test_llm_api_pytorch.py`
- `tests/integration/test_lists/qa/llm_function_core.txt`
- `tests/integration/test_lists/test-db/l0_dgx_b200.yml`
- `tests/integration/test_lists/test-db/l0_gb200_multi_gpus.yml`
- `tests/scripts/cute_dsl_kernels/moe_as_dense_gemm/run_moe_as_dense_gemm_fc1.py`
- `tests/scripts/cute_dsl_kernels/run_blockscaled_contiguous_gather_grouped_gemm_act_fusion.py`
- `tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py`

## Evidence Bundle

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