Use single precision in gain calculation, use pointers instead of span. (#8051)

2022-07-12 21:56:27 +02:00
parent a5bc8e2c6a
commit 0bdaca25ca
3 changed files with 19 additions and 21 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cuh
+++ b/src/tree/gpu_hist/evaluate_splits.cuh
@@ -68,7 +68,7 @@ class GPUHistEvaluator {
  // storage for sorted index of feature histogram, used for sort based splits.
  dh::device_vector<bst_feature_t> cat_sorted_idx_;
  // cached input for sorting the histogram, used for sort based splits.
-  using SortPair = thrust::tuple<uint32_t, double>;
+  using SortPair = thrust::tuple<uint32_t, float>;
  dh::device_vector<SortPair> sort_input_;
  // cache for feature index
  dh::device_vector<bst_feature_t> feature_idx_;