Use thrust functions instead of custom functions (#5544)

2020-04-16 21:41:16 +12:00
parent 6a169cd41a
commit e268fb0093
6 changed files with 82 additions and 306 deletions
--- a/src/data/ellpack_page.cu
+++ b/src/data/ellpack_page.cu
@@ -49,7 +49,9 @@ __global__ void CompressBinEllpackKernel(
    int ncuts = cut_rows[feature + 1] - cut_rows[feature];
    // Assigning the bin in current entry.
    // S.t.: fvalue < feature_cuts[bin]
-    bin = dh::UpperBound(feature_cuts, ncuts, fvalue);
+    bin = thrust::upper_bound(thrust::seq, feature_cuts, feature_cuts + ncuts,
+                              fvalue) -
+          feature_cuts;
    if (bin >= ncuts) {
      bin = ncuts - 1;
    }