---
id: pr-cccl-cub-7928
repo: NVIDIA/cccl
pr: 7928
title: Implement the new tuning API for `DispatchTopK`
author: bernhardmgruber
date: '2026-03-09'
url: https://github.com/NVIDIA/cccl/pull/7928
source_category: upstream-code
architectures:
- sm100
tags:
- benchmark_test
- memory_primitives
- moe_routing
- scheduler_autotune
- topk
techniques:
- autotuning
- tile-scheduling
hardware_features: []
kernel_types:
- memory
- moe
- scheduler
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: b28dc8eb34bcb926e278d72a07c6bb1f512fafde
inclusion_reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.
changed_paths:
- cub/benchmarks/bench/topk/keys.cu
- cub/benchmarks/bench/topk/pairs.cu
- cub/cub/agent/agent_topk.cuh
- cub/cub/device/device_topk.cuh
- cub/cub/device/dispatch/dispatch_topk.cuh
- cub/cub/device/dispatch/tuning/tuning_topk.cuh
- cub/test/catch2_test_device_topk_keys.cu
- cub/test/catch2_test_device_topk_pairs.cu
artifact_dir: evidence/pull-bundles/cccl-cub/gh-7928
head_ref_oid: 2e22b6defec2647f2c7959b3b6ae4deb8f47c93a
---
# Implement the new tuning API for `DispatchTopK`

- Repository: `NVIDIA/cccl`
- PR: [7928](https://github.com/NVIDIA/cccl/pull/7928)
- Decision: `Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.`
- Reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.

## Changed Paths

- `cub/benchmarks/bench/topk/keys.cu`
- `cub/benchmarks/bench/topk/pairs.cu`
- `cub/cub/agent/agent_topk.cuh`
- `cub/cub/device/device_topk.cuh`
- `cub/cub/device/dispatch/dispatch_topk.cuh`
- `cub/cub/device/dispatch/tuning/tuning_topk.cuh`
- `cub/test/catch2_test_device_topk_keys.cu`
- `cub/test/catch2_test_device_topk_pairs.cu`

## Evidence Bundle

`evidence/pull-bundles/cccl-cub/gh-7928`
