Skip to content

Commit 45f1dbb

Browse files
committed
Explain allow_reorder
1 parent cc3f0f3 commit 45f1dbb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

lib/Dialect/Triton/IR/Ops.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,7 @@ 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 to support different tensor layouts.
532533
return rewriter.create<ReshapeOp>(loc, resultType, value,
533534
/*allow_reorder=*/true);
534535
});

0 commit comments

Comments
 (0)