Is your feature request related to a problem? Please describe.
There are a few places in spark-rapids-jni where we use thrust::reduce_by_key. As a part of rapidsai/cudf#20860 cudf added cudf::detail::reduce_by_key. We should look into using that instead and see if it speeds things up.
Is your feature request related to a problem? Please describe.
There are a few places in spark-rapids-jni where we use
thrust::reduce_by_key. As a part of rapidsai/cudf#20860 cudf addedcudf::detail::reduce_by_key. We should look into using that instead and see if it speeds things up.