---
id: pr-tensorrt-llm-12236
repo: NVIDIA/TensorRT-LLM
pr: 12236
title: '[TRTLLM-10407][perf] Enable CuteDSL indexer_top_k in model'
author: limin2021
date: '2026-03-19'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12236
source_category: upstream-code
architectures:
- sm100
tags:
- attention_kv
- benchmark_test
- compiler_runtime
- cute dsl
- paged MQA
- scheduler_autotune
- sm100
techniques:
- autotuning
- tile-scheduling
hardware_features:
- blackwell
kernel_types:
- attention
- compiler
- scheduler
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: e940e58eb92c67c5fcad069c22469f605cdaabe3
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/attention_backend/sparse/dsa.py
- tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/__init__.py
- tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk.py
- tensorrt_llm/_torch/model_config.py
- tensorrt_llm/llmapi/llm_args.py
- tests/unittest/_torch/attention/sparse/test_dsa_indexer.py
- tests/unittest/_torch/thop/parallel/test_indexer_topk.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12236
head_ref_oid: 225223ba4ddc0a096fbb7308093b95a150992894
---
# [TRTLLM-10407][perf] Enable CuteDSL indexer_top_k in model

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12236](https://github.com/NVIDIA/TensorRT-LLM/pull/12236)
- 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/attention_backend/sparse/dsa.py`
- `tensorrt_llm/_torch/custom_ops/cute_dsl_custom_ops.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/__init__.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/blackwell/top_k/single_pass_multi_cta_radix_topk.py`
- `tensorrt_llm/_torch/model_config.py`
- `tensorrt_llm/llmapi/llm_args.py`
- `tests/unittest/_torch/attention/sparse/test_dsa_indexer.py`
- `tests/unittest/_torch/thop/parallel/test_indexer_topk.py`

## Evidence Bundle

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