jcf94 commented on a change in pull request #6184:
URL: https://github.com/apache/incubator-tvm/pull/6184#discussion_r467878505
##########
File path: include/tvm/auto_scheduler/search_policy.h
##########
@@ -89,46 +100,54 @@ class SearchCallback : public ObjectRef {
TVM_DEFINE_MUTABLE_OBJECT_REF_METHODS(SearchCallback, ObjectRef,
SearchCallbackNode);
};
+/*! \brief Attribute keys of ops used for SearchPolicy. */
+struct SearchPolicyKey {
+ /*! \brief Always apply unroll to the inner most iterator of the specificed
iterators. */
+ static constexpr const char* always_unroll_inner =
"auto_scheduler_always_unroll_inner";
+ /*! \brief The specified iterators will not be placed as the inner most
iterator. */
+ static constexpr const char* no_split_at_inner =
"auto_scheduler_no_split_at_inner";
+ /*! \brief The specified iterators will not be placed as the outter most
iterator. */
+ static constexpr const char* no_split_at_outer =
"auto_scheduler_no_split_at_outer";
Review comment:
Ok, then I'll remove them.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
