Fix external memory race in colmaker. (#4980)

* Move `GetColDensity` out of omp parallel block.
2019-10-25 04:11:13 -04:00
parent 96cd7ec2bb
commit 6ec7e300bd
9 changed files with 29 additions and 18 deletions
--- a/src/data/ellpack_page.cu
+++ b/src/data/ellpack_page.cu
@@ -121,7 +121,7 @@ void EllpackPageImpl::InitInfo(int device,

 // Initialize the buffer to stored compressed features.
 void EllpackPageImpl::InitCompressedData(int device, size_t num_rows) {
-  int num_symbols = matrix.info.n_bins + 1;
+  size_t num_symbols = matrix.info.n_bins + 1;

  // Required buffer size for storing data matrix in ELLPack format.
  size_t compressed_size_bytes = common::CompressedBufferWriter::CalculateBufferSize(