forked from OSchip/llvm-project
56 lines
2.0 KiB
C++
56 lines
2.0 KiB
C++
//===-- AnnotateConstant.cpp ----------------------------------------------===//
|
|
//
|
|
// 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 "PassDetail.h"
|
|
#include "flang/Optimizer/Dialect/FIROps.h"
|
|
#include "flang/Optimizer/Transforms/Passes.h"
|
|
#include "mlir/IR/BuiltinAttributes.h"
|
|
|
|
#define DEBUG_TYPE "flang-annotate-constant"
|
|
|
|
using namespace fir;
|
|
|
|
namespace {
|
|
struct AnnotateConstantOperands
|
|
: AnnotateConstantOperandsBase<AnnotateConstantOperands> {
|
|
void runOnOperation() override {
|
|
auto *context = &getContext();
|
|
mlir::Dialect *firDialect = context->getLoadedDialect("fir");
|
|
getOperation()->walk([&](mlir::Operation *op) {
|
|
// We filter out other dialects even though they may undergo merging of
|
|
// non-equal constant values by the canonicalizer as well.
|
|
if (op->getDialect() == firDialect) {
|
|
llvm::SmallVector<mlir::Attribute> attrs;
|
|
bool hasOneOrMoreConstOpnd = false;
|
|
for (mlir::Value opnd : op->getOperands()) {
|
|
if (auto constOp = mlir::dyn_cast_or_null<mlir::arith::ConstantOp>(
|
|
opnd.getDefiningOp())) {
|
|
attrs.push_back(constOp.getValue());
|
|
hasOneOrMoreConstOpnd = true;
|
|
} else if (auto addrOp = mlir::dyn_cast_or_null<fir::AddrOfOp>(
|
|
opnd.getDefiningOp())) {
|
|
attrs.push_back(addrOp.getSymbol());
|
|
hasOneOrMoreConstOpnd = true;
|
|
} else {
|
|
attrs.push_back(mlir::UnitAttr::get(context));
|
|
}
|
|
}
|
|
if (hasOneOrMoreConstOpnd)
|
|
op->setAttr("canonicalize_constant_operands",
|
|
mlir::ArrayAttr::get(context, attrs));
|
|
}
|
|
});
|
|
}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
std::unique_ptr<mlir::Pass> fir::createAnnotateConstantOperandsPass() {
|
|
return std::make_unique<AnnotateConstantOperands>();
|
|
}
|