1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
//===-- 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/FIRDialect.h"
#include "flang/Optimizer/Dialect/FIROps.h"
#include "flang/Optimizer/Transforms/Passes.h"
#include "mlir/IR/BuiltinAttributes.h"
namespace fir {
#define GEN_PASS_DEF_ANNOTATECONSTANTOPERANDS
#include "flang/Optimizer/Transforms/Passes.h.inc"
} // namespace fir
#define DEBUG_TYPE "flang-annotate-constant"
using namespace fir;
namespace {
struct AnnotateConstantOperands
: public impl::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>();
}
|
