From 95ef46c8eb5024bf6e79fa48585e0534a9bcdd35 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Tue, 21 Jul 2020 22:53:11 -0700 Subject: [PATCH] export functions for test --- include/tvm/auto_scheduler/compute_dag.h | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/include/tvm/auto_scheduler/compute_dag.h b/include/tvm/auto_scheduler/compute_dag.h index 3d5b272e3c99d..3cfccf4cd3521 100644 --- 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* 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); };