@@ -86,8 +86,7 @@ class SubstituteVarAndCollectOpaqueBlock : public StmtExprMutator {
86
86
/* ! \brief Simplify the binding of block realize and update the opaque block reuse mapping */
87
87
class IterMapSimplifyBlockBinding : public StmtExprMutator {
88
88
public:
89
- explicit IterMapSimplifyBlockBinding (MapNode* opaque_blocks,
90
- Map<Var, Range> loop_var2extent)
89
+ explicit IterMapSimplifyBlockBinding (MapNode* opaque_blocks, Map<Var, Range> loop_var2extent)
91
90
: opaque_blocks_(opaque_blocks), loop_var2extent_(loop_var2extent) {}
92
91
93
92
static For SimplifyBindings (Stmt stmt, const Array<StmtSRef>& loop_srefs,
@@ -320,8 +319,7 @@ Array<StmtSRef> Split(ScheduleState self, const StmtSRef& loop_sref,
320
319
return NullOpt;
321
320
}
322
321
},
323
- &opaque_block_reuse
324
- )(std::move (new_stmt));
322
+ &opaque_block_reuse)(std::move (new_stmt));
325
323
// Step 4. Update predicate to guard the loop
326
324
new_stmt =
327
325
BlockPredicateAppender (/* predicate=*/ substitute_value < loop->extent , &analyzer)(new_stmt);
@@ -407,9 +405,8 @@ StmtSRef Fuse(ScheduleState self, const Array<StmtSRef>& loop_srefs) {
407
405
}
408
406
fused_extent = analyzer.Simplify (fused_extent);
409
407
new_stmt = For (fused_var, 0 , fused_extent, ForKind::kSerial , new_stmt);
410
- new_stmt = IterMapSimplifyBlockBinding::SimplifyBindings (std::move (new_stmt), GetLoops
411
- (loop_srefs[0 ]),
412
- &opaque_block_reuse);
408
+ new_stmt = IterMapSimplifyBlockBinding::SimplifyBindings (
409
+ std::move (new_stmt), GetLoops (loop_srefs[0 ]), &opaque_block_reuse);
413
410
self->Replace (loop_srefs[0 ], new_stmt, opaque_block_reuse);
414
411
return self->stmt2ref .at (new_stmt.get ());
415
412
}
0 commit comments