diff --git a/include/mxnet/base.h b/include/mxnet/base.h index 1ef9c6bf8450..1eeffc1ab4b9 100644 --- a/include/mxnet/base.h +++ b/include/mxnet/base.h @@ -49,6 +49,19 @@ #endif #endif +/*! +* \brief define dllexport for Visual Studio +*/ +#ifdef _MSC_VER +#ifdef MXNET_EXPORTS +#define MXNET_API __declspec(dllexport) +#else +#define MXNET_API __declspec(dllimport) +#endif +#else +#define MXNET_API +#endif + /*! \brief namespace of mxnet */ namespace mxnet { /*! \brief mxnet cpu */ diff --git a/include/mxnet/engine.h b/include/mxnet/engine.h index 03eb45b54de0..195f5c05eb20 100644 --- a/include/mxnet/engine.h +++ b/include/mxnet/engine.h @@ -45,7 +45,7 @@ enum class FnProperty { /*! * \brief Dependency engine that schedules operations. */ -class Engine { +class MXNET_API Engine { public: /*! * \brief OnComplete Callback to the engine, diff --git a/include/mxnet/storage.h b/include/mxnet/storage.h index da7a8aaa5388..60bca03b0680 100644 --- a/include/mxnet/storage.h +++ b/include/mxnet/storage.h @@ -14,7 +14,7 @@ namespace mxnet { /*! * \brief Storage manager across multiple devices. */ -class Storage { +class MXNET_API Storage { public: /*! * \brief Storage handle.