//===-- RemoveShapeConstraints.cpp - Remove Shape Cstr and Assuming Ops ---===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// #include "mlir/Dialect/Shape/Transforms/Passes.h" #include "mlir/Dialect/Func/IR/FuncOps.h" #include "mlir/Dialect/Shape/IR/Shape.h" #include "mlir/Transforms/DialectConversion.h" #include "mlir/Transforms/GreedyPatternRewriteDriver.h" namespace mlir { #define GEN_PASS_DEF_REMOVESHAPECONSTRAINTS #include "mlir/Dialect/Shape/Transforms/Passes.h.inc" } // namespace mlir using namespace mlir; namespace { /// Removal patterns. class RemoveCstrBroadcastableOp : public OpRewritePattern { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(shape::CstrBroadcastableOp op, PatternRewriter &rewriter) const override { rewriter.replaceOpWithNewOp(op.getOperation(), true); return success(); } }; class RemoveCstrEqOp : public OpRewritePattern { public: using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(shape::CstrEqOp op, PatternRewriter &rewriter) const override { rewriter.replaceOpWithNewOp(op.getOperation(), true); return success(); } }; /// Removal pass. class RemoveShapeConstraintsPass : public impl::RemoveShapeConstraintsBase { void runOnOperation() override { MLIRContext &ctx = getContext(); RewritePatternSet patterns(&ctx); populateRemoveShapeConstraintsPatterns(patterns); (void)applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)); } }; } // namespace void mlir::populateRemoveShapeConstraintsPatterns(RewritePatternSet &patterns) { patterns.add( patterns.getContext()); } std::unique_ptr> mlir::createRemoveShapeConstraintsPass() { return std::make_unique(); }