Add arena initialization

oneapi-src · Jul 12, 2024 · ef0fb72 · ef0fb72
1 parent 6274d7d
commit ef0fb72
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cpp/daal/src/threading/threading.cpp b/cpp/daal/src/threading/threading.cpp
@@ -55,6 +55,7 @@ ThreaderEnvironment::~ThreaderEnvironment()
         std::cout << "delete task arena" << std::endl;
         delete reinterpret_cast<tbb::task_arena *>(_taskArena);
         _taskArena = nullptr;
+        std::cout << "after delete task arena" << std::endl;
     }
 }
 void ThreaderEnvironment::setNumberOfThreads(size_t value)
@@ -67,7 +68,9 @@ void ThreaderEnvironment::setNumberOfThreads(size_t value)
     }
     if (value > 1)
     {
-        _taskArena = reinterpret_cast<void *>(new tbb::task_arena(value));
+        tbb::task_arena* arenaPtr = new tbb::task_arena(value);
+        arenaPtr->initialize();
+        _taskArena = reinterpret_cast<void *>(arenaPtr);
     }
     _numberOfThreads = value;
 }