Integer gradient summation for GPU histogram algorithm. (#2681)

2017-09-08 15:07:29 +12:00
parent 15267eedf2
commit e6a9063344
15 changed files with 182 additions and 128 deletions
--- a/src/common/hist_util.h
+++ b/src/common/hist_util.h
@@ -33,8 +33,8 @@ struct GHistEntry {

  /*! \brief add a bst_gpair to the sum */
  inline void Add(const bst_gpair& e) {
-    sum_grad += e.grad;
-    sum_hess += e.hess;
+    sum_grad += e.GetGrad();
+    sum_hess += e.GetHess();
  }

  /*! \brief add a GHistEntry to the sum */