- Prepare for implementing approx on GPU. - Unify the code path between weighted and uniform sketching on DMatrix.