NVIDIA / NVIDIA/cccl

[CUB] Expose tuning parameters for `WarpReduceBatched`

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

Description

Currently the implementation switches between WSPRO algorithm and REDUX instructions depending on `Batches` and `LogicalWarpThreads` according to our (artificial) benchmark heuristics. We could expose a minimum number of threads (currently 32) and maximum number of batches (currently 8) for which REDUX is used (if possible for the given type and reduction operator). Alternatively we could have a simple on/off/heuristic switch.

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.