junrushao1994 commented on a change in pull request #9871:
URL: https://github.com/apache/tvm/pull/9871#discussion_r780588169
##########
File path: include/tvm/tir/schedule/schedule.h
##########
@@ -465,6 +465,25 @@ class ScheduleNode : public runtime::Object {
*/
virtual void SetScope(const BlockRV& block_rv, int buffer_index, const
String& storage_scope) = 0;
/******** Schedule: Blockize & Tensorize ********/
+ /*!
+ * \brief Convert the subtree rooted at a specific loop into a block.
+ * \param loop_rv the root of the subtree
+ * \return the new block
+ */
+ virtual BlockRV Blockize(const LoopRV& loop_rv) = 0;
+ /*!
+ * \brief Tensorize the computation enclosed by loop with tensor_intrin
+ * \param loop_rv the loop to be tensorized
+ * \param intrin the tensor intrinsic
+ */
+ virtual void Tensorize(const LoopRV& loop_rv, const TensorIntrin& intrin) =
0;
Review comment:
Shall we hide this method to internal impl given:
- TracedSchedule is the default go-to
- This method isn't well supported by TracedSchedule
--
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]
