add sync counter to avoid error in one of tests

apache · Feb 13, 2021 · 18fdf41 · 18fdf41
1 parent 0c7303d
commit 18fdf41
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/runtime/thread_pool.cc b/src/runtime/thread_pool.cc
@@ -365,9 +365,11 @@ TVM_REGISTER_GLOBAL("runtime.config_threadpool").set_body([](TVMArgs args, TVMRe
 int TVMBackendParallelLaunch(FTVMParallelLambda flambda, void* cdata, int num_task) {
   int num_workers = tvm::runtime::threading::MaxConcurrency();
   if (num_workers == 1) {
+    std::atomic<int32_t> sync_counter{0};
     TVMParallelGroupEnv env;
     env.num_task = 1;
-    (*flambda)(1, &env, cdata);
+    env.sync_handle = &sync_counter;
+    (*flambda)(0, &env, cdata);
     return 0;
   } else {
 #if !TVM_THREADPOOL_USE_OPENMP