Remove weird set thread.

dmlc · Nov 8, 2021 · 58f62e8 · 58f62e8
1 parent a9f36fd
commit 58f62e8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/common/threading_utils.h b/src/common/threading_utils.h
@@ -130,8 +130,7 @@ class BlockedSpace2d {
 template <typename Func>
 void ParallelFor2d(const BlockedSpace2d& space, int nthreads, Func func) {
   const size_t num_blocks_in_space = space.Size();
-  nthreads = std::min(nthreads, omp_get_max_threads());
-  nthreads = std::max(nthreads, 1);
+  CHECK_GE(nthreads, 1);
 
   dmlc::OMPException exc;
 #pragma omp parallel num_threads(nthreads)

diff --git a/tests/cpp/tree/test_approx.cc b/tests/cpp/tree/test_approx.cc
@@ -78,8 +78,9 @@ TEST(Approx, PredictionCache) {
   auto Xy = RandomDataGenerator{n_samples, n_features, 0}.GenerateDMatrix(true);
 
   {
+    omp_set_num_threads(1);
     GenericParameter ctx;
-    ctx.InitAllowUnknown(Args{});
+    ctx.InitAllowUnknown(Args{{"nthread", "8"}});
     std::unique_ptr<TreeUpdater> approx{
         TreeUpdater::Create("grow_global_approx_histmaker", &ctx, ObjInfo{ObjInfo::kRegression})};
     RegTree tree;