(tvm) branch main updated: [Runtime] Add "TVM_DLL" to NVTX header (#16809)

[wuwei]

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

wuwei 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 8ee8d0d0b8 [Runtime] Add "TVM_DLL" to NVTX header (#16809)
8ee8d0d0b8 is described below

commit 8ee8d0d0b8dbf3e77a0d67afbecc4274de6af642
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Thu Mar 28 20:22:14 2024 -0400

    [Runtime] Add "TVM_DLL" to NVTX header (#16809)
    
    This PR adds the `TVM_DLL` attribute to the nvtx header for windows
    build.
---
 include/tvm/runtime/nvtx.h | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/include/tvm/runtime/nvtx.h b/include/tvm/runtime/nvtx.h
index 17f8f4f28a..db99154b0b 100644
--- a/include/tvm/runtime/nvtx.h
+++ b/include/tvm/runtime/nvtx.h
@@ -19,6 +19,8 @@
 #ifndef TVM_RUNTIME_NVTX_H_
 #define TVM_RUNTIME_NVTX_H_
 
+#include <tvm/runtime/c_runtime_api.h>
+
 #include <string>
 namespace tvm {
 namespace runtime {
@@ -29,11 +31,11 @@ namespace runtime {
 class NVTXScopedRange {
  public:
   /*! \brief Enter an NVTX scoped range */
-  explicit NVTXScopedRange(const char* name);
+  TVM_DLL explicit NVTXScopedRange(const char* name);
   /*! \brief Enter an NVTX scoped range */
   explicit NVTXScopedRange(const std::string& name) : 
NVTXScopedRange(name.c_str()) {}
   /*! \brief Exist an NVTX scoped range */
-  ~NVTXScopedRange();
+  TVM_DLL ~NVTXScopedRange();
   NVTXScopedRange(const NVTXScopedRange& other) = delete;
   NVTXScopedRange(NVTXScopedRange&& other) = delete;
   NVTXScopedRange& operator=(const NVTXScopedRange& other) = delete;