diff --git a/include/tvm/runtime/disco/session.h b/include/tvm/runtime/disco/session.h index af29f74aaa..fee0c0ef30 100644 --- a/include/tvm/runtime/disco/session.h +++ b/include/tvm/runtime/disco/session.h @@ -237,7 +237,7 @@ class SessionObj : public Object { * \brief Managed reference to SessionObj * \sa SessionObj */ -class Session : public ObjectRef { +class TVM_DLL Session : public ObjectRef { public: /*! \brief Create a session backed by a thread pool of workers */ static TVM_DLL Session ThreadedSession(int num_workers);