junrushao1994 commented on a change in pull request #8467:
URL: https://github.com/apache/tvm/pull/8467#discussion_r669915102
##########
File path: include/tvm/tir/schedule/schedule.h
##########
@@ -151,6 +151,18 @@ class ScheduleNode : public runtime::Object {
* \return The corresponding loop sref
*/
virtual StmtSRef GetSRef(const LoopRV& loop_rv) const = 0;
+ /*!
+ * \brief Get the block srefs corresponding to an array of BlockRVs
+ * \param block_rvs The BlockRVs to be looked up
+ * \return The corresponding block srefs
+ */
+ virtual Array<StmtSRef> GetSRefs(const Array<BlockRV>& block_rvs) const = 0;
+ /*!
+ * \brief Get the loop srefs corresponding to an array of LoopRVs
+ * \param loop_rvs The LoopRVs to be looked up
+ * \return The corresponding loop srefs
+ */
+ virtual Array<StmtSRef> GetSRefs(const Array<LoopRV>& loop_rvs) const = 0;
Review comment:
GetSRefs seems like a simple extension of GetSRef with no additional
functionalities, so let's remove these two methods from the header.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
