[CUB] Expose tuning parameters for `WarpReduceBatched`
Open
- 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
Assessment
This issue has not been assessed yet.