nverke commented on code in PR #13721:
URL: https://github.com/apache/tvm/pull/13721#discussion_r1110191250
##########
include/tvm/meta_schedule/schedule_rule.h:
##########
@@ -210,6 +210,31 @@ class ScheduleRule : public runtime::ObjectRef {
Optional<Array<Integer>> vector_load_lens, Optional<Map<String,
ObjectRef>> reuse_read,
Optional<Map<String, ObjectRef>> reuse_write, bool
use_software_pipeline);
+ /*!
+ * \brief Extension of MultiLevelTiling for auto-tensorization with multiple
groups of candidate
+ * tensor core intrinsics
+ * \param intrin_groups A list of groups of tensor core intrinsics. The map
should contain key
+ * "compute" which represents the tensor intrin for computation. The value
of the map should be
+ * names of tensor intrinsics, must be registered via
+ * TensorIntrin.register(...) beforehand
+ * \param structure The tiling structure. Recommended:
+ * - 'SRSRS' on Hexagon
Review Comment:
This should be applicable outside of schedules that have input of NCHWc but
are you suggesting adding that to help people understand?
--
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]
