jinhongyii commented on code in PR #11585:
URL: https://github.com/apache/tvm/pull/11585#discussion_r889803782
##########
include/tvm/tir/schedule/schedule.h:
##########
@@ -277,19 +277,22 @@ class ScheduleNode : public runtime::Object {
* 3) All loops must start with 0.
* 4) The domain of a loop to be fused cannot depend on another loop to be
fused.
* \param loop_rvs The loops to be fused
+ * \param preserve_unit_iters Whether or not to preserve unit iterators in
block bindings
* \return The new loop after fusion
*/
- virtual LoopRV Fuse(const Array<LoopRV>& loop_rvs) = 0;
+ virtual LoopRV Fuse(const Array<LoopRV>& loop_rvs, bool preserve_unit_iters
= true) = 0;
Review Comment:
Will this change of default behavior affect unit tests? I'm not sure whether
the tests of split/fuse will fail because the bindings are not structural equal
.
--
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]
