Unverified Commit 87bd799e by Tianqi Chen Committed by GitHub

[RUNTIME] Move module export to the function level. (#4405)

parent e0810512
...@@ -111,15 +111,15 @@ class Module : public ObjectRef { ...@@ -111,15 +111,15 @@ class Module : public ObjectRef {
* *
* \endcode * \endcode
*/ */
class TVM_DLL ModuleNode : public Object { class ModuleNode : public Object {
public: public:
/*! \brief virtual destructor */ /*! \brief virtual destructor */
virtual ~ModuleNode() {} TVM_DLL virtual ~ModuleNode() {}
/*! /*!
* \return The per module type key. * \return The per module type key.
* \note This key is used to for serializing custom modules. * \note This key is used to for serializing custom modules.
*/ */
virtual const char* type_key() const = 0; TVM_DLL virtual const char* type_key() const = 0;
/*! /*!
* \brief Get a PackedFunc from module. * \brief Get a PackedFunc from module.
* *
...@@ -137,7 +137,7 @@ class TVM_DLL ModuleNode : public Object { ...@@ -137,7 +137,7 @@ class TVM_DLL ModuleNode : public Object {
* If the function need resource from the module(e.g. late linking), * If the function need resource from the module(e.g. late linking),
* it should capture sptr_to_self. * it should capture sptr_to_self.
*/ */
virtual PackedFunc GetFunction( TVM_DLL virtual PackedFunc GetFunction(
const std::string& name, const std::string& name,
const ObjectPtr<Object>& sptr_to_self) = 0; const ObjectPtr<Object>& sptr_to_self) = 0;
/*! /*!
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment