Bind TBB max threads num to OMP_NUM_THREADS environment variable value

Author: aobolensk

tasks/all/runner.cpp

@@ -1,4 +1,6 @@
+#define _CRT_SECURE_NO_WARNINGS
 #include <gtest/gtest.h>
+#include <tbb/global_control.h>
 
 #include <boost/mpi/communicator.hpp>
 #include <boost/mpi/environment.hpp>
@@ -56,6 +58,12 @@ int main(int argc, char** argv) {
   boost::mpi::environment env(argc, argv);
   boost::mpi::communicator world;
 
+  const char* omp_env = std::getenv("OMP_NUM_THREADS");
+  int num_threads = (omp_env != nullptr) ? std::atoi(omp_env) : 1;
+
+  // Limit the number of threads in TBB
+  tbb::global_control control(tbb::global_control::max_allowed_parallelism, num_threads);
+
   ::testing::InitGoogleTest(&argc, argv);
 
   auto& listeners = ::testing::UnitTest::GetInstance()->listeners();

tasks/tbb/runner.cpp

@@ -1,6 +1,14 @@
+#define _CRT_SECURE_NO_WARNINGS
 #include <gtest/gtest.h>
+#include <tbb/global_control.h>
+
+int main(int argc, char** argv) {
+  const char* omp_env = std::getenv("OMP_NUM_THREADS");
+  int num_threads = (omp_env != nullptr) ? std::atoi(omp_env) : 1;
+
+  // Limit the number of threads in TBB
+  tbb::global_control control(tbb::global_control::max_allowed_parallelism, num_threads);
 
-int main(int argc, char **argv) {
   ::testing::InitGoogleTest(&argc, argv);
   return RUN_ALL_TESTS();
 }