switching to new mgpu reduce_by_key