---
id: pr-tensorrt-llm-12506
repo: NVIDIA/TensorRT-LLM
pr: 12506
title: '[None][feat] Add PDL support to CuTE DSL top-k kernels'
author: limin2021
date: '2026-03-25'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12506
source_category: upstream-code
architectures:
- sm100
tags:
- arch_pipeline
- attention_kv
- compiler_runtime
- cute dsl
- moe_routing
- scheduler_autotune
techniques:
- autotuning
- pipeline-stages
- tile-scheduling
hardware_features:
- blackwell
- pdl
kernel_types:
- attention
- compiler
- decode
- gemm
- moe
- runtime
- scheduler
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 2d0a0660b4854ae71a38e283a493d610102c047a
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/cute_dsl_kernels/blackwell/top_k/filtered_top_k_decode_varlen.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
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12506
head_ref_oid: 82933dc7b85045b547cae3f62cd35e2e0e28b4f0
---
# [None][feat] Add PDL support to CuTE DSL top-k kernels

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12506](https://github.com/NVIDIA/TensorRT-LLM/pull/12506)
- 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/cute_dsl_kernels/blackwell/top_k/filtered_top_k_decode_varlen.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`

## Evidence Bundle

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