Enhance nvtx support. (#5636)

2020-05-06 22:54:24 +08:00
parent 67d267f9da
commit eaf2a00b5c
13 changed files with 98 additions and 87 deletions
--- 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();