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 + #include 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;