set num_threads of share_state (microsoft#3151)

shiyu1994 · Jul 18, 2020 · 67b982c · 67b982c
1 parent f5f27ca
commit 67b982c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/include/LightGBM/dataset.h b/include/LightGBM/dataset.h
@@ -294,11 +294,11 @@ struct TrainingShareStates {
       hist_buf;
 
   void SetMultiValBin(MultiValBin* bin) {
+    num_threads = OMP_NUM_THREADS();
     if (bin == nullptr) {
       return;
     }
     multi_val_bin.reset(bin);
-    num_threads = OMP_NUM_THREADS();
     num_bin_aligned =
         (bin->num_bin() + kAlignedSize - 1) / kAlignedSize * kAlignedSize;
     size_t new_size = static_cast<size_t>(num_bin_aligned) * 2 * num_threads;