[FEA]: cub::DeviceSegmentedSort with decomposer

[rootjalex]

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.

I would like to be able to run cub::DeviceSegmentedSort (and/or cub::DeviceSegmentedRadixSort) with a non-primitive type, ideally through a decomposer as in cub::DeviceRadixSort.

Describe the solution you'd like

Decomposer support added to cub::DeviceSegmentedSort and/or cub::DeviceSegmentedRadixSort. Thank you!

Describe alternatives you've considered

We've tried running DeviceRadixSort with the decomposer on our list of (i, k) tuples (the list is already sorted by i though), but this is a significant bottleneck.

Additional context

No response

[github-actions]

Hi @rootjalex!

Thanks for submitting this issue - the CCCL team has been notified and we'll get back to you as soon as we can!
In the mean time, feel free to add any relevant information to this issue.

[jrhemstad]

Thanks for the feature request @rootjalex. This is definitely something we plan to do, so hearing its important to you helps us prioritize this sooner!