export functions for test

apache · Jul 22, 2020 · 95ef46c · 95ef46c
1 parent 2b5bcea
commit 95ef46c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/include/tvm/auto_scheduler/compute_dag.h b/include/tvm/auto_scheduler/compute_dag.h
@@ -137,7 +137,7 @@ class AccessAnalyzer : public ObjectRef {
    * \param producers The return producer set
    * \note This function DOES NOT propagate the relation for inlined ops
    */
-  void GetDirectProducers(
+  TVM_DLL void GetDirectProducers(
       const te::Operation& op,
       std::unordered_set<te::Operation, ObjectHash, ObjectEqual>* producers) const;
 
@@ -147,14 +147,15 @@ class AccessAnalyzer : public ObjectRef {
    * \param target_op The target operation
    * \note This function propagates the relation for chains with multiple ops.
    */
-  int GetNumCommonOuterIterator(const te::Operation& op, const te::Operation& target_op) const;
+  TVM_DLL int GetNumCommonOuterIterator(
+      const te::Operation& op, const te::Operation& target_op) const;
 
   /*!
    * \brief Return whether two operations are elementwise-matched
    *  (e.g. conv2d and relu are elementwise matched)
    * \note This function propagates the relation for chains with multiple ops.
    */
-  bool ElementWiseMatch(const te::Operation& op, const te::Operation& target_op) const;
+  TVM_DLL bool ElementWiseMatch(const te::Operation& op, const te::Operation& target_op) const;
 
   TVM_DEFINE_OBJECT_REF_METHODS(AccessAnalyzer, ObjectRef, AccessAnalyzerNode);
 };