---
id: pr-tensorrt-llm-12354
repo: NVIDIA/TensorRT-LLM
pr: 12354
title: '[TRTLLM-10407][perf] Add cute dsl single pass multi cta cluster topk'
author: limin2021
date: '2026-03-23'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12354
source_category: upstream-code
architectures:
- sm100
tags:
- arch_pipeline
- benchmark kernel
- benchmark_test
- blackwell
- compiler_runtime
- cuda kernel
- cute dsl
- memory_primitives
- moe_routing
- sm100
techniques:
- pipeline-stages
hardware_features:
- blackwell
kernel_types:
- compiler
- memory
- moe
- runtime
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 4f929fef434e5920405c4c0bd7f0b2c1640daac8
inclusion_reason: 'Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.'
changed_paths:
- tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk_cluster.py
- tests/unittest/_torch/thop/parallel/test_indexer_topk.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12354
head_ref_oid: 5c73f6cf3bbc026d8e20213f6f17a4d3839d67b0
---
# [TRTLLM-10407][perf] Add cute dsl single pass multi cta cluster topk

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12354](https://github.com/NVIDIA/TensorRT-LLM/pull/12354)
- Decision: `Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.`
- Reason: Extract the architecture assumption first: SM target, TMA/WGMMA/tcgen path, cluster use, and pipeline stages.

## Changed Paths

- `tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk_cluster.py`
- `tests/unittest/_torch/thop/parallel/test_indexer_topk.py`

## Evidence Bundle

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