Set thread.

dmlc · May 6, 2020 · a1358e8 · a1358e8
1 parent 11f2a8f
commit a1358e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/cpp/data/test_sparse_page_dmatrix.cc b/tests/cpp/data/test_sparse_page_dmatrix.cc
@@ -94,7 +94,7 @@ TEST(SparsePageDMatrix, ThreadSafetyException) {
 
   bool exception = false;
   int threads = 1000;
-#pragma omp parallel for
+#pragma omp parallel for num_threads(threads)
   for (auto i = 0; i < threads; i++) {
     try {
       auto iter = dmat->GetBatches<SparsePage>().begin();