NVIDIA / NVIDIA/cccl

Multi-CTA `cub::DeviceBatchedTopK`: Carve out and add the `BlockPartition` building block

Open
#10,979 0 comments 0 reactions 1 assignee Claimed by @elstehle View on GitHub
Dominant language
C++
Stars
2.5k
Forks
486
Avg merge
2d 6h
Merged PRs (30d)
295

Description

The `BlockPartition` building block maps to two scenarios, as outlined in https://github.com/NVIDIA-dev/cccl_private/issues/659:
1. the `buffered` scenario where "selected" items are written to the user-provided output iterator, candidate items are written to a scratch "candidate" buffer and
2. the `last_filter` scenario with selected items being written front-to-back and the remaining candidates (tied items across the k-th boundary) being capped to not exceed the remaining "free" slots (e.g., growing back-to-front).

The issue can be closed once the PR that adds the `BlockPartition` building block has been merged.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.