Skip to content

Commit c6167ca

Browse files
committed
fix
1 parent 5d45b90 commit c6167ca

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

src/tir/schedule/primitive/loop_transformation.cc

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,7 @@ class SubstituteVarAndCollectOpaqueBlock : public StmtExprMutator {
8686
/*! \brief Simplify the binding of block realize and update the opaque block reuse mapping */
8787
class IterMapSimplifyBlockBinding : public StmtExprMutator {
8888
public:
89-
explicit IterMapSimplifyBlockBinding(MapNode* opaque_blocks,
90-
Map<Var, Range> loop_var2extent)
89+
explicit IterMapSimplifyBlockBinding(MapNode* opaque_blocks, Map<Var, Range> loop_var2extent)
9190
: opaque_blocks_(opaque_blocks), loop_var2extent_(loop_var2extent) {}
9291

9392
static For SimplifyBindings(Stmt stmt, const Array<StmtSRef>& loop_srefs,
@@ -320,8 +319,7 @@ Array<StmtSRef> Split(ScheduleState self, const StmtSRef& loop_sref,
320319
return NullOpt;
321320
}
322321
},
323-
&opaque_block_reuse
324-
)(std::move(new_stmt));
322+
&opaque_block_reuse)(std::move(new_stmt));
325323
// Step 4. Update predicate to guard the loop
326324
new_stmt =
327325
BlockPredicateAppender(/*predicate=*/substitute_value < loop->extent, &analyzer)(new_stmt);
@@ -407,9 +405,8 @@ StmtSRef Fuse(ScheduleState self, const Array<StmtSRef>& loop_srefs) {
407405
}
408406
fused_extent = analyzer.Simplify(fused_extent);
409407
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);
413410
self->Replace(loop_srefs[0], new_stmt, opaque_block_reuse);
414411
return self->stmt2ref.at(new_stmt.get());
415412
}

0 commit comments

Comments
 (0)