make ThreadPool thread local (#1152)

* make ThreadPool thread local * add return

make ThreadPool thread local (#1152)
* make ThreadPool thread local * add return
e8e84fa0 · Yida Wang · Tianqi Chen · d454f570 · e8e84fa0
Commit e8e84fa0 authored May 11, 2018 by Yida Wang Committed by Tianqi Chen May 11, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 4 deletions

src/runtime/thread_pool.cc
+3 -4

No files found.
--- a/src/runtime/thread_pool.cc
+++ b/src/runtime/thread_pool.cc
@@ -291,9 +291,8 @@ class ThreadPool {
    return res;
  }

-  static ThreadPool* Global() {
-    static ThreadPool inst;
-    return &inst;
+  static ThreadPool* ThreadLocal() {
+    return dmlc::ThreadLocalStore<ThreadPool>::Get();
  }

 private:
@@ -331,7 +330,7 @@ int TVMBackendParallelLaunch(
    FTVMParallelLambda flambda,
    void* cdata,
    int num_task) {
-  int res = tvm::runtime::ThreadPool::Global()->Launch(
+  int res = tvm::runtime::ThreadPool::ThreadLocal()->Launch(
      flambda, cdata, num_task, 1);
  return res;
 }