Fix scan.

dmlc · May 26, 2019 · 21d12ac · 21d12ac
1 parent e7630f0
commit 21d12ac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/common/hist_util.cc b/src/common/hist_util.cc
@@ -560,7 +560,7 @@ void GHistIndexBlockMatrix::Init(const GHistIndexMatrix& gmat,
   // scan row_ptr_offsets into pointer to blocks
   std::vector<uint32_t> row_ptr_block_ptr(std::move(row_ptr_block_offsets));
   for (uint32_t i = 1; i < row_ptr_block_ptr.size(); ++i) {
-    row_ptr_block_ptr[i] = row_ptr_block_ptr[i+1];
+    row_ptr_block_ptr[i] += row_ptr_block_ptr[i-1];
   }
 
   blocks_.resize(nblock);

diff --git a/tests/benchmark/block_hist_init.cc b/tests/benchmark/block_hist_init.cc
@@ -1,7 +1,9 @@
-#include <xgboost/c_api.h>
-#include <xgboost/logging.h>
 #include <dmlc/parameter.h>
 #include <dmlc/registry.h>
+
+#include <xgboost/c_api.h>
+#include <xgboost/logging.h>
+
 #include "../../src/common/common.h"
 #include "../../src/common/column_matrix.h"
 #include "../../src/common/hist_util.h"