| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 1 | //===- GlobalMerge.cpp - Internal globals merging -------------------------===// |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 2 | // |
| Chandler Carruth | 2946cd7 | 2019-01-19 08:50:56 +0000 | [diff] [blame] | 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 |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 6 | // |
| 7 | //===----------------------------------------------------------------------===// |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 8 | // |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 9 | // This pass merges globals with internal linkage into one. This way all the |
| 10 | // globals which were merged into a biggest one can be addressed using offsets |
| 11 | // from the same base pointer (no need for separate base pointer for each of the |
| 12 | // global). Such a transformation can significantly reduce the register pressure |
| 13 | // when many globals are involved. |
| 14 | // |
| Nadav Rotem | 465834c | 2012-07-24 10:51:42 +0000 | [diff] [blame] | 15 | // For example, consider the code which touches several global variables at |
| Eric Christopher | bf86fd3 | 2010-09-28 04:18:29 +0000 | [diff] [blame] | 16 | // once: |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 17 | // |
| 18 | // static int foo[N], bar[N], baz[N]; |
| 19 | // |
| 20 | // for (i = 0; i < N; ++i) { |
| 21 | // foo[i] = bar[i] * baz[i]; |
| 22 | // } |
| 23 | // |
| 24 | // On ARM the addresses of 3 arrays should be kept in the registers, thus |
| 25 | // this code has quite large register pressure (loop body): |
| 26 | // |
| 27 | // ldr r1, [r5], #4 |
| 28 | // ldr r2, [r6], #4 |
| 29 | // mul r1, r2, r1 |
| 30 | // str r1, [r0], #4 |
| 31 | // |
| 32 | // Pass converts the code to something like: |
| 33 | // |
| 34 | // static struct { |
| 35 | // int foo[N]; |
| 36 | // int bar[N]; |
| 37 | // int baz[N]; |
| 38 | // } merged; |
| 39 | // |
| 40 | // for (i = 0; i < N; ++i) { |
| 41 | // merged.foo[i] = merged.bar[i] * merged.baz[i]; |
| 42 | // } |
| 43 | // |
| 44 | // and in ARM code this becomes: |
| 45 | // |
| 46 | // ldr r0, [r5, #40] |
| 47 | // ldr r1, [r5, #80] |
| 48 | // mul r0, r1, r0 |
| 49 | // str r0, [r5], #4 |
| 50 | // |
| 51 | // note that we saved 2 registers here almostly "for free". |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 52 | // |
| 53 | // However, merging globals can have tradeoffs: |
| 54 | // - it confuses debuggers, tools, and users |
| 55 | // - it makes linker optimizations less useful (order files, LOHs, ...) |
| 56 | // - it forces usage of indexed addressing (which isn't necessarily "free") |
| 57 | // - it can increase register pressure when the uses are disparate enough. |
| Fangrui Song | f78650a | 2018-07-30 19:41:25 +0000 | [diff] [blame] | 58 | // |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 59 | // We use heuristics to discover the best global grouping we can (cf cl::opts). |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 60 | // |
| Eric Christopher | bf86fd3 | 2010-09-28 04:18:29 +0000 | [diff] [blame] | 61 | // ===---------------------------------------------------------------------===// |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 62 | |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 63 | #include "llvm/ADT/BitVector.h" |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 64 | #include "llvm/ADT/DenseMap.h" |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 65 | #include "llvm/ADT/SmallPtrSet.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 66 | #include "llvm/ADT/SmallVector.h" |
| Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 67 | #include "llvm/ADT/Statistic.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 68 | #include "llvm/ADT/StringRef.h" |
| 69 | #include "llvm/ADT/Triple.h" |
| 70 | #include "llvm/ADT/Twine.h" |
| Chandler Carruth | d990388 | 2015-01-14 11:23:27 +0000 | [diff] [blame] | 71 | #include "llvm/CodeGen/Passes.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 72 | #include "llvm/IR/BasicBlock.h" |
| Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 73 | #include "llvm/IR/Constants.h" |
| 74 | #include "llvm/IR/DataLayout.h" |
| 75 | #include "llvm/IR/DerivedTypes.h" |
| 76 | #include "llvm/IR/Function.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 77 | #include "llvm/IR/GlobalAlias.h" |
| 78 | #include "llvm/IR/GlobalValue.h" |
| Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 79 | #include "llvm/IR/GlobalVariable.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 80 | #include "llvm/IR/Instruction.h" |
| Chandler Carruth | 9fb823b | 2013-01-02 11:36:10 +0000 | [diff] [blame] | 81 | #include "llvm/IR/Module.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 82 | #include "llvm/IR/Type.h" |
| 83 | #include "llvm/IR/Use.h" |
| 84 | #include "llvm/IR/User.h" |
| Reid Kleckner | 05da2fe | 2019-11-13 13:15:01 -0800 | [diff] [blame] | 85 | #include "llvm/InitializePasses.h" |
| Simon Pilgrim | cdafe59 | 2020-05-18 16:56:53 +0100 | [diff] [blame] | 86 | #include "llvm/MC/SectionKind.h" |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 87 | #include "llvm/Pass.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 88 | #include "llvm/Support/Casting.h" |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 89 | #include "llvm/Support/CommandLine.h" |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 90 | #include "llvm/Support/Debug.h" |
| 91 | #include "llvm/Support/raw_ostream.h" |
| David Blaikie | 6054e65 | 2018-03-23 23:58:19 +0000 | [diff] [blame] | 92 | #include "llvm/Target/TargetLoweringObjectFile.h" |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 93 | #include "llvm/Target/TargetMachine.h" |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 94 | #include <algorithm> |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 95 | #include <cassert> |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 96 | #include <cstddef> |
| David Blaikie | b3bde2e | 2017-11-17 01:07:10 +0000 | [diff] [blame] | 97 | #include <cstdint> |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 98 | #include <string> |
| 99 | #include <vector> |
| 100 | |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 101 | using namespace llvm; |
| 102 | |
| Chandler Carruth | 964daaa | 2014-04-22 02:55:47 +0000 | [diff] [blame] | 103 | #define DEBUG_TYPE "global-merge" |
| 104 | |
| Ahmed Bougacha | b96444e | 2015-04-11 00:06:36 +0000 | [diff] [blame] | 105 | // FIXME: This is only useful as a last-resort way to disable the pass. |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 106 | static cl::opt<bool> |
| Jiangning Liu | 3e5b855 | 2014-06-11 06:35:26 +0000 | [diff] [blame] | 107 | EnableGlobalMerge("enable-global-merge", cl::Hidden, |
| Ahmed Bougacha | b96444e | 2015-04-11 00:06:36 +0000 | [diff] [blame] | 108 | cl::desc("Enable the global merge pass"), |
| Tim Northover | f804c17 | 2014-02-18 11:17:29 +0000 | [diff] [blame] | 109 | cl::init(true)); |
| 110 | |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 111 | static cl::opt<unsigned> |
| 112 | GlobalMergeMaxOffset("global-merge-max-offset", cl::Hidden, |
| 113 | cl::desc("Set maximum offset for global merge pass"), |
| 114 | cl::init(0)); |
| 115 | |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 116 | static cl::opt<bool> GlobalMergeGroupByUse( |
| 117 | "global-merge-group-by-use", cl::Hidden, |
| 118 | cl::desc("Improve global merge pass to look at uses"), cl::init(true)); |
| 119 | |
| 120 | static cl::opt<bool> GlobalMergeIgnoreSingleUse( |
| 121 | "global-merge-ignore-single-use", cl::Hidden, |
| 122 | cl::desc("Improve global merge pass to ignore globals only used alone"), |
| 123 | cl::init(true)); |
| 124 | |
| Tim Northover | f804c17 | 2014-02-18 11:17:29 +0000 | [diff] [blame] | 125 | static cl::opt<bool> |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 126 | EnableGlobalMergeOnConst("global-merge-on-const", cl::Hidden, |
| Jakub Staszak | 6b36db0 | 2013-07-22 21:11:30 +0000 | [diff] [blame] | 127 | cl::desc("Enable global merge pass on constants"), |
| 128 | cl::init(false)); |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 129 | |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 130 | // FIXME: this could be a transitional option, and we probably need to remove |
| 131 | // it if only we are sure this optimization could always benefit all targets. |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 132 | static cl::opt<cl::boolOrDefault> |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 133 | EnableGlobalMergeOnExternal("global-merge-on-external", cl::Hidden, |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 134 | cl::desc("Enable global merge pass on external linkage")); |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 135 | |
| Eric Christopher | ed47b22 | 2015-02-23 19:28:45 +0000 | [diff] [blame] | 136 | STATISTIC(NumMerged, "Number of globals merged"); |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 137 | |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 138 | namespace { |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 139 | |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 140 | class GlobalMerge : public FunctionPass { |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 141 | const TargetMachine *TM = nullptr; |
| 142 | |
| Eric Christopher | ed47b22 | 2015-02-23 19:28:45 +0000 | [diff] [blame] | 143 | // FIXME: Infer the maximum possible offset depending on the actual users |
| 144 | // (these max offsets are different for the users inside Thumb or ARM |
| 145 | // functions), see the code that passes in the offset in the ARM backend |
| 146 | // for more information. |
| 147 | unsigned MaxOffset; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 148 | |
| Ahmed Bougacha | 8207641 | 2015-06-04 20:39:23 +0000 | [diff] [blame] | 149 | /// Whether we should try to optimize for size only. |
| 150 | /// Currently, this applies a dead simple heuristic: only consider globals |
| 151 | /// used in minsize functions for merging. |
| 152 | /// FIXME: This could learn about optsize, and be used in the cost model. |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 153 | bool OnlyOptimizeForSize = false; |
| Ahmed Bougacha | 8207641 | 2015-06-04 20:39:23 +0000 | [diff] [blame] | 154 | |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 155 | /// Whether we should merge global variables that have external linkage. |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 156 | bool MergeExternalGlobals = false; |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 157 | |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 158 | bool IsMachO; |
| 159 | |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 160 | bool doMerge(SmallVectorImpl<GlobalVariable*> &Globals, |
| Silviu Baranga | a055aab | 2013-01-07 12:31:25 +0000 | [diff] [blame] | 161 | Module &M, bool isConst, unsigned AddrSpace) const; |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 162 | |
| Adrian Prantl | 5f8f34e4 | 2018-05-01 15:54:18 +0000 | [diff] [blame] | 163 | /// Merge everything in \p Globals for which the corresponding bit |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 164 | /// in \p GlobalSet is set. |
| David Blaikie | 47bf5c0 | 2015-08-21 22:19:06 +0000 | [diff] [blame] | 165 | bool doMerge(const SmallVectorImpl<GlobalVariable *> &Globals, |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 166 | const BitVector &GlobalSet, Module &M, bool isConst, |
| 167 | unsigned AddrSpace) const; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 168 | |
| Adrian Prantl | 5f8f34e4 | 2018-05-01 15:54:18 +0000 | [diff] [blame] | 169 | /// Check if the given variable has been identified as must keep |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 170 | /// \pre setMustKeepGlobalVariables must have been called on the Module that |
| 171 | /// contains GV |
| 172 | bool isMustKeepGlobalVariable(const GlobalVariable *GV) const { |
| 173 | return MustKeepGlobalVariables.count(GV); |
| 174 | } |
| 175 | |
| 176 | /// Collect every variables marked as "used" or used in a landing pad |
| 177 | /// instruction for this Module. |
| 178 | void setMustKeepGlobalVariables(Module &M); |
| 179 | |
| 180 | /// Collect every variables marked as "used" |
| Eli Friedman | d6baff6 | 2018-07-25 22:03:35 +0000 | [diff] [blame] | 181 | void collectUsedGlobalVariables(Module &M, StringRef Name); |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 182 | |
| Quentin Colombet | 2393cb9 | 2013-03-19 21:46:49 +0000 | [diff] [blame] | 183 | /// Keep track of the GlobalVariable that must not be merged away |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 184 | SmallPtrSet<const GlobalVariable *, 16> MustKeepGlobalVariables; |
| 185 | |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 186 | public: |
| 187 | static char ID; // Pass identification, replacement for typeid. |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 188 | |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 189 | explicit GlobalMerge() |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 190 | : FunctionPass(ID), MaxOffset(GlobalMergeMaxOffset) { |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 191 | initializeGlobalMergePass(*PassRegistry::getPassRegistry()); |
| 192 | } |
| 193 | |
| 194 | explicit GlobalMerge(const TargetMachine *TM, unsigned MaximalOffset, |
| 195 | bool OnlyOptimizeForSize, bool MergeExternalGlobals) |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 196 | : FunctionPass(ID), TM(TM), MaxOffset(MaximalOffset), |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 197 | OnlyOptimizeForSize(OnlyOptimizeForSize), |
| 198 | MergeExternalGlobals(MergeExternalGlobals) { |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 199 | initializeGlobalMergePass(*PassRegistry::getPassRegistry()); |
| 200 | } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 201 | |
| Craig Topper | 3e4c697 | 2014-03-05 09:10:37 +0000 | [diff] [blame] | 202 | bool doInitialization(Module &M) override; |
| 203 | bool runOnFunction(Function &F) override; |
| 204 | bool doFinalization(Module &M) override; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 205 | |
| Mehdi Amini | 117296c | 2016-10-01 02:56:57 +0000 | [diff] [blame] | 206 | StringRef getPassName() const override { return "Merge internal globals"; } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 207 | |
| Craig Topper | 3e4c697 | 2014-03-05 09:10:37 +0000 | [diff] [blame] | 208 | void getAnalysisUsage(AnalysisUsage &AU) const override { |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 209 | AU.setPreservesCFG(); |
| 210 | FunctionPass::getAnalysisUsage(AU); |
| 211 | } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 212 | }; |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 213 | |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 214 | } // end anonymous namespace |
| 215 | |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 216 | char GlobalMerge::ID = 0; |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 217 | |
| Matthias Braun | 1527baa | 2017-05-25 21:26:32 +0000 | [diff] [blame] | 218 | INITIALIZE_PASS(GlobalMerge, DEBUG_TYPE, "Merge global variables", false, false) |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 219 | |
| 220 | bool GlobalMerge::doMerge(SmallVectorImpl<GlobalVariable*> &Globals, |
| Silviu Baranga | a055aab | 2013-01-07 12:31:25 +0000 | [diff] [blame] | 221 | Module &M, bool isConst, unsigned AddrSpace) const { |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 222 | auto &DL = M.getDataLayout(); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 223 | // FIXME: Find better heuristics |
| Fangrui Song | efd94c5 | 2019-04-23 14:51:27 +0000 | [diff] [blame] | 224 | llvm::stable_sort( |
| 225 | Globals, [&DL](const GlobalVariable *GV1, const GlobalVariable *GV2) { |
| 226 | return DL.getTypeAllocSize(GV1->getValueType()) < |
| 227 | DL.getTypeAllocSize(GV2->getValueType()); |
| 228 | }); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 229 | |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 230 | // If we want to just blindly group all globals together, do so. |
| 231 | if (!GlobalMergeGroupByUse) { |
| 232 | BitVector AllGlobals(Globals.size()); |
| 233 | AllGlobals.set(); |
| 234 | return doMerge(Globals, AllGlobals, M, isConst, AddrSpace); |
| 235 | } |
| 236 | |
| 237 | // If we want to be smarter, look at all uses of each global, to try to |
| 238 | // discover all sets of globals used together, and how many times each of |
| Benjamin Kramer | df005cb | 2015-08-08 18:27:36 +0000 | [diff] [blame] | 239 | // these sets occurred. |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 240 | // |
| 241 | // Keep this reasonably efficient, by having an append-only list of all sets |
| 242 | // discovered so far (UsedGlobalSet), and mapping each "together-ness" unit of |
| 243 | // code (currently, a Function) to the set of globals seen so far that are |
| 244 | // used together in that unit (GlobalUsesByFunction). |
| 245 | // |
| Haicheng Wu | b09308d | 2018-04-26 17:56:50 +0000 | [diff] [blame] | 246 | // When we look at the Nth global, we know that any new set is either: |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 247 | // - the singleton set {N}, containing this global only, or |
| 248 | // - the union of {N} and a previously-discovered set, containing some |
| 249 | // combination of the previous N-1 globals. |
| 250 | // Using that knowledge, when looking at the Nth global, we can keep: |
| 251 | // - a reference to the singleton set {N} (CurGVOnlySetIdx) |
| 252 | // - a list mapping each previous set to its union with {N} (EncounteredUGS), |
| 253 | // if it actually occurs. |
| 254 | |
| 255 | // We keep track of the sets of globals used together "close enough". |
| 256 | struct UsedGlobalSet { |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 257 | BitVector Globals; |
| Eugene Zelenko | f193332 | 2017-09-22 23:46:57 +0000 | [diff] [blame] | 258 | unsigned UsageCount = 1; |
| 259 | |
| 260 | UsedGlobalSet(size_t Size) : Globals(Size) {} |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 261 | }; |
| 262 | |
| 263 | // Each set is unique in UsedGlobalSets. |
| 264 | std::vector<UsedGlobalSet> UsedGlobalSets; |
| 265 | |
| 266 | // Avoid repeating the create-global-set pattern. |
| 267 | auto CreateGlobalSet = [&]() -> UsedGlobalSet & { |
| 268 | UsedGlobalSets.emplace_back(Globals.size()); |
| 269 | return UsedGlobalSets.back(); |
| 270 | }; |
| 271 | |
| 272 | // The first set is the empty set. |
| 273 | CreateGlobalSet().UsageCount = 0; |
| 274 | |
| 275 | // We define "close enough" to be "in the same function". |
| 276 | // FIXME: Grouping uses by function is way too aggressive, so we should have |
| 277 | // a better metric for distance between uses. |
| 278 | // The obvious alternative would be to group by BasicBlock, but that's in |
| 279 | // turn too conservative.. |
| 280 | // Anything in between wouldn't be trivial to compute, so just stick with |
| 281 | // per-function grouping. |
| 282 | |
| 283 | // The value type is an index into UsedGlobalSets. |
| 284 | // The default (0) conveniently points to the empty set. |
| 285 | DenseMap<Function *, size_t /*UsedGlobalSetIdx*/> GlobalUsesByFunction; |
| 286 | |
| 287 | // Now, look at each merge-eligible global in turn. |
| 288 | |
| 289 | // Keep track of the sets we already encountered to which we added the |
| 290 | // current global. |
| 291 | // Each element matches the same-index element in UsedGlobalSets. |
| 292 | // This lets us efficiently tell whether a set has already been expanded to |
| 293 | // include the current global. |
| 294 | std::vector<size_t> EncounteredUGS; |
| 295 | |
| 296 | for (size_t GI = 0, GE = Globals.size(); GI != GE; ++GI) { |
| 297 | GlobalVariable *GV = Globals[GI]; |
| 298 | |
| 299 | // Reset the encountered sets for this global... |
| 300 | std::fill(EncounteredUGS.begin(), EncounteredUGS.end(), 0); |
| 301 | // ...and grow it in case we created new sets for the previous global. |
| 302 | EncounteredUGS.resize(UsedGlobalSets.size()); |
| 303 | |
| 304 | // We might need to create a set that only consists of the current global. |
| 305 | // Keep track of its index into UsedGlobalSets. |
| 306 | size_t CurGVOnlySetIdx = 0; |
| 307 | |
| 308 | // For each global, look at all its Uses. |
| 309 | for (auto &U : GV->uses()) { |
| 310 | // This Use might be a ConstantExpr. We're interested in Instruction |
| 311 | // users, so look through ConstantExpr... |
| 312 | Use *UI, *UE; |
| 313 | if (ConstantExpr *CE = dyn_cast<ConstantExpr>(U.getUser())) { |
| Oliver Stannard | 8379e29 | 2015-06-08 16:55:31 +0000 | [diff] [blame] | 314 | if (CE->use_empty()) |
| 315 | continue; |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 316 | UI = &*CE->use_begin(); |
| 317 | UE = nullptr; |
| 318 | } else if (isa<Instruction>(U.getUser())) { |
| 319 | UI = &U; |
| 320 | UE = UI->getNext(); |
| 321 | } else { |
| 322 | continue; |
| 323 | } |
| 324 | |
| 325 | // ...to iterate on all the instruction users of the global. |
| 326 | // Note that we iterate on Uses and not on Users to be able to getNext(). |
| 327 | for (; UI != UE; UI = UI->getNext()) { |
| 328 | Instruction *I = dyn_cast<Instruction>(UI->getUser()); |
| 329 | if (!I) |
| 330 | continue; |
| 331 | |
| 332 | Function *ParentFn = I->getParent()->getParent(); |
| Ahmed Bougacha | 8207641 | 2015-06-04 20:39:23 +0000 | [diff] [blame] | 333 | |
| 334 | // If we're only optimizing for size, ignore non-minsize functions. |
| Evandro Menezes | 85bd397 | 2019-04-04 22:40:06 +0000 | [diff] [blame] | 335 | if (OnlyOptimizeForSize && !ParentFn->hasMinSize()) |
| Ahmed Bougacha | 8207641 | 2015-06-04 20:39:23 +0000 | [diff] [blame] | 336 | continue; |
| 337 | |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 338 | size_t UGSIdx = GlobalUsesByFunction[ParentFn]; |
| 339 | |
| 340 | // If this is the first global the basic block uses, map it to the set |
| 341 | // consisting of this global only. |
| 342 | if (!UGSIdx) { |
| 343 | // If that set doesn't exist yet, create it. |
| 344 | if (!CurGVOnlySetIdx) { |
| 345 | CurGVOnlySetIdx = UsedGlobalSets.size(); |
| 346 | CreateGlobalSet().Globals.set(GI); |
| 347 | } else { |
| 348 | ++UsedGlobalSets[CurGVOnlySetIdx].UsageCount; |
| 349 | } |
| 350 | |
| 351 | GlobalUsesByFunction[ParentFn] = CurGVOnlySetIdx; |
| 352 | continue; |
| 353 | } |
| 354 | |
| 355 | // If we already encountered this BB, just increment the counter. |
| 356 | if (UsedGlobalSets[UGSIdx].Globals.test(GI)) { |
| 357 | ++UsedGlobalSets[UGSIdx].UsageCount; |
| 358 | continue; |
| 359 | } |
| 360 | |
| 361 | // If not, the previous set wasn't actually used in this function. |
| 362 | --UsedGlobalSets[UGSIdx].UsageCount; |
| 363 | |
| 364 | // If we already expanded the previous set to include this global, just |
| 365 | // reuse that expanded set. |
| 366 | if (size_t ExpandedIdx = EncounteredUGS[UGSIdx]) { |
| 367 | ++UsedGlobalSets[ExpandedIdx].UsageCount; |
| 368 | GlobalUsesByFunction[ParentFn] = ExpandedIdx; |
| 369 | continue; |
| 370 | } |
| 371 | |
| 372 | // If not, create a new set consisting of the union of the previous set |
| 373 | // and this global. Mark it as encountered, so we can reuse it later. |
| 374 | GlobalUsesByFunction[ParentFn] = EncounteredUGS[UGSIdx] = |
| 375 | UsedGlobalSets.size(); |
| 376 | |
| 377 | UsedGlobalSet &NewUGS = CreateGlobalSet(); |
| 378 | NewUGS.Globals.set(GI); |
| 379 | NewUGS.Globals |= UsedGlobalSets[UGSIdx].Globals; |
| 380 | } |
| 381 | } |
| 382 | } |
| 383 | |
| 384 | // Now we found a bunch of sets of globals used together. We accumulated |
| 385 | // the number of times we encountered the sets (i.e., the number of blocks |
| 386 | // that use that exact set of globals). |
| 387 | // |
| 388 | // Multiply that by the size of the set to give us a crude profitability |
| 389 | // metric. |
| Fangrui Song | efd94c5 | 2019-04-23 14:51:27 +0000 | [diff] [blame] | 390 | llvm::stable_sort(UsedGlobalSets, |
| 391 | [](const UsedGlobalSet &UGS1, const UsedGlobalSet &UGS2) { |
| 392 | return UGS1.Globals.count() * UGS1.UsageCount < |
| 393 | UGS2.Globals.count() * UGS2.UsageCount; |
| 394 | }); |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 395 | |
| 396 | // We can choose to merge all globals together, but ignore globals never used |
| 397 | // with another global. This catches the obviously non-profitable cases of |
| 398 | // having a single global, but is aggressive enough for any other case. |
| 399 | if (GlobalMergeIgnoreSingleUse) { |
| 400 | BitVector AllGlobals(Globals.size()); |
| 401 | for (size_t i = 0, e = UsedGlobalSets.size(); i != e; ++i) { |
| 402 | const UsedGlobalSet &UGS = UsedGlobalSets[e - i - 1]; |
| 403 | if (UGS.UsageCount == 0) |
| 404 | continue; |
| 405 | if (UGS.Globals.count() > 1) |
| 406 | AllGlobals |= UGS.Globals; |
| 407 | } |
| 408 | return doMerge(Globals, AllGlobals, M, isConst, AddrSpace); |
| 409 | } |
| 410 | |
| 411 | // Starting from the sets with the best (=biggest) profitability, find a |
| 412 | // good combination. |
| 413 | // The ideal (and expensive) solution can only be found by trying all |
| 414 | // combinations, looking for the one with the best profitability. |
| 415 | // Don't be smart about it, and just pick the first compatible combination, |
| 416 | // starting with the sets with the best profitability. |
| 417 | BitVector PickedGlobals(Globals.size()); |
| 418 | bool Changed = false; |
| 419 | |
| 420 | for (size_t i = 0, e = UsedGlobalSets.size(); i != e; ++i) { |
| 421 | const UsedGlobalSet &UGS = UsedGlobalSets[e - i - 1]; |
| 422 | if (UGS.UsageCount == 0) |
| 423 | continue; |
| 424 | if (PickedGlobals.anyCommon(UGS.Globals)) |
| 425 | continue; |
| 426 | PickedGlobals |= UGS.Globals; |
| 427 | // If the set only contains one global, there's no point in merging. |
| 428 | // Ignore the global for inclusion in other sets though, so keep it in |
| 429 | // PickedGlobals. |
| 430 | if (UGS.Globals.count() < 2) |
| 431 | continue; |
| 432 | Changed |= doMerge(Globals, UGS.Globals, M, isConst, AddrSpace); |
| 433 | } |
| 434 | |
| 435 | return Changed; |
| 436 | } |
| 437 | |
| David Blaikie | 47bf5c0 | 2015-08-21 22:19:06 +0000 | [diff] [blame] | 438 | bool GlobalMerge::doMerge(const SmallVectorImpl<GlobalVariable *> &Globals, |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 439 | const BitVector &GlobalSet, Module &M, bool isConst, |
| 440 | unsigned AddrSpace) const { |
| David Blaikie | 47bf5c0 | 2015-08-21 22:19:06 +0000 | [diff] [blame] | 441 | assert(Globals.size() > 1); |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 442 | |
| Chris Lattner | 229907c | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 443 | Type *Int32Ty = Type::getInt32Ty(M.getContext()); |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 444 | Type *Int8Ty = Type::getInt8Ty(M.getContext()); |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 445 | auto &DL = M.getDataLayout(); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 446 | |
| Nicola Zaghen | d34e60c | 2018-05-14 12:53:11 +0000 | [diff] [blame] | 447 | LLVM_DEBUG(dbgs() << " Trying to merge set, starts with #" |
| 448 | << GlobalSet.find_first() << "\n"); |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 449 | |
| Haicheng Wu | 69ba061 | 2018-05-19 18:00:02 +0000 | [diff] [blame] | 450 | bool Changed = false; |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 451 | ssize_t i = GlobalSet.find_first(); |
| 452 | while (i != -1) { |
| 453 | ssize_t j = 0; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 454 | uint64_t MergedSize = 0; |
| Jay Foad | b804a2b | 2011-07-12 14:06:48 +0000 | [diff] [blame] | 455 | std::vector<Type*> Tys; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 456 | std::vector<Constant*> Inits; |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 457 | std::vector<unsigned> StructIdxs; |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 458 | |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 459 | bool HasExternal = false; |
| Adrian Prantl | 622bddb | 2016-11-11 22:09:25 +0000 | [diff] [blame] | 460 | StringRef FirstExternalName; |
| Guillaume Chatelet | 0e62011 | 2019-10-15 11:24:36 +0000 | [diff] [blame] | 461 | Align MaxAlign; |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 462 | unsigned CurIdx = 0; |
| Ahmed Bougacha | 279e3ee | 2015-04-18 01:21:58 +0000 | [diff] [blame] | 463 | for (j = i; j != -1; j = GlobalSet.find_next(j)) { |
| David Blaikie | 9ed57a9 | 2015-08-21 22:00:44 +0000 | [diff] [blame] | 464 | Type *Ty = Globals[j]->getValueType(); |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 465 | |
| Eli Friedman | 3769639 | 2018-08-29 23:46:26 +0000 | [diff] [blame] | 466 | // Make sure we use the same alignment AsmPrinter would use. |
| Guillaume Chatelet | 368a5e3 | 2020-06-29 11:24:36 +0000 | [diff] [blame] | 467 | Align Alignment = DL.getPreferredAlign(Globals[j]); |
| Guillaume Chatelet | 0e62011 | 2019-10-15 11:24:36 +0000 | [diff] [blame] | 468 | unsigned Padding = alignTo(MergedSize, Alignment) - MergedSize; |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 469 | MergedSize += Padding; |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 470 | MergedSize += DL.getTypeAllocSize(Ty); |
| Bob Wilson | 4c8ab19 | 2010-11-17 21:25:36 +0000 | [diff] [blame] | 471 | if (MergedSize > MaxOffset) { |
| 472 | break; |
| 473 | } |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 474 | if (Padding) { |
| 475 | Tys.push_back(ArrayType::get(Int8Ty, Padding)); |
| 476 | Inits.push_back(ConstantAggregateZero::get(Tys.back())); |
| 477 | ++CurIdx; |
| 478 | } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 479 | Tys.push_back(Ty); |
| 480 | Inits.push_back(Globals[j]->getInitializer()); |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 481 | StructIdxs.push_back(CurIdx++); |
| 482 | |
| Guillaume Chatelet | 0e62011 | 2019-10-15 11:24:36 +0000 | [diff] [blame] | 483 | MaxAlign = std::max(MaxAlign, Alignment); |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 484 | |
| 485 | if (Globals[j]->hasExternalLinkage() && !HasExternal) { |
| 486 | HasExternal = true; |
| Adrian Prantl | 622bddb | 2016-11-11 22:09:25 +0000 | [diff] [blame] | 487 | FirstExternalName = Globals[j]->getName(); |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 488 | } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 489 | } |
| 490 | |
| Haicheng Wu | 69ba061 | 2018-05-19 18:00:02 +0000 | [diff] [blame] | 491 | // Exit early if there is only one global to merge. |
| 492 | if (Tys.size() < 2) { |
| 493 | i = j; |
| 494 | continue; |
| 495 | } |
| 496 | |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 497 | // If merged variables doesn't have external linkage, we needn't to expose |
| 498 | // the symbol after merging. |
| 499 | GlobalValue::LinkageTypes Linkage = HasExternal |
| 500 | ? GlobalValue::ExternalLinkage |
| 501 | : GlobalValue::InternalLinkage; |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 502 | // Use a packed struct so we can control alignment. |
| 503 | StructType *MergedTy = StructType::get(M.getContext(), Tys, true); |
| Chris Lattner | e40007a | 2010-09-05 21:18:45 +0000 | [diff] [blame] | 504 | Constant *MergedInit = ConstantStruct::get(MergedTy, Inits); |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 505 | |
| Adrian Prantl | 6cb849e | 2016-11-11 21:48:09 +0000 | [diff] [blame] | 506 | // On Darwin external linkage needs to be preserved, otherwise |
| 507 | // dsymutil cannot preserve the debug info for the merged |
| 508 | // variables. If they have external linkage, use the symbol name |
| 509 | // of the first variable merged as the suffix of global symbol |
| 510 | // name. This avoids a link-time naming conflict for the |
| 511 | // _MergedGlobals symbols. |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 512 | Twine MergedName = |
| 513 | (IsMachO && HasExternal) |
| Adrian Prantl | 622bddb | 2016-11-11 22:09:25 +0000 | [diff] [blame] | 514 | ? "_MergedGlobals_" + FirstExternalName |
| Adrian Prantl | 554fd99 | 2016-11-11 17:50:09 +0000 | [diff] [blame] | 515 | : "_MergedGlobals"; |
| 516 | auto MergedLinkage = IsMachO ? Linkage : GlobalValue::PrivateLinkage; |
| 517 | auto *MergedGV = new GlobalVariable( |
| 518 | M, MergedTy, isConst, MergedLinkage, MergedInit, MergedName, nullptr, |
| 519 | GlobalVariable::NotThreadLocal, AddrSpace); |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 520 | |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 521 | MergedGV->setAlignment(MaxAlign); |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 522 | MergedGV->setSection(Globals[i]->getSection()); |
| Peter Collingbourne | d4135bb | 2016-09-13 01:12:59 +0000 | [diff] [blame] | 523 | |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 524 | const StructLayout *MergedLayout = DL.getStructLayout(MergedTy); |
| David Blaikie | 6614d8d | 2015-09-14 20:29:26 +0000 | [diff] [blame] | 525 | for (ssize_t k = i, idx = 0; k != j; k = GlobalSet.find_next(k), ++idx) { |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 526 | GlobalValue::LinkageTypes Linkage = Globals[k]->getLinkage(); |
| Benjamin Kramer | adcd026 | 2020-01-28 20:23:46 +0100 | [diff] [blame] | 527 | std::string Name(Globals[k]->getName()); |
| Michael Spang | a2fb2c0 | 2020-01-28 12:43:07 -0800 | [diff] [blame] | 528 | GlobalValue::VisibilityTypes Visibility = Globals[k]->getVisibility(); |
| Martin Storsjo | 9ca8b57 | 2018-02-12 21:14:21 +0000 | [diff] [blame] | 529 | GlobalValue::DLLStorageClassTypes DLLStorage = |
| 530 | Globals[k]->getDLLStorageClass(); |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 531 | |
| Peter Collingbourne | d4135bb | 2016-09-13 01:12:59 +0000 | [diff] [blame] | 532 | // Copy metadata while adjusting any debug info metadata by the original |
| 533 | // global's offset within the merged global. |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 534 | MergedGV->copyMetadata(Globals[k], |
| 535 | MergedLayout->getElementOffset(StructIdxs[idx])); |
| Peter Collingbourne | d4135bb | 2016-09-13 01:12:59 +0000 | [diff] [blame] | 536 | |
| Chris Lattner | e40007a | 2010-09-05 21:18:45 +0000 | [diff] [blame] | 537 | Constant *Idx[2] = { |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 538 | ConstantInt::get(Int32Ty, 0), |
| 539 | ConstantInt::get(Int32Ty, StructIdxs[idx]), |
| Chris Lattner | e40007a | 2010-09-05 21:18:45 +0000 | [diff] [blame] | 540 | }; |
| David Blaikie | 4a2e73b | 2015-04-02 18:55:32 +0000 | [diff] [blame] | 541 | Constant *GEP = |
| 542 | ConstantExpr::getInBoundsGetElementPtr(MergedTy, MergedGV, Idx); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 543 | Globals[k]->replaceAllUsesWith(GEP); |
| 544 | Globals[k]->eraseFromParent(); |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 545 | |
| John Brawn | 0bef27d | 2015-08-12 13:36:48 +0000 | [diff] [blame] | 546 | // When the linkage is not internal we must emit an alias for the original |
| 547 | // variable name as it may be accessed from another object. On non-Mach-O |
| 548 | // we can also emit an alias for internal linkage as it's safe to do so. |
| 549 | // It's not safe on Mach-O as the alias (and thus the portion of the |
| 550 | // MergedGlobals variable) may be dead stripped at link time. |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 551 | if (Linkage != GlobalValue::InternalLinkage || !IsMachO) { |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 552 | GlobalAlias *GA = GlobalAlias::create(Tys[StructIdxs[idx]], AddrSpace, |
| 553 | Linkage, Name, GEP, &M); |
| Michael Spang | a2fb2c0 | 2020-01-28 12:43:07 -0800 | [diff] [blame] | 554 | GA->setVisibility(Visibility); |
| Martin Storsjo | 9ca8b57 | 2018-02-12 21:14:21 +0000 | [diff] [blame] | 555 | GA->setDLLStorageClass(DLLStorage); |
| John Brawn | 0bef27d | 2015-08-12 13:36:48 +0000 | [diff] [blame] | 556 | } |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 557 | |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 558 | NumMerged++; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 559 | } |
| Haicheng Wu | 69ba061 | 2018-05-19 18:00:02 +0000 | [diff] [blame] | 560 | Changed = true; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 561 | i = j; |
| 562 | } |
| 563 | |
| Haicheng Wu | 69ba061 | 2018-05-19 18:00:02 +0000 | [diff] [blame] | 564 | return Changed; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 565 | } |
| 566 | |
| Eli Friedman | d6baff6 | 2018-07-25 22:03:35 +0000 | [diff] [blame] | 567 | void GlobalMerge::collectUsedGlobalVariables(Module &M, StringRef Name) { |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 568 | // Extract global variables from llvm.used array |
| Eli Friedman | d6baff6 | 2018-07-25 22:03:35 +0000 | [diff] [blame] | 569 | const GlobalVariable *GV = M.getGlobalVariable(Name); |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 570 | if (!GV || !GV->hasInitializer()) return; |
| 571 | |
| 572 | // Should be an array of 'i8*'. |
| Rafael Espindola | 74f2e46 | 2013-04-22 14:58:02 +0000 | [diff] [blame] | 573 | const ConstantArray *InitList = cast<ConstantArray>(GV->getInitializer()); |
| 574 | |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 575 | for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) |
| 576 | if (const GlobalVariable *G = |
| 577 | dyn_cast<GlobalVariable>(InitList->getOperand(i)->stripPointerCasts())) |
| 578 | MustKeepGlobalVariables.insert(G); |
| 579 | } |
| 580 | |
| 581 | void GlobalMerge::setMustKeepGlobalVariables(Module &M) { |
| Eli Friedman | d6baff6 | 2018-07-25 22:03:35 +0000 | [diff] [blame] | 582 | collectUsedGlobalVariables(M, "llvm.used"); |
| 583 | collectUsedGlobalVariables(M, "llvm.compiler.used"); |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 584 | |
| Reid Kleckner | f8d1d12 | 2016-10-19 19:56:22 +0000 | [diff] [blame] | 585 | for (Function &F : M) { |
| 586 | for (BasicBlock &BB : F) { |
| 587 | Instruction *Pad = BB.getFirstNonPHI(); |
| 588 | if (!Pad->isEHPad()) |
| 589 | continue; |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 590 | |
| Reid Kleckner | f8d1d12 | 2016-10-19 19:56:22 +0000 | [diff] [blame] | 591 | // Keep globals used by landingpads and catchpads. |
| 592 | for (const Use &U : Pad->operands()) { |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 593 | if (const GlobalVariable *GV = |
| Reid Kleckner | f8d1d12 | 2016-10-19 19:56:22 +0000 | [diff] [blame] | 594 | dyn_cast<GlobalVariable>(U->stripPointerCasts())) |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 595 | MustKeepGlobalVariables.insert(GV); |
| Reid Kleckner | f8d1d12 | 2016-10-19 19:56:22 +0000 | [diff] [blame] | 596 | } |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 597 | } |
| 598 | } |
| 599 | } |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 600 | |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 601 | bool GlobalMerge::doInitialization(Module &M) { |
| Tim Northover | f804c17 | 2014-02-18 11:17:29 +0000 | [diff] [blame] | 602 | if (!EnableGlobalMerge) |
| 603 | return false; |
| 604 | |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 605 | IsMachO = Triple(M.getTargetTriple()).isOSBinFormatMachO(); |
| 606 | |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 607 | auto &DL = M.getDataLayout(); |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 608 | DenseMap<std::pair<unsigned, StringRef>, SmallVector<GlobalVariable *, 16>> |
| 609 | Globals, ConstGlobals, BSSGlobals; |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 610 | bool Changed = false; |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 611 | setMustKeepGlobalVariables(M); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 612 | |
| 613 | // Grab all non-const globals. |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 614 | for (auto &GV : M.globals()) { |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 615 | // Merge is safe for "normal" internal or external globals only |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 616 | if (GV.isDeclaration() || GV.isThreadLocal() || GV.hasImplicitSection()) |
| Jiangning Liu | b2ae37f | 2014-06-11 06:44:53 +0000 | [diff] [blame] | 617 | continue; |
| 618 | |
| John Brawn | 6671616 | 2017-06-02 10:24:14 +0000 | [diff] [blame] | 619 | // It's not safe to merge globals that may be preempted |
| 620 | if (TM && !TM->shouldAssumeDSOLocal(M, &GV)) |
| 621 | continue; |
| 622 | |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 623 | if (!(MergeExternalGlobals && GV.hasExternalLinkage()) && |
| 624 | !GV.hasInternalLinkage()) |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 625 | continue; |
| 626 | |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 627 | PointerType *PT = dyn_cast<PointerType>(GV.getType()); |
| Silviu Baranga | a055aab | 2013-01-07 12:31:25 +0000 | [diff] [blame] | 628 | assert(PT && "Global variable is not a pointer!"); |
| 629 | |
| 630 | unsigned AddressSpace = PT->getAddressSpace(); |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 631 | StringRef Section = GV.getSection(); |
| Silviu Baranga | a055aab | 2013-01-07 12:31:25 +0000 | [diff] [blame] | 632 | |
| Anton Korobeynikov | 6bcea06 | 2010-07-26 18:45:39 +0000 | [diff] [blame] | 633 | // Ignore all 'special' globals. |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 634 | if (GV.getName().startswith("llvm.") || |
| 635 | GV.getName().startswith(".llvm.")) |
| Anton Korobeynikov | 6bcea06 | 2010-07-26 18:45:39 +0000 | [diff] [blame] | 636 | continue; |
| 637 | |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 638 | // Ignore all "required" globals: |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 639 | if (isMustKeepGlobalVariable(&GV)) |
| Quentin Colombet | 8fc3409 | 2013-03-18 22:30:07 +0000 | [diff] [blame] | 640 | continue; |
| 641 | |
| Eli Friedman | 0887cf9 | 2018-07-25 20:58:01 +0000 | [diff] [blame] | 642 | Type *Ty = GV.getValueType(); |
| Mehdi Amini | f6727b0 | 2015-07-07 18:49:25 +0000 | [diff] [blame] | 643 | if (DL.getTypeAllocSize(Ty) < MaxOffset) { |
| Peter Collingbourne | fe12d0e | 2016-05-19 04:38:56 +0000 | [diff] [blame] | 644 | if (TM && |
| Huihui Zhang | 2f41065 | 2018-08-30 00:49:50 +0000 | [diff] [blame] | 645 | TargetLoweringObjectFile::getKindForGlobal(&GV, *TM).isBSS()) |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 646 | BSSGlobals[{AddressSpace, Section}].push_back(&GV); |
| Duncan P. N. Exon Smith | 530d040 | 2015-10-09 18:57:47 +0000 | [diff] [blame] | 647 | else if (GV.isConstant()) |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 648 | ConstGlobals[{AddressSpace, Section}].push_back(&GV); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 649 | else |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 650 | Globals[{AddressSpace, Section}].push_back(&GV); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 651 | } |
| 652 | } |
| 653 | |
| David Blaikie | 47bf5c0 | 2015-08-21 22:19:06 +0000 | [diff] [blame] | 654 | for (auto &P : Globals) |
| 655 | if (P.second.size() > 1) |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 656 | Changed |= doMerge(P.second, M, false, P.first.first); |
| Silviu Baranga | a055aab | 2013-01-07 12:31:25 +0000 | [diff] [blame] | 657 | |
| David Blaikie | 47bf5c0 | 2015-08-21 22:19:06 +0000 | [diff] [blame] | 658 | for (auto &P : BSSGlobals) |
| 659 | if (P.second.size() > 1) |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 660 | Changed |= doMerge(P.second, M, false, P.first.first); |
| Bob Wilson | 881b45c | 2010-11-17 21:25:39 +0000 | [diff] [blame] | 661 | |
| David Blaikie | d486000 | 2015-08-25 17:01:36 +0000 | [diff] [blame] | 662 | if (EnableGlobalMergeOnConst) |
| 663 | for (auto &P : ConstGlobals) |
| 664 | if (P.second.size() > 1) |
| Eli Friedman | 1ba5e9a | 2018-08-02 23:54:16 +0000 | [diff] [blame] | 665 | Changed |= doMerge(P.second, M, true, P.first.first); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 666 | |
| 667 | return Changed; |
| 668 | } |
| 669 | |
| Devang Patel | 76c8563 | 2011-10-17 17:17:43 +0000 | [diff] [blame] | 670 | bool GlobalMerge::runOnFunction(Function &F) { |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 671 | return false; |
| 672 | } |
| 673 | |
| Quentin Colombet | 2393cb9 | 2013-03-19 21:46:49 +0000 | [diff] [blame] | 674 | bool GlobalMerge::doFinalization(Module &M) { |
| 675 | MustKeepGlobalVariables.clear(); |
| 676 | return false; |
| 677 | } |
| 678 | |
| Ahmed Bougacha | 8207641 | 2015-06-04 20:39:23 +0000 | [diff] [blame] | 679 | Pass *llvm::createGlobalMergePass(const TargetMachine *TM, unsigned Offset, |
| John Brawn | 8b95424 | 2015-08-03 12:08:41 +0000 | [diff] [blame] | 680 | bool OnlyOptimizeForSize, |
| 681 | bool MergeExternalByDefault) { |
| 682 | bool MergeExternal = (EnableGlobalMergeOnExternal == cl::BOU_UNSET) ? |
| 683 | MergeExternalByDefault : (EnableGlobalMergeOnExternal == cl::BOU_TRUE); |
| 684 | return new GlobalMerge(TM, Offset, OnlyOptimizeForSize, MergeExternal); |
| Anton Korobeynikov | 19edda0 | 2010-07-24 21:52:08 +0000 | [diff] [blame] | 685 | } |