MLIR  20.0.0git
AddComdats.cpp
Go to the documentation of this file.
1 //===- AddComdats.cpp - Add comdats to linkonce functions -----------------===//
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-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
11 #include "mlir/Pass/Pass.h"
12 
13 namespace mlir {
14 namespace LLVM {
15 #define GEN_PASS_DEF_LLVMADDCOMDATS
16 #include "mlir/Dialect/LLVMIR/Transforms/Passes.h.inc"
17 } // namespace LLVM
18 } // namespace mlir
19 
20 using namespace mlir;
21 
22 static void addComdat(LLVM::LLVMFuncOp &op, OpBuilder &builder,
23  SymbolTable &symbolTable, ModuleOp &module) {
24  const char *comdatName = "__llvm_comdat";
25  mlir::LLVM::ComdatOp comdatOp =
26  symbolTable.lookup<mlir::LLVM::ComdatOp>(comdatName);
27  if (!comdatOp) {
28  PatternRewriter::InsertionGuard guard(builder);
29  builder.setInsertionPointToStart(module.getBody());
30  comdatOp =
31  builder.create<mlir::LLVM::ComdatOp>(module.getLoc(), comdatName);
32  symbolTable.insert(comdatOp);
33  }
34 
35  PatternRewriter::InsertionGuard guard(builder);
36  builder.setInsertionPointToStart(&comdatOp.getBody().back());
37  auto selectorOp = builder.create<mlir::LLVM::ComdatSelectorOp>(
38  comdatOp.getLoc(), op.getSymName(), mlir::LLVM::comdat::Comdat::Any);
39  op.setComdatAttr(mlir::SymbolRefAttr::get(
40  builder.getContext(), comdatName,
41  mlir::FlatSymbolRefAttr::get(selectorOp.getSymNameAttr())));
42 }
43 
44 namespace {
45 struct AddComdatsPass : public LLVM::impl::LLVMAddComdatsBase<AddComdatsPass> {
46  void runOnOperation() override {
47  OpBuilder builder{&getContext()};
48  ModuleOp mod = getOperation();
49 
50  std::unique_ptr<SymbolTable> symbolTable;
51  auto getSymTab = [&]() -> SymbolTable & {
52  if (!symbolTable)
53  symbolTable = std::make_unique<SymbolTable>(mod);
54  return *symbolTable;
55  };
56  for (auto op : mod.getBody()->getOps<LLVM::LLVMFuncOp>()) {
57  if (op.getLinkage() == LLVM::Linkage::Linkonce ||
58  op.getLinkage() == LLVM::Linkage::LinkonceODR) {
59  addComdat(op, builder, getSymTab(), mod);
60  }
61  }
62  }
63 };
64 } // namespace
static void addComdat(LLVM::LLVMFuncOp &op, OpBuilder &builder, SymbolTable &symbolTable, ModuleOp &module)
Definition: AddComdats.cpp:22
static MLIRContext * getContext(OpFoldResult val)
MLIRContext * getContext() const
Definition: Builders.h:56
static FlatSymbolRefAttr get(StringAttr value)
Construct a symbol reference for the given value name.
This class helps build Operations.
Definition: Builders.h:216
void setInsertionPointToStart(Block *block)
Sets the insertion point to the start of the specified block.
Definition: Builders.h:440
Operation * create(const OperationState &state)
Creates an operation given the fields represented as an OperationState.
Definition: Builders.cpp:497
This class allows for representing and managing the symbol table used by operations with the 'SymbolT...
Definition: SymbolTable.h:24
Operation * lookup(StringRef name) const
Look up a symbol with the specified name, returning null if no such name exists.
StringAttr insert(Operation *symbol, Block::iterator insertPt={})
Insert a new symbol into the table, and rename it as necessary to avoid collisions.
Include the generated interface declarations.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...