jcf94 commented on a change in pull request #6142:
URL: https://github.com/apache/incubator-tvm/pull/6142#discussion_r460686060
##########
File path: include/tvm/auto_scheduler/loop_state.h
##########
@@ -356,9 +356,30 @@ class State : public ObjectRef {
* \note If we do split on an iterator which has stages attached at it(by
compute_at), the inner
* most iterator of split results will become the new attach point.
*/
- TVM_DLL Array<Iterator> split(int stage_id, const Iterator& it,
- const Array<Optional<Integer>>& lengths,
- bool inner_to_outer = true);
+ Array<Iterator> split(int stage_id, const Iterator& it, const
Array<Optional<Integer>>& lengths,
+ bool inner_to_outer = true);
+ /*!
+ * \brief Schedule primitive extends to split step.
+ * \param stage_id The index of the stage to be split.
+ * \param it The iterator to be split.
+ * \param src_step_id The index of the split step to follow in the history.
+ * \param n_split The number of split level.
+ * \return The splitted new Iterators.
+ */
+ Array<Iterator> follow_split(int stage_id, const Iterator& it, int
src_step_id, int n_split);
Review comment:
Add `TVM_DLL` hint before this function, see other functions as the
reference.
----------------------------------------------------------------
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]
