59 lines · cpp
1//===- FormExpressions.cpp - Form C-style expressions --------*- C++ -*-===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file implements a pass that forms EmitC operations modeling C operators10// into C-style expressions using the emitc.expression op.11//12//===----------------------------------------------------------------------===//13 14#include "mlir/Dialect/EmitC/IR/EmitC.h"15#include "mlir/Dialect/EmitC/Transforms/Passes.h"16#include "mlir/Dialect/EmitC/Transforms/Transforms.h"17#include "mlir/Transforms/GreedyPatternRewriteDriver.h"18 19namespace mlir {20namespace emitc {21#define GEN_PASS_DEF_FORMEXPRESSIONSPASS22#include "mlir/Dialect/EmitC/Transforms/Passes.h.inc"23} // namespace emitc24} // namespace mlir25 26using namespace mlir;27using namespace emitc;28 29namespace {30struct FormExpressionsPass31 : public emitc::impl::FormExpressionsPassBase<FormExpressionsPass> {32 void runOnOperation() override {33 Operation *rootOp = getOperation();34 MLIRContext *context = rootOp->getContext();35 36 // Wrap each C operator op with an expression op.37 OpBuilder builder(context);38 auto matchFun = [&](Operation *op) {39 if (isa<emitc::CExpressionInterface>(*op) &&40 !op->getParentOfType<emitc::ExpressionOp>() &&41 op->getNumResults() == 1)42 createExpression(op, builder);43 };44 rootOp->walk(matchFun);45 46 // Fold expressions where possible.47 RewritePatternSet patterns(context);48 populateExpressionPatterns(patterns);49 50 if (failed(applyPatternsGreedily(rootOp, std::move(patterns))))51 return signalPassFailure();52 }53 54 void getDependentDialects(DialectRegistry ®istry) const override {55 registry.insert<emitc::EmitCDialect>();56 }57};58} // namespace59