fix #508

microsoft · May 13, 2017 · e9cfece · e9cfece
1 parent 5b0ae60
commit e9cfece
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/treelearner/feature_histogram.hpp b/src/treelearner/feature_histogram.hpp
@@ -374,13 +374,17 @@ class HistogramPool {
     }
     uint64_t num_total_bin = train_data->NumTotalBin();
     Log::Info("Total Bins %d", num_total_bin);
-    int old_cache_size = cache_size_;
+    int old_cache_size = static_cast<int>(pool_.size());
     Reset(cache_size, total_size);
-    pool_.resize(cache_size);
-    data_.resize(cache_size);
+
+    if (cache_size > old_cache_size) {
+      pool_.resize(cache_size);
+      data_.resize(cache_size);
+    }
+
     OMP_INIT_EX();
     #pragma omp parallel for schedule(static)
-    for (int i = old_cache_size; i < cache_size_; ++i) {
+    for (int i = old_cache_size; i < cache_size; ++i) {
       OMP_LOOP_EX_BEGIN();
       pool_[i].reset(new FeatureHistogram[train_data->num_features()]);
       data_[i].resize(num_total_bin);