[FEA]: WarpExchange (shuffle version) where ITEMS_PER_THREAD != LOGICAL_WARP_THREADS

[el1goluj]

Is this a duplicate?

• I confirmed there appear to be no duplicate issues for this request and that I agree to the Code of Conduct

Area

CUB

Is your feature request related to a problem? Please describe.

Currently, the shuffle version of WarpExchange only compiles if ITEMS_PER_THREAD == LOGICAL_WARP_THREADS. It would be desirable to have a version where there is no restriction on ITEMS_PER_THREAD (at least if ITEMS_PER_THREAD <= 32).

Describe the solution you'd like

A version of WarpExchange that uses shuffle and allows ITEMS_PER_THREAD != LOGICAL_WARP_THREADS. If it is for ITEMS_PER_THREAD <= LOGICAL_WARP_THREADS, it would be good.

Describe alternatives you've considered

No response

Additional context

No response