Pass *this to in_parallel in OpenMP::impl_thread_pool_size()

sandialabs · Feb 2, 2023 · c005e60 · c005e60
1 parent 5bb7e0a
commit c005e60
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/src/OpenMP/Kokkos_OpenMP.cpp b/core/src/OpenMP/Kokkos_OpenMP.cpp
@@ -100,14 +100,15 @@ bool OpenMP::in_parallel(OpenMP const &exec_space) noexcept {
 
 int OpenMP::impl_thread_pool_size() const noexcept {
 #ifdef KOKKOS_ENABLE_DEPRECATED_CODE_3
-  return OpenMP::in_parallel()
+  return OpenMP::in_parallel(*this)
              ? omp_get_num_threads()
              : (Impl::t_openmp_instance
                     ? Impl::t_openmp_instance->m_pool_size
                     : impl_internal_space_instance()->m_pool_size);
 #else
-  return OpenMP::in_parallel() ? omp_get_num_threads()
-                               : impl_internal_space_instance()->m_pool_size;
+  return OpenMP::in_parallel(*this)
+             ? omp_get_num_threads()
+             : impl_internal_space_instance()->m_pool_size;
 #endif
 }