(tvm) branch main updated: [Runtime] Add TVM_DLL to threading backend funcs (#16630)

[yongwww]

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

yongwww 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 8194b484e7 [Runtime] Add TVM_DLL to threading backend funcs (#16630)
8194b484e7 is described below

commit 8194b484e75a33fefdbbd7851d9fbf5886dec504
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Fri Feb 23 23:43:59 2024 -0500

    [Runtime] Add TVM_DLL to threading backend funcs (#16630)
    
    This PR adds the TVM_DLL attribute to the functions
    in `threading_backend.h` to make it work with Windows
    packaging.
---
 include/tvm/runtime/threading_backend.h | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/include/tvm/runtime/threading_backend.h 
b/include/tvm/runtime/threading_backend.h
index e56c130b2c..4d09f43f95 100644
--- a/include/tvm/runtime/threading_backend.h
+++ b/include/tvm/runtime/threading_backend.h
@@ -76,14 +76,14 @@ class ThreadGroup {
    *        `worker_callback` will only be called for values >= 1. This
    *        allows use of the main thread as a worker.
    */
-  ThreadGroup(int num_workers, std::function<void(int)> worker_callback,
-              bool exclude_worker0 = false);
-  ~ThreadGroup();
+  TVM_DLL ThreadGroup(int num_workers, std::function<void(int)> 
worker_callback,
+                      bool exclude_worker0 = false);
+  TVM_DLL ~ThreadGroup();
 
   /*!
    * \brief Blocks until all non-main threads in the pool finish.
    */
-  void Join();
+  TVM_DLL void Join();
 
   enum AffinityMode : int {
     kBig = 1,
@@ -106,8 +106,8 @@ class ThreadGroup {
    *
    * \return The number of workers to use.
    */
-  int Configure(AffinityMode mode, int nthreads, bool exclude_worker0,
-                std::vector<unsigned int> cpus = {});
+  TVM_DLL int Configure(AffinityMode mode, int nthreads, bool exclude_worker0,
+                        std::vector<unsigned int> cpus = {});
 
  private:
   Impl* impl_;
@@ -116,22 +116,22 @@ class ThreadGroup {
 /*!
  * \brief Platform-agnostic no-op.
  */
-void Yield();
+TVM_DLL void Yield();
 /*!
  * \return the maximum number of effective workers for this system.
  */
-int MaxConcurrency();
+TVM_DLL int MaxConcurrency();
 /*!
  * \brief Setting the maximum number of available cores.
  */
-void SetMaxConcurrency(int value);
+TVM_DLL void SetMaxConcurrency(int value);
 /*!
  * \brief Reset the threads in the pool. All current threads are destroyed and
  * new ones are created.
  *
  * Note that this does nothing when openmp is used.
  */
-void ResetThreadPool();
+TVM_DLL void ResetThreadPool();
 
 /*!
  * \brief Configuring the CPU affinity mode for the working threads.
@@ -147,7 +147,7 @@ TVM_DLL void 
Configure(tvm::runtime::threading::ThreadGroup::AffinityMode mode,
  * \brief Get the number of threads being used by the TVM runtime
  * \returns The number of threads used.
  */
-int32_t NumThreads();
+TVM_DLL int32_t NumThreads();
 
 }  // namespace threading