Commit b7a74edb authored by codrut3's avatar codrut3 Committed by Jonathan Hseu
Browse files

Use cub::ReduceByKey to count partition indices (#14665) 

* Use cub::ReduceByKey to count how many times each partition index appears.

This implements a suggestion by @ekelsen. It replaces the
previously custom-made counting method and is likely more
efficient.

* Remove CubReduceAdd and use instead cub::Sum.
parent 9a6999c9
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment