vinx13 commented on a change in pull request #10538:
URL: https://github.com/apache/tvm/pull/10538#discussion_r826415315
##########
File path: include/tvm/tir/schedule/schedule.h
##########
@@ -521,6 +522,21 @@ class ScheduleNode : public runtime::Object {
*/
virtual void Unannotate(const BlockRV& block_rv, const String& ann_key) = 0;
+ /******** Schedule: Layout transformation ********/
+ /*!
+ * \brief Apply a transformation represented by IndexMap to buffer
+ * \details The indices and the access region to the target buffer is
transformed by the given
+ * index_map. The index_map is used to infer the new shape of the buffer.
Buffer must be either
+ * a function parameter, or allocated in a block (it cannot be a buffer
subregion created via
+ * 'match_buffer').
Review comment:
Good point, there are indeed consistency issue, we can make sure that
the target buffer is not used in `match_buffer`
--
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]
