Merge pull request rapidsai#395 from ChuckHastings/bug_snmg_sort_omp_…

…threads [REVIEW] run omp_get_num_threads in a parallel context
oorliu · Jul 22, 2019 · e038d75 · e038d75
2 parents b9679ee + ba9ef44
commit e038d75
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -22,6 +22,7 @@
 - PR #378 cugraph was unable to import device_of_gpu_pointer
 - PR #384 Fixed bug in snmg coo2csr causing error in dask-cugraph tests.
 - PR #382 Disabled vertex id check to allow Azure deployment
+- PR #395 run omp_ge_num_threads in a parallel context
 
 
 # cuGraph 0.8.0 (27 June 2019)

diff --git a/cpp/src/sort/sort_impl.cuh b/cpp/src/sort/sort_impl.cuh
@@ -457,8 +457,13 @@ namespace cusort {
       // Used for partitioning the output and ensuring that each GPU sorts a near equal number of elements.
       Length_t average_array_size = (keyCount + num_gpus - 1) / num_gpus;
 
-      int original_number_threads = omp_get_num_threads();
-
+      int original_number_threads = 0;
+#pragma omp parallel
+      {
+        if (omp_get_thread_num() == 0)
+          original_number_threads = omp_get_num_threads();
+      }
+
       omp_set_num_threads(num_gpus);
 
 #pragma omp parallel