Skip to content

Commit

Permalink
Bump to f27feffa3d44c834a184ed3ea4b704316fd1769e
Browse files Browse the repository at this point in the history
  • Loading branch information
yzhang93 committed Dec 31, 2024
1 parent 0b1df36 commit a93947b
Show file tree
Hide file tree
Showing 17 changed files with 20 additions and 20 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ struct RedundantLoadStoreOptimizationPass
patterns.add<SetInboundsToReadOp, SetInboundsToWriteOp>(
patterns.getContext());

(void)applyPatternsAndFoldGreedily(op, std::move(patterns));
(void)applyPatternsGreedily(op, std::move(patterns));
IRRewriter rewriter(&getContext());
vector::transferOpflowOpt(rewriter, op);
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -1042,14 +1042,14 @@ struct CanonicalizeVectorForAIEVecPass
RewritePatternSet patterns(context);
patterns.add<CanonicalizeTrivialReadAccessSubviewOpPattern,
CanonicalizeTrivialWriteAccessSubviewOpPattern>(context);
(void)applyPatternsAndFoldGreedily(op, std::move(patterns));
(void)applyPatternsGreedily(op, std::move(patterns));
}
{
// These must run before 'populateVectorBroadcastLoweringPatterns'
// so that broadcasts can be matched before conversion to insert.
RewritePatternSet patterns(context);
populateBubbleSignExtensionsLate(patterns);
(void)applyPatternsAndFoldGreedily(op, std::move(patterns));
(void)applyPatternsGreedily(op, std::move(patterns));
}

{
Expand All @@ -1067,7 +1067,7 @@ struct CanonicalizeVectorForAIEVecPass
mlir::vector::populateShapeCastFoldingPatterns(patterns);
mlir::vector::populateDropUnitDimWithShapeCastPatterns(patterns);
mlir::vector::populateVectorBroadcastLoweringPatterns(patterns);
(void)applyPatternsAndFoldGreedily(op, std::move(patterns));
(void)applyPatternsGreedily(op, std::move(patterns));
}

{
Expand All @@ -1076,7 +1076,7 @@ struct CanonicalizeVectorForAIEVecPass
// cycles.
RewritePatternSet patterns(context);
populateBubbleSignExtensionsLate(patterns);
(void)applyPatternsAndFoldGreedily(op, std::move(patterns));
(void)applyPatternsGreedily(op, std::move(patterns));
}
}
};
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@ LogicalResult assignNonLocalTiles(RewriterBase &rewriter, Operation *op,
// Find and fill the tile candidates.
RewritePatternSet fillTilePatterns(context);
fillTilePatterns.insert<FillTiles>(context, deviceModel);
if (failed(applyPatternsAndFoldGreedily(op, std::move(fillTilePatterns)))) {
if (failed(applyPatternsGreedily(op, std::move(fillTilePatterns)))) {
return op->emitOpError()
<< "collection of tile candidates for logical objectFifos failed";
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ void AMDAIECanonicalizeDoublyStridedOpPass::runOnOperation() {
}
populateCanonicalizeDoublyStridedOpPatterns(patterns, foldSingleDims,
deviceModel);
if (failed(applyPatternsAndFoldGreedily(parentOp, std::move(patterns)))) {
if (failed(applyPatternsGreedily(parentOp, std::move(patterns)))) {
parentOp->emitOpError(
"failed to canonicalize doubly strided DMA operations");
return signalPassFailure();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ void AMDAIECombineStridedOpsPass::runOnOperation() {
MLIRContext *context = &getContext();
RewritePatternSet patterns(context);
populateStridedOpCombinationPattern(patterns);
if (failed(applyPatternsAndFoldGreedily(parentOp, std::move(patterns)))) {
if (failed(applyPatternsGreedily(parentOp, std::move(patterns)))) {
parentOp->emitOpError("failed to combine strided operations");
return signalPassFailure();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -478,7 +478,7 @@ void AMDAIEDistributeCoresAndObjectFifosPass::runOnOperation() {
// possible.
RewritePatternSet unrollLocalLoopsPatterns(context);
unrollLocalLoopsPatterns.insert<AMDAIEUnrollLocalLoops>(context);
if (failed(applyPatternsAndFoldGreedily(
if (failed(applyPatternsGreedily(
moduleOp, std::move(unrollLocalLoopsPatterns)))) {
moduleOp.emitOpError()
<< "loop unrolling of loops selected for parallel execution failed";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ void AMDAIEDmaCompositionPass::runOnOperation() {
populateStridedOpCombinationPattern(patterns);
populateCanonicalizeDoublyStridedOpPatterns(patterns, false, deviceModel);

if (failed(applyPatternsAndFoldGreedily(parentOp, std::move(patterns)))) {
if (failed(applyPatternsGreedily(parentOp, std::move(patterns)))) {
parentOp->emitOpError("failed to compose strided operations");
return signalPassFailure();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -649,7 +649,7 @@ void AMDAIEDmaLoopSubsumptionPass::runOnOperation() {
onlyZeroStrideOnOuterDim);
}

if (failed(applyPatternsAndFoldGreedily(parentOp, std::move(patterns)))) {
if (failed(applyPatternsGreedily(parentOp, std::move(patterns)))) {
parentOp->emitOpError("failed to subsume some loops into DMA operations");
return signalPassFailure();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -277,7 +277,7 @@ void AMDAIELowerToUKernelsPass::runOnOperation() {
patterns.insert<LowerToUKernelPattern<linalg::FillOp>>(context, allTargets,
pathToUkernels);
if (failed(
applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)))) {
applyPatternsGreedily(getOperation(), std::move(patterns)))) {
return signalPassFailure();
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,7 @@ void AMDAIEPropagateDataLayoutPass::runOnOperation() {
patterns, [](OpOperand *opOperand) { return true; });
patterns.add<RemoveOutsDependency>(context);

if (failed(applyPatternsAndFoldGreedily(getOperation(), std::move(patterns))))
if (failed(applyPatternsGreedily(getOperation(), std::move(patterns))))
return signalPassFailure();
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ class AMDAIESinkIntoCorePass
void runOnOperation() override {
RewritePatternSet patterns(&getContext());
patterns.insert<SinkingPattern>(&getContext());
if (failed(applyPatternsAndFoldGreedily(getOperation(),
if (failed(applyPatternsGreedily(getOperation(),
std::move(patterns)))) {
signalPassFailure();
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ void AMDAIEVectorizationPass::runOnOperation() {
vector::populateVectorTransferPermutationMapLoweringPatterns(
vectorizationPatterns);

(void)applyPatternsAndFoldGreedily(funcOp, std::move(vectorizationPatterns));
(void)applyPatternsGreedily(funcOp, std::move(vectorizationPatterns));
}
} // namespace

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ void AMDAIEBridgeToAIRPass::runOnOperation() {
.insert<LinalgCopyToMemRefCopy, SCFForAllToParallelOp, AffineApplyOnSym>(
context);
if (failed(
applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)))) {
applyPatternsGreedily(getOperation(), std::move(patterns)))) {
return signalPassFailure();
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ void AMDAIECleanupPass::runOnOperation() {

RewritePatternSet patterns(context);
populateCleanupPatterns(patterns);
if (failed(applyPatternsAndFoldGreedily(funcOp, std::move(patterns)))) {
if (failed(applyPatternsGreedily(funcOp, std::move(patterns)))) {
return signalPassFailure();
}
{
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -317,7 +317,7 @@ void AMDAIEDecomposeLinalgExtPackUnPackToAIRPass::runOnOperation() {
RewritePatternSet patterns(ctx);
patterns.add<LowerPackPattern, LowerUnPackPattern>(ctx);
if (failed(
applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)))) {
applyPatternsGreedily(getOperation(), std::move(patterns)))) {
return signalPassFailure();
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -804,7 +804,7 @@ LogicalResult initAIELaunchConfig(FunctionOpInterface funcOp,
// Resolve those away.
RewritePatternSet patterns(funcOp.getContext());
memref::populateResolveRankedShapedTypeResultDimsPatterns(patterns);
return applyPatternsAndFoldGreedily(funcOp, std::move(patterns));
return applyPatternsGreedily(funcOp, std::move(patterns));
}

} // namespace mlir::iree_compiler::AMDAIE
2 changes: 1 addition & 1 deletion third_party/iree
Submodule iree updated 176 files

0 comments on commit a93947b

Please sign in to comment.