[CUB]: Gather requirements and prepare high level design for device resident problem size for cub::DeviceRadixSort

This issue can be closed with a list of design alternatives we considered with pros and cons for each, e.g., performance, temporary storage size, interface changes.

Note: this applies to the onesweep kernel implementation only. We can have older architectures use the onesweep in this case.