Remove omp_get_max_threads in data. (#7588)

2022-01-24 02:44:07 +08:00
parent f84291c1e1
commit 5817840858
18 changed files with 97 additions and 92 deletions
--- a/tests/cpp/data/test_data.cc
+++ b/tests/cpp/data/test_data.cc
@@ -1,3 +1,6 @@
+/*!
+ * Copyright 2019-2022 by XGBoost Contributors
+ */
 #include <gtest/gtest.h>
 #include <dmlc/filesystem.h>
 #include <fstream>
@@ -66,7 +69,7 @@ TEST(SparsePage, PushCSCAfterTranspose) {
  SparsePage page; // Consolidated sparse page
  for (const auto &batch : dmat->GetBatches<xgboost::SparsePage>()) {
    // Transpose each batch and push
-    SparsePage tmp = batch.GetTranspose(ncols);
+    SparsePage tmp = batch.GetTranspose(ncols, common::OmpGetNumThreads(0));
    page.PushCSC(tmp);
  }