Update dnnl_thread.hpp

Intel-tensorflow · Nov 25, 2024 · 209214a · 209214a
1 parent a78fe63
commit 209214a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/common/dnnl_thread.hpp b/src/common/dnnl_thread.hpp
@@ -330,6 +330,7 @@ static inline void parallel(int nthr, const std::function<void(int, int)> &f) {
 #elif DNNL_CPU_THREADING_RUNTIME == DNNL_RUNTIME_THREADPOOL
     using namespace dnnl::impl::threadpool_utils;
     dnnl::threadpool_interop::threadpool_iface *tp = get_active_threadpool();
+    dnnl::threadpool_interop::threadpool_iface *tp_null = std::nullptr_t;
     if (!tp || dnnl_in_parallel()) {
         threadpool_utils::deactivate_threadpool();
         for (int ithr = 0; ithr < nthr; ithr++) {
@@ -341,7 +342,7 @@ static inline void parallel(int nthr, const std::function<void(int, int)> &f) {
                 & dnnl::threadpool_interop::threadpool_iface::ASYNCHRONOUS;
         counting_barrier_t b;
         if (async) b.init(nthr);
-        tp->parallel_for(nthr, [&, tp](int ithr, int nthr) {
+        tp->parallel_for(nthr, [&, tp_null](int ithr, int nthr) {
             bool is_master = threadpool_utils::get_active_threadpool() == tp;
             if (!is_master) {
                 threadpool_utils::activate_threadpool(tp);