From 67b982c921f605c2c722c2411588539b1c8c43ae Mon Sep 17 00:00:00 2001
From: Ubuntu
 <shiyu@gbdt-shiyu.ren3kv4wanvufliwrpy4k03lsf.xx.internal.cloudapp.net>
Date: Sat, 18 Jul 2020 15:32:04 +0000
Subject: [PATCH] set num_threads of share_state (#3151)

---
 include/LightGBM/dataset.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/LightGBM/dataset.h b/include/LightGBM/dataset.h
index 802b44b9fc2e..3cf82c2aa1d3 100644
--- 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;