OpenMp检测并行区域之前嵌套并行性中的线程数

如何在并行区域启动之前检测OpenMp中的线程数？ 如果我使用嵌套并行性，则环境变量OMP_NUM_THREADS看起来像4,64 。

 get_nested_num_threads(); #pragma omp parallel { // starting 4 threads #pragma omp parallel { // starting 64 threads for each of the 4 } } 

这个答案导致我使用以下代码查询OMP_NUM_THREADS ：

 #include  #include  int get_nested_num_threads(){ char delimiter[] = ","; char *ptr = NULL; char *num_threads = NULL; num_threads = getenv("OMP_NUM_THREADS"); int threads=1, nested=0; ptr = strtok(num_threads, delimiter); while ( ptr != NULL ){ threads *= atoi(ptr); ptr = strtok(NULL,delimiter); nested += 1; } assert( nested <= 2 ); return threads; } 

不幸的是，如果我调用getenv("OMP_NUM_THREADS")那么我会观察到4,4而不是4,64的嵌套并行性。 这对我来说真的很奇怪。 你对此有解释吗？

Answer 1

我已经解决了它，打开一个嵌套的并行区域来查询所有线程：

 int get_nested_num_threads(){ int threads=1; #pragma omp parallel shared(threads) { #pragma omp single { threads = omp_get_num_threads(); #pragma omp parallel shared(threads) { #pragma omp single { threads *= omp_get_num_threads(); } } } } return threads; } 

据我所知，在这种情况下，您不需要在C中使用firstprivate和lastprivate 。 但是你必须在Fortran中做到这一点。