Skip to content

Commit 18219bf

Browse files
committed
Explain allow_reorder
1 parent cc3f0f3 commit 18219bf

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lib/Dialect/Triton/IR/Ops.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,8 @@ struct CanonicalizeReshapeReduceOpPattern final : OpRewritePattern<ReduceOp> {
529529
reshapes.begin(),
530530
[loc = reduceOp.getLoc(), &rewriter](auto pair) -> Value {
531531
auto &[value, resultType] = pair;
532+
// Set allow_reorder despite being a NOP to support different kinds of
533+
// tensor layouts.
532534
return rewriter.create<ReshapeOp>(loc, resultType, value,
533535
/*allow_reorder=*/true);
534536
});

0 commit comments

Comments
 (0)