Tune the small reduce_by_key case to be able to fit on sm_1x.