Create task arena on stack

uxlfoundation · Nov 19, 2024 · 119652d · 119652d
1 parent a25cb07
commit 119652d
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/cpp/daal/src/threading/threading.cpp b/cpp/daal/src/threading/threading.cpp
@@ -268,19 +268,22 @@ DAAL_EXPORT void _daal_static_threader_for(size_t n, const void * a, daal::funct
         const size_t nthreads           = _daal_threader_get_max_threads();
         const size_t nblocks_per_thread = n / nthreads + !!(n % nthreads);
 
-        tbb::parallel_for(
-            tbb::blocked_range<size_t>(0, nthreads, 1),
-            [&](tbb::blocked_range<size_t> r) {
-                const size_t tid   = r.begin();
-                const size_t begin = tid * nblocks_per_thread;
-                const size_t end   = n < begin + nblocks_per_thread ? n : begin + nblocks_per_thread;
-
-                for (size_t i = begin; i < end; ++i)
-                {
-                    func(i, tid, a);
-                }
-            },
-            tbb::static_partitioner());
+        tbb::task_arena tmp_arena(nthreads);
+        tmp_arena.execute([&] {
+            tbb::parallel_for(
+                tbb::blocked_range<size_t>(0, nthreads, 1),
+                [&](tbb::blocked_range<size_t> r) {
+                    const size_t tid   = r.begin();
+                    const size_t begin = tid * nblocks_per_thread;
+                    const size_t end   = n < begin + nblocks_per_thread ? n : begin + nblocks_per_thread;
+
+                    for (size_t i = begin; i < end; ++i)
+                    {
+                        func(i, tid, a);
+                    }
+                },
+                tbb::static_partitioner());
+        });
     }
     else
     {