[tvm] branch main updated: [WIN] export void Configure(...) symbol for Windows (#12091)

[junrushao]

This is an automated email from the ASF dual-hosted git repository.

junrushao pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new de3c0f4b30 [WIN] export void Configure(...) symbol for Windows (#12091)
de3c0f4b30 is described below

commit de3c0f4b30b33eb54e851f94d518ca3349423cfa
Author: Valery Chernov <black.che...@gmail.com>
AuthorDate: Thu Jul 14 20:07:10 2022 +0300

    [WIN] export void Configure(...) symbol for Windows (#12091)
    
    Extends TVM API for clients of Windows. It helps configure thread pool on 
native side on Windows OS.
    
    Co-authored-by: Valery Chernov <valery.cher...@deelvin.com>
---
 include/tvm/runtime/threading_backend.h | 4 ++--
 src/runtime/thread_pool.cc              | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/include/tvm/runtime/threading_backend.h 
b/include/tvm/runtime/threading_backend.h
index c23ba835fc..77d6730c09 100644
--- a/include/tvm/runtime/threading_backend.h
+++ b/include/tvm/runtime/threading_backend.h
@@ -137,8 +137,8 @@ void ResetThreadPool();
  * \param nthreads The number of threads to use (0 = use all).
  * \param cpus A list of CPUs is used to set the 'cpu affinity' for the worker 
threads.
  */
-void Configure(tvm::runtime::threading::ThreadGroup::AffinityMode mode, int 
nthreads,
-               std::vector<unsigned int> cpus);
+TVM_DLL void Configure(tvm::runtime::threading::ThreadGroup::AffinityMode 
mode, int nthreads,
+                       std::vector<unsigned int> cpus);
 
 /*!
  * \brief Get the number of threads being used by the TVM runtime
diff --git a/src/runtime/thread_pool.cc b/src/runtime/thread_pool.cc
index 4692e06734..6dc87cda11 100644
--- a/src/runtime/thread_pool.cc
+++ b/src/runtime/thread_pool.cc
@@ -458,8 +458,8 @@ void ResetThreadPool() { 
tvm::runtime::ThreadPool::ThreadLocal()->Reset(); }
  * \param cpus cpus A list of CPUs is used to set the 'cpu affinity' for the 
worker threads.
  *
  */
-void Configure(tvm::runtime::threading::ThreadGroup::AffinityMode mode, int 
nthreads,
-               std::vector<unsigned int> cpus) {
+TVM_DLL void Configure(tvm::runtime::threading::ThreadGroup::AffinityMode 
mode, int nthreads,
+                       std::vector<unsigned int> cpus) {
   tvm::runtime::threading::SetMaxConcurrency(cpus.size());
 #if !TVM_THREADPOOL_USE_OPENMP
   tvm::runtime::ThreadPool::ThreadLocal()->UpdateWorkerConfiguration(mode, 
nthreads, cpus);