Use double for GPU Hist node sum. (#7507)

2021-12-22 08:41:35 +08:00
parent eabec370e4
commit 7f399eac8b
5 changed files with 85 additions and 98 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cuh
+++ b/src/tree/gpu_hist/evaluate_splits.cuh
@@ -15,7 +15,7 @@ namespace tree {
 template <typename GradientSumT>
 struct EvaluateSplitInputs {
  int nidx;
-  GradientSumT parent_sum;
+  GradientPairPrecise parent_sum;
  GPUTrainingParam param;
  common::Span<const bst_feature_t> feature_set;
  common::Span<FeatureType const> feature_types;