Modularize distributed radix sort code

Refactor the distributed radix sort code into a separate function so that it is more modular, testable, reusable. Furthermore, it fixes a small rounding error edge case in the merge code.

Signed-off-by: Matthew Cong <mcong@nvidia.com>

Author: matthewdcong