Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 1 | //===--- BackendUtil.cpp - LLVM Backend Utilities -------------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
Daniel Dunbar | c1b1729 | 2010-06-15 17:48:49 +0000 | [diff] [blame] | 10 | #include "clang/CodeGen/BackendUtil.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 11 | #include "clang/Basic/Diagnostic.h" |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 12 | #include "clang/Basic/LangOptions.h" |
Chandler Carruth | 3a02247 | 2012-12-04 09:13:33 +0000 | [diff] [blame] | 13 | #include "clang/Basic/TargetOptions.h" |
Saleem Abdulrasool | 10a4972 | 2016-04-08 16:52:00 +0000 | [diff] [blame] | 14 | #include "clang/Frontend/CodeGenOptions.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 15 | #include "clang/Frontend/FrontendDiagnostic.h" |
Kostya Serebryany | ce2c726 | 2013-12-27 08:11:08 +0000 | [diff] [blame] | 16 | #include "clang/Frontend/Utils.h" |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 17 | #include "clang/Lex/HeaderSearchOptions.h" |
Hal Finkel | 1a328f5 | 2016-12-15 02:19:17 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/SmallSet.h" |
Eric Christopher | 583a1f7 | 2015-09-26 01:25:08 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/StringExtras.h" |
Saleem Abdulrasool | 62849c6 | 2014-05-08 02:28:32 +0000 | [diff] [blame] | 20 | #include "llvm/ADT/StringSwitch.h" |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 21 | #include "llvm/ADT/Triple.h" |
Chandler Carruth | 418bd1a | 2015-01-15 02:16:55 +0000 | [diff] [blame] | 22 | #include "llvm/Analysis/TargetLibraryInfo.h" |
Chandler Carruth | aab5ec0 | 2015-01-31 11:18:46 +0000 | [diff] [blame] | 23 | #include "llvm/Analysis/TargetTransformInfo.h" |
Teresa Johnson | ffc4e24 | 2016-11-11 05:35:12 +0000 | [diff] [blame] | 24 | #include "llvm/Bitcode/BitcodeReader.h" |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 25 | #include "llvm/Bitcode/BitcodeWriter.h" |
| 26 | #include "llvm/Bitcode/BitcodeWriterPass.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 27 | #include "llvm/CodeGen/RegAllocRegistry.h" |
| 28 | #include "llvm/CodeGen/SchedulerRegistry.h" |
David Blaikie | 7a4f7f5 | 2018-01-09 22:03:47 +0000 | [diff] [blame] | 29 | #include "llvm/CodeGen/TargetSubtargetInfo.h" |
Chandler Carruth | ffd5551 | 2013-01-02 11:45:17 +0000 | [diff] [blame] | 30 | #include "llvm/IR/DataLayout.h" |
Chandler Carruth | 0a50c49 | 2014-01-12 11:11:50 +0000 | [diff] [blame] | 31 | #include "llvm/IR/IRPrintingPasses.h" |
Chandler Carruth | 9828e69 | 2015-02-13 09:57:03 +0000 | [diff] [blame] | 32 | #include "llvm/IR/LegacyPassManager.h" |
Chandler Carruth | ffd5551 | 2013-01-02 11:45:17 +0000 | [diff] [blame] | 33 | #include "llvm/IR/Module.h" |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 34 | #include "llvm/IR/ModuleSummaryIndex.h" |
Chandler Carruth | ca88474 | 2014-01-13 09:26:48 +0000 | [diff] [blame] | 35 | #include "llvm/IR/Verifier.h" |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 36 | #include "llvm/LTO/LTOBackend.h" |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 37 | #include "llvm/MC/MCAsmInfo.h" |
Evan Cheng | eeb486d | 2011-06-29 01:14:32 +0000 | [diff] [blame] | 38 | #include "llvm/MC/SubtargetFeature.h" |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 39 | #include "llvm/Passes/PassBuilder.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 40 | #include "llvm/Support/CommandLine.h" |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 41 | #include "llvm/Support/MemoryBuffer.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 42 | #include "llvm/Support/PrettyStackTrace.h" |
Evan Cheng | 494eb06 | 2011-08-24 18:09:14 +0000 | [diff] [blame] | 43 | #include "llvm/Support/TargetRegistry.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 44 | #include "llvm/Support/Timer.h" |
| 45 | #include "llvm/Support/raw_ostream.h" |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 46 | #include "llvm/Target/TargetMachine.h" |
| 47 | #include "llvm/Target/TargetOptions.h" |
Gor Nishanov | 97e3b6d | 2016-10-03 22:44:48 +0000 | [diff] [blame] | 48 | #include "llvm/Transforms/Coroutines.h" |
David Blaikie | 7a4f7f5 | 2018-01-09 22:03:47 +0000 | [diff] [blame] | 49 | #include "llvm/Transforms/GCOVProfiler.h" |
Rafael Espindola | 56a7dab0 | 2011-08-02 21:51:02 +0000 | [diff] [blame] | 50 | #include "llvm/Transforms/IPO.h" |
Chandler Carruth | b72c19f | 2016-08-17 03:09:11 +0000 | [diff] [blame] | 51 | #include "llvm/Transforms/IPO/AlwaysInliner.h" |
Rafael Espindola | 56a7dab0 | 2011-08-02 21:51:02 +0000 | [diff] [blame] | 52 | #include "llvm/Transforms/IPO/PassManagerBuilder.h" |
Tim Shen | 50fedec | 2017-06-01 23:27:51 +0000 | [diff] [blame] | 53 | #include "llvm/Transforms/IPO/ThinLTOBitcodeWriter.h" |
Chandler Carruth | 3a02247 | 2012-12-04 09:13:33 +0000 | [diff] [blame] | 54 | #include "llvm/Transforms/Instrumentation.h" |
Chandler Carruth | afce449 | 2017-11-14 01:47:24 +0000 | [diff] [blame] | 55 | #include "llvm/Transforms/Instrumentation/BoundsChecking.h" |
Michael Gottesman | 90cae77 | 2013-01-28 01:36:00 +0000 | [diff] [blame] | 56 | #include "llvm/Transforms/ObjCARC.h" |
Rafael Espindola | 56a7dab0 | 2011-08-02 21:51:02 +0000 | [diff] [blame] | 57 | #include "llvm/Transforms/Scalar.h" |
Chandler Carruth | 4ddaadc | 2016-03-11 09:02:43 +0000 | [diff] [blame] | 58 | #include "llvm/Transforms/Scalar/GVN.h" |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 59 | #include "llvm/Transforms/Utils/NameAnonGlobals.h" |
Saleem Abdulrasool | 76a4b95 | 2015-01-09 05:10:20 +0000 | [diff] [blame] | 60 | #include "llvm/Transforms/Utils/SymbolRewriter.h" |
Ahmed Charles | dfca6f9 | 2014-03-09 11:36:40 +0000 | [diff] [blame] | 61 | #include <memory> |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 62 | using namespace clang; |
| 63 | using namespace llvm; |
| 64 | |
| 65 | namespace { |
| 66 | |
Davide Italiano | 945de43 | 2017-02-13 16:07:05 +0000 | [diff] [blame] | 67 | // Default filename used for profile generation. |
| 68 | static constexpr StringLiteral DefaultProfileGenName = "default_%m.profraw"; |
| 69 | |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 70 | class EmitAssemblyHelper { |
David Blaikie | 9c902b5 | 2011-09-25 23:23:43 +0000 | [diff] [blame] | 71 | DiagnosticsEngine &Diags; |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 72 | const HeaderSearchOptions &HSOpts; |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 73 | const CodeGenOptions &CodeGenOpts; |
Nick Lewycky | 432add5 | 2011-12-02 22:17:00 +0000 | [diff] [blame] | 74 | const clang::TargetOptions &TargetOpts; |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 75 | const LangOptions &LangOpts; |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 76 | Module *TheModule; |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 77 | |
| 78 | Timer CodeGenerationTime; |
| 79 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 80 | std::unique_ptr<raw_pwrite_stream> OS; |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 81 | |
Chandler Carruth | d294bdb | 2015-02-01 12:26:23 +0000 | [diff] [blame] | 82 | TargetIRAnalysis getTargetIRAnalysis() const { |
Chandler Carruth | aab5ec0 | 2015-01-31 11:18:46 +0000 | [diff] [blame] | 83 | if (TM) |
Chandler Carruth | d294bdb | 2015-02-01 12:26:23 +0000 | [diff] [blame] | 84 | return TM->getTargetIRAnalysis(); |
Chandler Carruth | aab5ec0 | 2015-01-31 11:18:46 +0000 | [diff] [blame] | 85 | |
Chandler Carruth | d294bdb | 2015-02-01 12:26:23 +0000 | [diff] [blame] | 86 | return TargetIRAnalysis(); |
Chandler Carruth | aab5ec0 | 2015-01-31 11:18:46 +0000 | [diff] [blame] | 87 | } |
| 88 | |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 89 | void CreatePasses(legacy::PassManager &MPM, legacy::FunctionPassManager &FPM); |
Nadav Rotem | ec57ab3 | 2012-10-24 00:53:38 +0000 | [diff] [blame] | 90 | |
Rafael Espindola | 2f16bc1 | 2015-04-14 15:15:49 +0000 | [diff] [blame] | 91 | /// Generates the TargetMachine. |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 92 | /// Leaves TM unchanged if it is unable to create the target machine. |
Nadav Rotem | dc06b2d | 2012-10-24 03:52:31 +0000 | [diff] [blame] | 93 | /// Some of our clang tests specify triples which are not built |
| 94 | /// into clang. This is okay because these tests check the generated |
| 95 | /// IR, and they require DataLayout which depends on the triple. |
| 96 | /// In this case, we allow this method to fail and not report an error. |
| 97 | /// When MustCreateTM is used, we print an error if we are unable to load |
| 98 | /// the requested target. |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 99 | void CreateTargetMachine(bool MustCreateTM); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 100 | |
Rafael Espindola | 2f16bc1 | 2015-04-14 15:15:49 +0000 | [diff] [blame] | 101 | /// Add passes necessary to emit assembly or LLVM IR. |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 102 | /// |
| 103 | /// \return True on success. |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 104 | bool AddEmitPasses(legacy::PassManager &CodeGenPasses, BackendAction Action, |
| 105 | raw_pwrite_stream &OS); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 106 | |
| 107 | public: |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 108 | EmitAssemblyHelper(DiagnosticsEngine &_Diags, |
| 109 | const HeaderSearchOptions &HeaderSearchOpts, |
| 110 | const CodeGenOptions &CGOpts, |
Nick Lewycky | 432add5 | 2011-12-02 22:17:00 +0000 | [diff] [blame] | 111 | const clang::TargetOptions &TOpts, |
Teresa Johnson | 4b4f4b9 | 2016-01-08 17:04:29 +0000 | [diff] [blame] | 112 | const LangOptions &LOpts, Module *M) |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 113 | : Diags(_Diags), HSOpts(HeaderSearchOpts), CodeGenOpts(CGOpts), |
| 114 | TargetOpts(TOpts), LangOpts(LOpts), TheModule(M), |
| 115 | CodeGenerationTime("codegen", "Code Generation Time") {} |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 116 | |
| 117 | ~EmitAssemblyHelper() { |
Alp Toker | f4e2238 | 2013-12-20 20:26:53 +0000 | [diff] [blame] | 118 | if (CodeGenOpts.DisableFree) |
David Blaikie | a97eaa1 | 2014-08-29 16:53:14 +0000 | [diff] [blame] | 119 | BuryPointer(std::move(TM)); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 120 | } |
| 121 | |
Ahmed Charles | b898432 | 2014-03-07 20:03:18 +0000 | [diff] [blame] | 122 | std::unique_ptr<TargetMachine> TM; |
Alp Toker | f4e2238 | 2013-12-20 20:26:53 +0000 | [diff] [blame] | 123 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 124 | void EmitAssembly(BackendAction Action, |
| 125 | std::unique_ptr<raw_pwrite_stream> OS); |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 126 | |
| 127 | void EmitAssemblyWithNewPassManager(BackendAction Action, |
| 128 | std::unique_ptr<raw_pwrite_stream> OS); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 129 | }; |
| 130 | |
Alexey Samsonov | c6515b6 | 2012-12-28 09:31:34 +0000 | [diff] [blame] | 131 | // We need this wrapper to access LangOpts and CGOpts from extension functions |
| 132 | // that we add to the PassManagerBuilder. |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 133 | class PassManagerBuilderWrapper : public PassManagerBuilder { |
| 134 | public: |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 135 | PassManagerBuilderWrapper(const Triple &TargetTriple, |
| 136 | const CodeGenOptions &CGOpts, |
Alexey Samsonov | 9ab7362 | 2012-12-03 19:12:58 +0000 | [diff] [blame] | 137 | const LangOptions &LangOpts) |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 138 | : PassManagerBuilder(), TargetTriple(TargetTriple), CGOpts(CGOpts), |
| 139 | LangOpts(LangOpts) {} |
| 140 | const Triple &getTargetTriple() const { return TargetTriple; } |
Alexey Samsonov | 9ab7362 | 2012-12-03 19:12:58 +0000 | [diff] [blame] | 141 | const CodeGenOptions &getCGOpts() const { return CGOpts; } |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 142 | const LangOptions &getLangOpts() const { return LangOpts; } |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 143 | |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 144 | private: |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 145 | const Triple &TargetTriple; |
Alexey Samsonov | 9ab7362 | 2012-12-03 19:12:58 +0000 | [diff] [blame] | 146 | const CodeGenOptions &CGOpts; |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 147 | const LangOptions &LangOpts; |
| 148 | }; |
Alexander Kornienko | ab9db51 | 2015-06-22 23:07:51 +0000 | [diff] [blame] | 149 | } |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 150 | |
Dan Gohman | 5932ce2 | 2012-01-17 20:54:51 +0000 | [diff] [blame] | 151 | static void addObjCARCAPElimPass(const PassManagerBuilder &Builder, PassManagerBase &PM) { |
| 152 | if (Builder.OptLevel > 0) |
| 153 | PM.add(createObjCARCAPElimPass()); |
| 154 | } |
| 155 | |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 156 | static void addObjCARCExpandPass(const PassManagerBuilder &Builder, PassManagerBase &PM) { |
| 157 | if (Builder.OptLevel > 0) |
| 158 | PM.add(createObjCARCExpandPass()); |
| 159 | } |
| 160 | |
| 161 | static void addObjCARCOptPass(const PassManagerBuilder &Builder, PassManagerBase &PM) { |
| 162 | if (Builder.OptLevel > 0) |
| 163 | PM.add(createObjCARCOptPass()); |
| 164 | } |
| 165 | |
Diego Novillo | b56be64 | 2014-03-03 20:06:18 +0000 | [diff] [blame] | 166 | static void addAddDiscriminatorsPass(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 167 | legacy::PassManagerBase &PM) { |
Diego Novillo | b56be64 | 2014-03-03 20:06:18 +0000 | [diff] [blame] | 168 | PM.add(createAddDiscriminatorsPass()); |
| 169 | } |
| 170 | |
Nuno Lopes | a425589 | 2012-05-22 17:19:45 +0000 | [diff] [blame] | 171 | static void addBoundsCheckingPass(const PassManagerBuilder &Builder, |
Vitaly Buka | 9d4eb6f | 2016-06-02 00:24:20 +0000 | [diff] [blame] | 172 | legacy::PassManagerBase &PM) { |
Chandler Carruth | 00a301d | 2017-11-14 01:30:04 +0000 | [diff] [blame] | 173 | PM.add(createBoundsCheckingLegacyPass()); |
Nuno Lopes | a425589 | 2012-05-22 17:19:45 +0000 | [diff] [blame] | 174 | } |
| 175 | |
Kostya Serebryany | 75b4f9e | 2014-11-11 22:15:07 +0000 | [diff] [blame] | 176 | static void addSanitizerCoveragePass(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 177 | legacy::PassManagerBase &PM) { |
Kostya Serebryany | 75b4f9e | 2014-11-11 22:15:07 +0000 | [diff] [blame] | 178 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 179 | static_cast<const PassManagerBuilderWrapper&>(Builder); |
| 180 | const CodeGenOptions &CGOpts = BuilderWrapper.getCGOpts(); |
Alexey Samsonov | 3f3b3ab | 2015-05-07 18:31:29 +0000 | [diff] [blame] | 181 | SanitizerCoverageOptions Opts; |
| 182 | Opts.CoverageType = |
| 183 | static_cast<SanitizerCoverageOptions::Type>(CGOpts.SanitizeCoverageType); |
| 184 | Opts.IndirectCalls = CGOpts.SanitizeCoverageIndirectCalls; |
| 185 | Opts.TraceBB = CGOpts.SanitizeCoverageTraceBB; |
| 186 | Opts.TraceCmp = CGOpts.SanitizeCoverageTraceCmp; |
Kostya Serebryany | 3b41971 | 2016-08-30 01:27:03 +0000 | [diff] [blame] | 187 | Opts.TraceDiv = CGOpts.SanitizeCoverageTraceDiv; |
| 188 | Opts.TraceGep = CGOpts.SanitizeCoverageTraceGep; |
Alexey Samsonov | 3f3b3ab | 2015-05-07 18:31:29 +0000 | [diff] [blame] | 189 | Opts.Use8bitCounters = CGOpts.SanitizeCoverage8bitCounters; |
Kostya Serebryany | d4590c7 | 2016-02-17 21:34:43 +0000 | [diff] [blame] | 190 | Opts.TracePC = CGOpts.SanitizeCoverageTracePC; |
Kostya Serebryany | 60cdd61 | 2016-09-14 01:39:49 +0000 | [diff] [blame] | 191 | Opts.TracePCGuard = CGOpts.SanitizeCoverageTracePCGuard; |
Kostya Serebryany | 50fb618 | 2017-05-05 23:28:18 +0000 | [diff] [blame] | 192 | Opts.NoPrune = CGOpts.SanitizeCoverageNoPrune; |
Kostya Serebryany | 2c2fb88 | 2017-06-08 22:58:19 +0000 | [diff] [blame] | 193 | Opts.Inline8bitCounters = CGOpts.SanitizeCoverageInline8bitCounters; |
Kostya Serebryany | 6145776 | 2017-07-28 00:10:10 +0000 | [diff] [blame] | 194 | Opts.PCTable = CGOpts.SanitizeCoveragePCTable; |
Matt Morehouse | 5c7fc76 | 2017-08-18 18:43:30 +0000 | [diff] [blame] | 195 | Opts.StackDepth = CGOpts.SanitizeCoverageStackDepth; |
Alexey Samsonov | 3f3b3ab | 2015-05-07 18:31:29 +0000 | [diff] [blame] | 196 | PM.add(createSanitizerCoverageModulePass(Opts)); |
Kostya Serebryany | 75b4f9e | 2014-11-11 22:15:07 +0000 | [diff] [blame] | 197 | } |
| 198 | |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 199 | // Check if ASan should use GC-friendly instrumentation for globals. |
| 200 | // First of all, there is no point if -fdata-sections is off (expect for MachO, |
| 201 | // where this is not a factor). Also, on ELF this feature requires an assembler |
| 202 | // extension that only works with -integrated-as at the moment. |
| 203 | static bool asanUseGlobalsGC(const Triple &T, const CodeGenOptions &CGOpts) { |
Evgeniy Stepanov | d991cdd | 2017-05-09 21:57:43 +0000 | [diff] [blame] | 204 | if (!CGOpts.SanitizeAddressGlobalsDeadStripping) |
| 205 | return false; |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 206 | switch (T.getObjectFormat()) { |
| 207 | case Triple::MachO: |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 208 | case Triple::COFF: |
Evgeniy Stepanov | c7b9094 | 2017-04-26 00:51:06 +0000 | [diff] [blame] | 209 | return true; |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 210 | case Triple::ELF: |
| 211 | return CGOpts.DataSections && !CGOpts.DisableIntegratedAS; |
| 212 | default: |
| 213 | return false; |
| 214 | } |
| 215 | } |
| 216 | |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 217 | static void addAddressSanitizerPasses(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 218 | legacy::PassManagerBase &PM) { |
Yury Gribov | 5bfeca1 | 2015-11-11 10:45:48 +0000 | [diff] [blame] | 219 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 220 | static_cast<const PassManagerBuilderWrapper&>(Builder); |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 221 | const Triple &T = BuilderWrapper.getTargetTriple(); |
Yury Gribov | 5bfeca1 | 2015-11-11 10:45:48 +0000 | [diff] [blame] | 222 | const CodeGenOptions &CGOpts = BuilderWrapper.getCGOpts(); |
| 223 | bool Recover = CGOpts.SanitizeRecover.has(SanitizerKind::Address); |
Vitaly Buka | 9d4eb6f | 2016-06-02 00:24:20 +0000 | [diff] [blame] | 224 | bool UseAfterScope = CGOpts.SanitizeAddressUseAfterScope; |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 225 | bool UseGlobalsGC = asanUseGlobalsGC(T, CGOpts); |
Vitaly Buka | 9d4eb6f | 2016-06-02 00:24:20 +0000 | [diff] [blame] | 226 | PM.add(createAddressSanitizerFunctionPass(/*CompileKernel*/ false, Recover, |
| 227 | UseAfterScope)); |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 228 | PM.add(createAddressSanitizerModulePass(/*CompileKernel*/ false, Recover, |
| 229 | UseGlobalsGC)); |
Alexander Potapenko | b9b73ef | 2015-06-19 12:19:07 +0000 | [diff] [blame] | 230 | } |
| 231 | |
| 232 | static void addKernelAddressSanitizerPasses(const PassManagerBuilder &Builder, |
| 233 | legacy::PassManagerBase &PM) { |
Vitaly Buka | 9d4eb6f | 2016-06-02 00:24:20 +0000 | [diff] [blame] | 234 | PM.add(createAddressSanitizerFunctionPass( |
| 235 | /*CompileKernel*/ true, |
| 236 | /*Recover*/ true, /*UseAfterScope*/ false)); |
Yury Gribov | 5bfeca1 | 2015-11-11 10:45:48 +0000 | [diff] [blame] | 237 | PM.add(createAddressSanitizerModulePass(/*CompileKernel*/true, |
| 238 | /*Recover*/true)); |
Kostya Serebryany | 8855ff6 | 2011-11-16 17:34:26 +0000 | [diff] [blame] | 239 | } |
| 240 | |
Evgeniy Stepanov | 12817e5 | 2017-12-09 01:32:07 +0000 | [diff] [blame] | 241 | static void addHWAddressSanitizerPasses(const PassManagerBuilder &Builder, |
| 242 | legacy::PassManagerBase &PM) { |
Evgeniy Stepanov | 3fd1b1a | 2017-12-20 19:05:44 +0000 | [diff] [blame] | 243 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 244 | static_cast<const PassManagerBuilderWrapper &>(Builder); |
| 245 | const CodeGenOptions &CGOpts = BuilderWrapper.getCGOpts(); |
| 246 | bool Recover = CGOpts.SanitizeRecover.has(SanitizerKind::HWAddress); |
| 247 | PM.add(createHWAddressSanitizerPass(Recover)); |
Evgeniy Stepanov | 12817e5 | 2017-12-09 01:32:07 +0000 | [diff] [blame] | 248 | } |
| 249 | |
Evgeniy Stepanov | aea92e5 | 2012-12-03 13:20:43 +0000 | [diff] [blame] | 250 | static void addMemorySanitizerPass(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 251 | legacy::PassManagerBase &PM) { |
Evgeniy Stepanov | ad8ab3d | 2012-12-24 08:42:34 +0000 | [diff] [blame] | 252 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 253 | static_cast<const PassManagerBuilderWrapper&>(Builder); |
| 254 | const CodeGenOptions &CGOpts = BuilderWrapper.getCGOpts(); |
Evgeniy Stepanov | 5b5370a | 2016-11-07 21:02:11 +0000 | [diff] [blame] | 255 | int TrackOrigins = CGOpts.SanitizeMemoryTrackOrigins; |
| 256 | bool Recover = CGOpts.SanitizeRecover.has(SanitizerKind::Memory); |
| 257 | PM.add(createMemorySanitizerPass(TrackOrigins, Recover)); |
Evgeniy Stepanov | 1028467 | 2013-01-31 09:53:29 +0000 | [diff] [blame] | 258 | |
| 259 | // MemorySanitizer inserts complex instrumentation that mostly follows |
| 260 | // the logic of the original code, but operates on "shadow" values. |
| 261 | // It can benefit from re-running some general purpose optimization passes. |
| 262 | if (Builder.OptLevel > 0) { |
| 263 | PM.add(createEarlyCSEPass()); |
| 264 | PM.add(createReassociatePass()); |
| 265 | PM.add(createLICMPass()); |
| 266 | PM.add(createGVNPass()); |
| 267 | PM.add(createInstructionCombiningPass()); |
| 268 | PM.add(createDeadStoreEliminationPass()); |
| 269 | } |
Evgeniy Stepanov | aea92e5 | 2012-12-03 13:20:43 +0000 | [diff] [blame] | 270 | } |
| 271 | |
Kostya Serebryany | 28a7a11 | 2012-03-01 22:27:08 +0000 | [diff] [blame] | 272 | static void addThreadSanitizerPass(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 273 | legacy::PassManagerBase &PM) { |
Alexey Samsonov | 1cf07ea | 2014-06-02 18:08:08 +0000 | [diff] [blame] | 274 | PM.add(createThreadSanitizerPass()); |
Kostya Serebryany | 28a7a11 | 2012-03-01 22:27:08 +0000 | [diff] [blame] | 275 | } |
| 276 | |
Peter Collingbourne | c377275 | 2013-08-07 22:47:34 +0000 | [diff] [blame] | 277 | static void addDataFlowSanitizerPass(const PassManagerBuilder &Builder, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 278 | legacy::PassManagerBase &PM) { |
Peter Collingbourne | 276be3c | 2013-08-14 18:54:18 +0000 | [diff] [blame] | 279 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 280 | static_cast<const PassManagerBuilderWrapper&>(Builder); |
Alexey Samsonov | ae5804f | 2014-10-15 20:22:54 +0000 | [diff] [blame] | 281 | const LangOptions &LangOpts = BuilderWrapper.getLangOpts(); |
Alexey Samsonov | a511cdd | 2015-02-04 17:40:08 +0000 | [diff] [blame] | 282 | PM.add(createDataFlowSanitizerPass(LangOpts.SanitizerBlacklistFiles)); |
Peter Collingbourne | c377275 | 2013-08-07 22:47:34 +0000 | [diff] [blame] | 283 | } |
| 284 | |
Derek Bruening | 256c2e1 | 2016-04-21 21:32:04 +0000 | [diff] [blame] | 285 | static void addEfficiencySanitizerPass(const PassManagerBuilder &Builder, |
| 286 | legacy::PassManagerBase &PM) { |
| 287 | const PassManagerBuilderWrapper &BuilderWrapper = |
| 288 | static_cast<const PassManagerBuilderWrapper&>(Builder); |
| 289 | const LangOptions &LangOpts = BuilderWrapper.getLangOpts(); |
| 290 | EfficiencySanitizerOptions Opts; |
| 291 | if (LangOpts.Sanitize.has(SanitizerKind::EfficiencyCacheFrag)) |
| 292 | Opts.ToolType = EfficiencySanitizerOptions::ESAN_CacheFrag; |
Derek Bruening | 293772e | 2016-05-25 00:41:24 +0000 | [diff] [blame] | 293 | else if (LangOpts.Sanitize.has(SanitizerKind::EfficiencyWorkingSet)) |
| 294 | Opts.ToolType = EfficiencySanitizerOptions::ESAN_WorkingSet; |
Derek Bruening | 256c2e1 | 2016-04-21 21:32:04 +0000 | [diff] [blame] | 295 | PM.add(createEfficiencySanitizerPass(Opts)); |
| 296 | } |
| 297 | |
Chandler Carruth | 57bb7c7 | 2015-01-24 02:25:21 +0000 | [diff] [blame] | 298 | static TargetLibraryInfoImpl *createTLII(llvm::Triple &TargetTriple, |
| 299 | const CodeGenOptions &CodeGenOpts) { |
| 300 | TargetLibraryInfoImpl *TLII = new TargetLibraryInfoImpl(TargetTriple); |
Rafael Espindola | b8a71c5 | 2014-08-21 17:58:42 +0000 | [diff] [blame] | 301 | if (!CodeGenOpts.SimplifyLibCalls) |
Chandler Carruth | 57bb7c7 | 2015-01-24 02:25:21 +0000 | [diff] [blame] | 302 | TLII->disableAllFunctions(); |
Chad Rosier | 7dbc9cf | 2016-01-06 14:35:46 +0000 | [diff] [blame] | 303 | else { |
| 304 | // Disable individual libc/libm calls in TargetLibraryInfo. |
David L. Jones | 7a7dd03 | 2017-01-23 23:16:58 +0000 | [diff] [blame] | 305 | LibFunc F; |
Chad Rosier | 7dbc9cf | 2016-01-06 14:35:46 +0000 | [diff] [blame] | 306 | for (auto &FuncName : CodeGenOpts.getNoBuiltinFuncs()) |
| 307 | if (TLII->getLibFunc(FuncName, F)) |
| 308 | TLII->setUnavailable(F); |
| 309 | } |
Michael Zolotukhin | c888dd0 | 2015-03-17 20:03:11 +0000 | [diff] [blame] | 310 | |
| 311 | switch (CodeGenOpts.getVecLib()) { |
| 312 | case CodeGenOptions::Accelerate: |
| 313 | TLII->addVectorizableFunctionsFromVecLib(TargetLibraryInfoImpl::Accelerate); |
| 314 | break; |
Matt Masten | 6731dea | 2016-07-29 16:44:24 +0000 | [diff] [blame] | 315 | case CodeGenOptions::SVML: |
| 316 | TLII->addVectorizableFunctionsFromVecLib(TargetLibraryInfoImpl::SVML); |
| 317 | break; |
Michael Zolotukhin | c888dd0 | 2015-03-17 20:03:11 +0000 | [diff] [blame] | 318 | default: |
| 319 | break; |
| 320 | } |
Chandler Carruth | 57bb7c7 | 2015-01-24 02:25:21 +0000 | [diff] [blame] | 321 | return TLII; |
Rafael Espindola | b8a71c5 | 2014-08-21 17:58:42 +0000 | [diff] [blame] | 322 | } |
| 323 | |
Saleem Abdulrasool | 76a4b95 | 2015-01-09 05:10:20 +0000 | [diff] [blame] | 324 | static void addSymbolRewriterPass(const CodeGenOptions &Opts, |
Chandler Carruth | 8f4f509 | 2015-02-13 09:47:49 +0000 | [diff] [blame] | 325 | legacy::PassManager *MPM) { |
Saleem Abdulrasool | 76a4b95 | 2015-01-09 05:10:20 +0000 | [diff] [blame] | 326 | llvm::SymbolRewriter::RewriteDescriptorList DL; |
| 327 | |
| 328 | llvm::SymbolRewriter::RewriteMapParser MapParser; |
| 329 | for (const auto &MapFile : Opts.RewriteMapFiles) |
| 330 | MapParser.parse(MapFile, &DL); |
| 331 | |
| 332 | MPM->add(createRewriteSymbolsPass(DL)); |
| 333 | } |
| 334 | |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 335 | static CodeGenOpt::Level getCGOptLevel(const CodeGenOptions &CodeGenOpts) { |
| 336 | switch (CodeGenOpts.OptimizationLevel) { |
| 337 | default: |
| 338 | llvm_unreachable("Invalid optimization level!"); |
| 339 | case 0: |
| 340 | return CodeGenOpt::None; |
| 341 | case 1: |
| 342 | return CodeGenOpt::Less; |
| 343 | case 2: |
| 344 | return CodeGenOpt::Default; // O2/Os/Oz |
| 345 | case 3: |
| 346 | return CodeGenOpt::Aggressive; |
| 347 | } |
| 348 | } |
| 349 | |
Rafael Espindola | aa0226e | 2017-08-03 02:16:28 +0000 | [diff] [blame] | 350 | static Optional<llvm::CodeModel::Model> |
| 351 | getCodeModel(const CodeGenOptions &CodeGenOpts) { |
| 352 | unsigned CodeModel = llvm::StringSwitch<unsigned>(CodeGenOpts.CodeModel) |
| 353 | .Case("small", llvm::CodeModel::Small) |
| 354 | .Case("kernel", llvm::CodeModel::Kernel) |
| 355 | .Case("medium", llvm::CodeModel::Medium) |
| 356 | .Case("large", llvm::CodeModel::Large) |
| 357 | .Case("default", ~1u) |
| 358 | .Default(~0u); |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 359 | assert(CodeModel != ~0u && "invalid code model!"); |
Rafael Espindola | aa0226e | 2017-08-03 02:16:28 +0000 | [diff] [blame] | 360 | if (CodeModel == ~1u) |
| 361 | return None; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 362 | return static_cast<llvm::CodeModel::Model>(CodeModel); |
| 363 | } |
| 364 | |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 365 | static TargetMachine::CodeGenFileType getCodeGenFileType(BackendAction Action) { |
| 366 | if (Action == Backend_EmitObj) |
| 367 | return TargetMachine::CGFT_ObjectFile; |
| 368 | else if (Action == Backend_EmitMCNull) |
| 369 | return TargetMachine::CGFT_Null; |
| 370 | else { |
| 371 | assert(Action == Backend_EmitAssembly && "Invalid action!"); |
| 372 | return TargetMachine::CGFT_AssemblyFile; |
| 373 | } |
| 374 | } |
| 375 | |
| 376 | static void initTargetOptions(llvm::TargetOptions &Options, |
| 377 | const CodeGenOptions &CodeGenOpts, |
| 378 | const clang::TargetOptions &TargetOpts, |
| 379 | const LangOptions &LangOpts, |
| 380 | const HeaderSearchOptions &HSOpts) { |
| 381 | Options.ThreadModel = |
| 382 | llvm::StringSwitch<llvm::ThreadModel::Model>(CodeGenOpts.ThreadModel) |
| 383 | .Case("posix", llvm::ThreadModel::POSIX) |
| 384 | .Case("single", llvm::ThreadModel::Single); |
| 385 | |
| 386 | // Set float ABI type. |
| 387 | assert((CodeGenOpts.FloatABI == "soft" || CodeGenOpts.FloatABI == "softfp" || |
| 388 | CodeGenOpts.FloatABI == "hard" || CodeGenOpts.FloatABI.empty()) && |
| 389 | "Invalid Floating Point ABI!"); |
| 390 | Options.FloatABIType = |
| 391 | llvm::StringSwitch<llvm::FloatABI::ABIType>(CodeGenOpts.FloatABI) |
| 392 | .Case("soft", llvm::FloatABI::Soft) |
| 393 | .Case("softfp", llvm::FloatABI::Soft) |
| 394 | .Case("hard", llvm::FloatABI::Hard) |
| 395 | .Default(llvm::FloatABI::Default); |
| 396 | |
| 397 | // Set FP fusion mode. |
| 398 | switch (LangOpts.getDefaultFPContractMode()) { |
| 399 | case LangOptions::FPC_Off: |
Adam Nemet | 03af424 | 2017-04-20 17:09:35 +0000 | [diff] [blame] | 400 | // Preserve any contraction performed by the front-end. (Strict performs |
| 401 | // splitting of the muladd instrinsic in the backend.) |
| 402 | Options.AllowFPOpFusion = llvm::FPOpFusion::Standard; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 403 | break; |
| 404 | case LangOptions::FPC_On: |
| 405 | Options.AllowFPOpFusion = llvm::FPOpFusion::Standard; |
| 406 | break; |
| 407 | case LangOptions::FPC_Fast: |
| 408 | Options.AllowFPOpFusion = llvm::FPOpFusion::Fast; |
| 409 | break; |
| 410 | } |
| 411 | |
| 412 | Options.UseInitArray = CodeGenOpts.UseInitArray; |
| 413 | Options.DisableIntegratedAS = CodeGenOpts.DisableIntegratedAS; |
Saleem Abdulrasool | 5444890 | 2017-06-09 00:40:30 +0000 | [diff] [blame] | 414 | Options.CompressDebugSections = CodeGenOpts.getCompressDebugSections(); |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 415 | Options.RelaxELFRelocations = CodeGenOpts.RelaxELFRelocations; |
| 416 | |
| 417 | // Set EABI version. |
Yuka Takahashi | dc77150 | 2017-07-01 07:57:23 +0000 | [diff] [blame] | 418 | Options.EABIVersion = TargetOpts.EABIVersion; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 419 | |
| 420 | if (LangOpts.SjLjExceptions) |
| 421 | Options.ExceptionModel = llvm::ExceptionHandling::SjLj; |
Martell Malone | c950c65 | 2017-11-29 07:25:12 +0000 | [diff] [blame] | 422 | if (LangOpts.SEHExceptions) |
| 423 | Options.ExceptionModel = llvm::ExceptionHandling::WinEH; |
| 424 | if (LangOpts.DWARFExceptions) |
| 425 | Options.ExceptionModel = llvm::ExceptionHandling::DwarfCFI; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 426 | |
| 427 | Options.NoInfsFPMath = CodeGenOpts.NoInfsFPMath; |
| 428 | Options.NoNaNsFPMath = CodeGenOpts.NoNaNsFPMath; |
| 429 | Options.NoZerosInBSS = CodeGenOpts.NoZeroInitializedInBSS; |
| 430 | Options.UnsafeFPMath = CodeGenOpts.UnsafeFPMath; |
| 431 | Options.StackAlignmentOverride = CodeGenOpts.StackAlignment; |
| 432 | Options.FunctionSections = CodeGenOpts.FunctionSections; |
| 433 | Options.DataSections = CodeGenOpts.DataSections; |
| 434 | Options.UniqueSectionNames = CodeGenOpts.UniqueSectionNames; |
| 435 | Options.EmulatedTLS = CodeGenOpts.EmulatedTLS; |
| 436 | Options.DebuggerTuning = CodeGenOpts.getDebuggerTuning(); |
Sean Eveson | 5110d4f | 2018-01-08 13:42:26 +0000 | [diff] [blame] | 437 | Options.EmitStackSizeSection = CodeGenOpts.StackSizeSection; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 438 | |
David Blaikie | 8150355 | 2017-04-21 23:35:36 +0000 | [diff] [blame] | 439 | if (CodeGenOpts.EnableSplitDwarf) |
| 440 | Options.MCOptions.SplitDwarfFile = CodeGenOpts.SplitDwarfFile; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 441 | Options.MCOptions.MCRelaxAll = CodeGenOpts.RelaxAll; |
| 442 | Options.MCOptions.MCSaveTempLabels = CodeGenOpts.SaveTempLabels; |
| 443 | Options.MCOptions.MCUseDwarfDirectory = !CodeGenOpts.NoDwarfDirectoryAsm; |
| 444 | Options.MCOptions.MCNoExecStack = CodeGenOpts.NoExecStack; |
| 445 | Options.MCOptions.MCIncrementalLinkerCompatible = |
| 446 | CodeGenOpts.IncrementalLinkerCompatible; |
| 447 | Options.MCOptions.MCPIECopyRelocations = CodeGenOpts.PIECopyRelocations; |
| 448 | Options.MCOptions.MCFatalWarnings = CodeGenOpts.FatalWarnings; |
| 449 | Options.MCOptions.AsmVerbose = CodeGenOpts.AsmVerbose; |
| 450 | Options.MCOptions.PreserveAsmComments = CodeGenOpts.PreserveAsmComments; |
| 451 | Options.MCOptions.ABIName = TargetOpts.ABI; |
| 452 | for (const auto &Entry : HSOpts.UserEntries) |
| 453 | if (!Entry.IsFramework && |
| 454 | (Entry.Group == frontend::IncludeDirGroup::Quoted || |
| 455 | Entry.Group == frontend::IncludeDirGroup::Angled || |
| 456 | Entry.Group == frontend::IncludeDirGroup::System)) |
| 457 | Options.MCOptions.IASSearchPaths.push_back( |
| 458 | Entry.IgnoreSysRoot ? Entry.Path : HSOpts.Sysroot + Entry.Path); |
| 459 | } |
David Blaikie | 7a4f7f5 | 2018-01-09 22:03:47 +0000 | [diff] [blame] | 460 | static Optional<GCOVOptions> getGCOVOptions(const CodeGenOptions &CodeGenOpts) { |
| 461 | if (CodeGenOpts.DisableGCov) |
| 462 | return None; |
| 463 | if (!CodeGenOpts.EmitGcovArcs && !CodeGenOpts.EmitGcovNotes) |
| 464 | return None; |
| 465 | // Not using 'GCOVOptions::getDefault' allows us to avoid exiting if |
| 466 | // LLVM's -default-gcov-version flag is set to something invalid. |
| 467 | GCOVOptions Options; |
| 468 | Options.EmitNotes = CodeGenOpts.EmitGcovNotes; |
| 469 | Options.EmitData = CodeGenOpts.EmitGcovArcs; |
| 470 | llvm::copy(CodeGenOpts.CoverageVersion, std::begin(Options.Version)); |
| 471 | Options.UseCfgChecksum = CodeGenOpts.CoverageExtraChecksum; |
| 472 | Options.NoRedZone = CodeGenOpts.DisableRedZone; |
| 473 | Options.FunctionNamesInData = !CodeGenOpts.CoverageNoFunctionNamesInData; |
| 474 | Options.ExitBlockBeforeBody = CodeGenOpts.CoverageExitBlockBeforeBody; |
| 475 | return Options; |
| 476 | } |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 477 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 478 | void EmitAssemblyHelper::CreatePasses(legacy::PassManager &MPM, |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 479 | legacy::FunctionPassManager &FPM) { |
Chandler Carruth | 93786da | 2016-12-23 00:23:01 +0000 | [diff] [blame] | 480 | // Handle disabling of all LLVM passes, where we want to preserve the |
| 481 | // internal module before any optimization. |
Steven Wu | 546a196 | 2015-07-17 20:09:56 +0000 | [diff] [blame] | 482 | if (CodeGenOpts.DisableLLVMPasses) |
| 483 | return; |
| 484 | |
Marcin Koscielnicki | 9a063e7 | 2016-11-22 20:03:35 +0000 | [diff] [blame] | 485 | // Figure out TargetLibraryInfo. This needs to be added to MPM and FPM |
| 486 | // manually (and not via PMBuilder), since some passes (eg. InstrProfiling) |
| 487 | // are inserted before PMBuilder ones - they'd get the default-constructed |
| 488 | // TLI with an unknown target otherwise. |
Teresa Johnson | aff2232 | 2015-12-07 19:21:34 +0000 | [diff] [blame] | 489 | Triple TargetTriple(TheModule->getTargetTriple()); |
Marcin Koscielnicki | 9a063e7 | 2016-11-22 20:03:35 +0000 | [diff] [blame] | 490 | std::unique_ptr<TargetLibraryInfoImpl> TLII( |
| 491 | createTLII(TargetTriple, CodeGenOpts)); |
Teresa Johnson | aff2232 | 2015-12-07 19:21:34 +0000 | [diff] [blame] | 492 | |
Evgeniy Stepanov | df217a2 | 2017-04-24 19:34:12 +0000 | [diff] [blame] | 493 | PassManagerBuilderWrapper PMBuilder(TargetTriple, CodeGenOpts, LangOpts); |
| 494 | |
Chandler Carruth | fcd3314 | 2016-12-23 01:24:49 +0000 | [diff] [blame] | 495 | // At O0 and O1 we only run the always inliner which is more efficient. At |
| 496 | // higher optimization levels we run the normal inliner. |
| 497 | if (CodeGenOpts.OptimizationLevel <= 1) { |
Mehdi Amini | 7f87307 | 2017-01-06 23:18:09 +0000 | [diff] [blame] | 498 | bool InsertLifetimeIntrinsics = (CodeGenOpts.OptimizationLevel != 0 && |
| 499 | !CodeGenOpts.DisableLifetimeMarkers); |
Chandler Carruth | fcd3314 | 2016-12-23 01:24:49 +0000 | [diff] [blame] | 500 | PMBuilder.Inliner = createAlwaysInlinerLegacyPass(InsertLifetimeIntrinsics); |
| 501 | } else { |
Dehao Chen | ce39fdd | 2017-03-21 19:55:46 +0000 | [diff] [blame] | 502 | // We do not want to inline hot callsites for SamplePGO module-summary build |
| 503 | // because profile annotation will happen again in ThinLTO backend, and we |
| 504 | // want the IR of the hot path to match the profile. |
Chandler Carruth | fcd3314 | 2016-12-23 01:24:49 +0000 | [diff] [blame] | 505 | PMBuilder.Inliner = createFunctionInliningPass( |
Dehao Chen | ce39fdd | 2017-03-21 19:55:46 +0000 | [diff] [blame] | 506 | CodeGenOpts.OptimizationLevel, CodeGenOpts.OptimizeSize, |
| 507 | (!CodeGenOpts.SampleProfileFile.empty() && |
| 508 | CodeGenOpts.EmitSummaryIndex)); |
Teresa Johnson | aff2232 | 2015-12-07 19:21:34 +0000 | [diff] [blame] | 509 | } |
| 510 | |
Chandler Carruth | fcd3314 | 2016-12-23 01:24:49 +0000 | [diff] [blame] | 511 | PMBuilder.OptLevel = CodeGenOpts.OptimizationLevel; |
Chris Lattner | ecf0ba5 | 2011-05-21 23:50:44 +0000 | [diff] [blame] | 512 | PMBuilder.SizeLevel = CodeGenOpts.OptimizeSize; |
Nick Lewycky | d3f3e4f | 2013-06-25 01:49:44 +0000 | [diff] [blame] | 513 | PMBuilder.SLPVectorize = CodeGenOpts.VectorizeSLP; |
| 514 | PMBuilder.LoopVectorize = CodeGenOpts.VectorizeLoop; |
Andrew Trick | b2a8472 | 2011-04-05 18:49:32 +0000 | [diff] [blame] | 515 | |
Chris Lattner | ecf0ba5 | 2011-05-21 23:50:44 +0000 | [diff] [blame] | 516 | PMBuilder.DisableUnrollLoops = !CodeGenOpts.UnrollLoops; |
Nick Lewycky | f04f237 | 2014-10-24 00:49:29 +0000 | [diff] [blame] | 517 | PMBuilder.MergeFunctions = CodeGenOpts.MergeFunctions; |
Teresa Johnson | a0d2940 | 2016-03-15 00:04:44 +0000 | [diff] [blame] | 518 | PMBuilder.PrepareForThinLTO = CodeGenOpts.EmitSummaryIndex; |
Teresa Johnson | 8749d804 | 2015-07-06 16:23:00 +0000 | [diff] [blame] | 519 | PMBuilder.PrepareForLTO = CodeGenOpts.PrepareForLTO; |
Hal Finkel | ce0697f | 2013-11-17 16:03:29 +0000 | [diff] [blame] | 520 | PMBuilder.RerollLoops = CodeGenOpts.RerollLoops; |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 521 | |
Marcin Koscielnicki | 9a063e7 | 2016-11-22 20:03:35 +0000 | [diff] [blame] | 522 | MPM.add(new TargetLibraryInfoWrapperPass(*TLII)); |
| 523 | |
Justin Lebar | 5a7df9c | 2016-04-27 19:12:56 +0000 | [diff] [blame] | 524 | if (TM) |
Stanislav Mekhanoshin | 61da067 | 2017-01-26 16:49:21 +0000 | [diff] [blame] | 525 | TM->adjustPassManager(PMBuilder); |
Justin Lebar | 5a7df9c | 2016-04-27 19:12:56 +0000 | [diff] [blame] | 526 | |
Dehao Chen | 7810d4f | 2017-02-21 20:36:21 +0000 | [diff] [blame] | 527 | if (CodeGenOpts.DebugInfoForProfiling || |
| 528 | !CodeGenOpts.SampleProfileFile.empty()) |
| 529 | PMBuilder.addExtension(PassManagerBuilder::EP_EarlyAsPossible, |
| 530 | addAddDiscriminatorsPass); |
Diego Novillo | b56be64 | 2014-03-03 20:06:18 +0000 | [diff] [blame] | 531 | |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 532 | // In ObjC ARC mode, add the main ARC optimization passes. |
| 533 | if (LangOpts.ObjCAutoRefCount) { |
| 534 | PMBuilder.addExtension(PassManagerBuilder::EP_EarlyAsPossible, |
| 535 | addObjCARCExpandPass); |
Dan Gohman | 5932ce2 | 2012-01-17 20:54:51 +0000 | [diff] [blame] | 536 | PMBuilder.addExtension(PassManagerBuilder::EP_ModuleOptimizerEarly, |
| 537 | addObjCARCAPElimPass); |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 538 | PMBuilder.addExtension(PassManagerBuilder::EP_ScalarOptimizerLate, |
| 539 | addObjCARCOptPass); |
| 540 | } |
Kostya Serebryany | 8855ff6 | 2011-11-16 17:34:26 +0000 | [diff] [blame] | 541 | |
Alexey Samsonov | edf99a9 | 2014-11-07 22:29:38 +0000 | [diff] [blame] | 542 | if (LangOpts.Sanitize.has(SanitizerKind::LocalBounds)) { |
Nuno Lopes | a425589 | 2012-05-22 17:19:45 +0000 | [diff] [blame] | 543 | PMBuilder.addExtension(PassManagerBuilder::EP_ScalarOptimizerLate, |
| 544 | addBoundsCheckingPass); |
| 545 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 546 | addBoundsCheckingPass); |
| 547 | } |
| 548 | |
Alexey Samsonov | dfa908c | 2015-05-07 22:34:06 +0000 | [diff] [blame] | 549 | if (CodeGenOpts.SanitizeCoverageType || |
| 550 | CodeGenOpts.SanitizeCoverageIndirectCalls || |
| 551 | CodeGenOpts.SanitizeCoverageTraceCmp) { |
Kostya Serebryany | 75b4f9e | 2014-11-11 22:15:07 +0000 | [diff] [blame] | 552 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 553 | addSanitizerCoveragePass); |
| 554 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 555 | addSanitizerCoveragePass); |
| 556 | } |
| 557 | |
Alexey Samsonov | edf99a9 | 2014-11-07 22:29:38 +0000 | [diff] [blame] | 558 | if (LangOpts.Sanitize.has(SanitizerKind::Address)) { |
Kostya Serebryany | 7e247f2 | 2012-10-15 14:22:56 +0000 | [diff] [blame] | 559 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 560 | addAddressSanitizerPasses); |
Kostya Serebryany | d476857 | 2011-11-30 22:20:21 +0000 | [diff] [blame] | 561 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
Alexey Samsonov | 0e96bec | 2012-11-29 22:36:21 +0000 | [diff] [blame] | 562 | addAddressSanitizerPasses); |
Kostya Serebryany | 8855ff6 | 2011-11-16 17:34:26 +0000 | [diff] [blame] | 563 | } |
Kostya Serebryany | 28a7a11 | 2012-03-01 22:27:08 +0000 | [diff] [blame] | 564 | |
Alexander Potapenko | b9b73ef | 2015-06-19 12:19:07 +0000 | [diff] [blame] | 565 | if (LangOpts.Sanitize.has(SanitizerKind::KernelAddress)) { |
| 566 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 567 | addKernelAddressSanitizerPasses); |
| 568 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 569 | addKernelAddressSanitizerPasses); |
| 570 | } |
| 571 | |
Evgeniy Stepanov | 12817e5 | 2017-12-09 01:32:07 +0000 | [diff] [blame] | 572 | if (LangOpts.Sanitize.has(SanitizerKind::HWAddress)) { |
| 573 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 574 | addHWAddressSanitizerPasses); |
| 575 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 576 | addHWAddressSanitizerPasses); |
| 577 | } |
| 578 | |
Alexey Samsonov | edf99a9 | 2014-11-07 22:29:38 +0000 | [diff] [blame] | 579 | if (LangOpts.Sanitize.has(SanitizerKind::Memory)) { |
Evgeniy Stepanov | aea92e5 | 2012-12-03 13:20:43 +0000 | [diff] [blame] | 580 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 581 | addMemorySanitizerPass); |
| 582 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 583 | addMemorySanitizerPass); |
| 584 | } |
| 585 | |
Alexey Samsonov | edf99a9 | 2014-11-07 22:29:38 +0000 | [diff] [blame] | 586 | if (LangOpts.Sanitize.has(SanitizerKind::Thread)) { |
Kostya Serebryany | d18cb50 | 2012-03-23 23:25:23 +0000 | [diff] [blame] | 587 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
Kostya Serebryany | 28a7a11 | 2012-03-01 22:27:08 +0000 | [diff] [blame] | 588 | addThreadSanitizerPass); |
| 589 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 590 | addThreadSanitizerPass); |
| 591 | } |
| 592 | |
Alexey Samsonov | edf99a9 | 2014-11-07 22:29:38 +0000 | [diff] [blame] | 593 | if (LangOpts.Sanitize.has(SanitizerKind::DataFlow)) { |
Peter Collingbourne | c377275 | 2013-08-07 22:47:34 +0000 | [diff] [blame] | 594 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 595 | addDataFlowSanitizerPass); |
| 596 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 597 | addDataFlowSanitizerPass); |
| 598 | } |
| 599 | |
Gor Nishanov | 97e3b6d | 2016-10-03 22:44:48 +0000 | [diff] [blame] | 600 | if (LangOpts.CoroutinesTS) |
| 601 | addCoroutinePassesToExtensionPoints(PMBuilder); |
| 602 | |
Derek Bruening | 256c2e1 | 2016-04-21 21:32:04 +0000 | [diff] [blame] | 603 | if (LangOpts.Sanitize.hasOneOf(SanitizerKind::Efficiency)) { |
| 604 | PMBuilder.addExtension(PassManagerBuilder::EP_OptimizerLast, |
| 605 | addEfficiencySanitizerPass); |
| 606 | PMBuilder.addExtension(PassManagerBuilder::EP_EnabledOnOptLevel0, |
| 607 | addEfficiencySanitizerPass); |
| 608 | } |
| 609 | |
Chris Lattner | 5c12367 | 2011-05-21 20:40:11 +0000 | [diff] [blame] | 610 | // Set up the per-function pass manager. |
Marcin Koscielnicki | 9a063e7 | 2016-11-22 20:03:35 +0000 | [diff] [blame] | 611 | FPM.add(new TargetLibraryInfoWrapperPass(*TLII)); |
Chris Lattner | 5c12367 | 2011-05-21 20:40:11 +0000 | [diff] [blame] | 612 | if (CodeGenOpts.VerifyModule) |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 613 | FPM.add(createVerifierPass()); |
Andrew Trick | b2a8472 | 2011-04-05 18:49:32 +0000 | [diff] [blame] | 614 | |
Chris Lattner | 5c12367 | 2011-05-21 20:40:11 +0000 | [diff] [blame] | 615 | // Set up the per-module pass manager. |
Saleem Abdulrasool | 76a4b95 | 2015-01-09 05:10:20 +0000 | [diff] [blame] | 616 | if (!CodeGenOpts.RewriteMapFiles.empty()) |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 617 | addSymbolRewriterPass(CodeGenOpts, &MPM); |
Chris Lattner | d98cec5 | 2011-02-18 22:20:38 +0000 | [diff] [blame] | 618 | |
David Blaikie | 7a4f7f5 | 2018-01-09 22:03:47 +0000 | [diff] [blame] | 619 | if (Optional<GCOVOptions> Options = getGCOVOptions(CodeGenOpts)) { |
| 620 | MPM.add(createGCOVProfilerPass(*Options)); |
Benjamin Kramer | 8c30592 | 2016-02-02 11:06:51 +0000 | [diff] [blame] | 621 | if (CodeGenOpts.getDebugInfo() == codegenoptions::NoDebugInfo) |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 622 | MPM.add(createStripSymbolsPass(true)); |
Nick Lewycky | 207bce3 | 2011-04-21 23:44:07 +0000 | [diff] [blame] | 623 | } |
Nadav Rotem | dc06b2d | 2012-10-24 03:52:31 +0000 | [diff] [blame] | 624 | |
Rong Xu | 9837ef5 | 2016-02-04 18:39:09 +0000 | [diff] [blame] | 625 | if (CodeGenOpts.hasProfileClangInstr()) { |
Justin Bogner | 970ac60 | 2014-12-08 19:04:51 +0000 | [diff] [blame] | 626 | InstrProfOptions Options; |
| 627 | Options.NoRedZone = CodeGenOpts.DisableRedZone; |
Justin Bogner | a71e681 | 2015-04-30 23:49:42 +0000 | [diff] [blame] | 628 | Options.InstrProfileOutput = CodeGenOpts.InstrProfileOutput; |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 629 | MPM.add(createInstrProfilingLegacyPass(Options)); |
Justin Bogner | 970ac60 | 2014-12-08 19:04:51 +0000 | [diff] [blame] | 630 | } |
Rong Xu | 522b5cb | 2016-02-29 18:54:59 +0000 | [diff] [blame] | 631 | if (CodeGenOpts.hasProfileIRInstr()) { |
Xinliang David Li | b65f8ae | 2016-07-23 04:28:59 +0000 | [diff] [blame] | 632 | PMBuilder.EnablePGOInstrGen = true; |
Rong Xu | 522b5cb | 2016-02-29 18:54:59 +0000 | [diff] [blame] | 633 | if (!CodeGenOpts.InstrProfileOutput.empty()) |
| 634 | PMBuilder.PGOInstrGen = CodeGenOpts.InstrProfileOutput; |
| 635 | else |
Davide Italiano | 945de43 | 2017-02-13 16:07:05 +0000 | [diff] [blame] | 636 | PMBuilder.PGOInstrGen = DefaultProfileGenName; |
Rong Xu | 522b5cb | 2016-02-29 18:54:59 +0000 | [diff] [blame] | 637 | } |
Rong Xu | 9c6f153 | 2016-03-02 20:59:36 +0000 | [diff] [blame] | 638 | if (CodeGenOpts.hasProfileIRUse()) |
| 639 | PMBuilder.PGOInstrUse = CodeGenOpts.ProfileInstrumentUsePath; |
Justin Bogner | 970ac60 | 2014-12-08 19:04:51 +0000 | [diff] [blame] | 640 | |
Dehao Chen | 5717aff | 2016-12-14 21:41:04 +0000 | [diff] [blame] | 641 | if (!CodeGenOpts.SampleProfileFile.empty()) |
| 642 | PMBuilder.PGOSampleUse = CodeGenOpts.SampleProfileFile; |
Diego Novillo | d3ef108 | 2015-08-25 15:25:13 +0000 | [diff] [blame] | 643 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 644 | PMBuilder.populateFunctionPassManager(FPM); |
| 645 | PMBuilder.populateModulePassManager(MPM); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 646 | } |
| 647 | |
David Blaikie | 6e2ec5f | 2017-04-19 20:08:21 +0000 | [diff] [blame] | 648 | static void setCommandLineOpts(const CodeGenOptions &CodeGenOpts) { |
Yaxun Liu | b5e80c3 | 2016-04-12 20:22:32 +0000 | [diff] [blame] | 649 | SmallVector<const char *, 16> BackendArgs; |
| 650 | BackendArgs.push_back("clang"); // Fake program name. |
| 651 | if (!CodeGenOpts.DebugPass.empty()) { |
| 652 | BackendArgs.push_back("-debug-pass"); |
| 653 | BackendArgs.push_back(CodeGenOpts.DebugPass.c_str()); |
| 654 | } |
| 655 | if (!CodeGenOpts.LimitFloatPrecision.empty()) { |
| 656 | BackendArgs.push_back("-limit-float-precision"); |
| 657 | BackendArgs.push_back(CodeGenOpts.LimitFloatPrecision.c_str()); |
| 658 | } |
| 659 | for (const std::string &BackendOption : CodeGenOpts.BackendOptions) |
| 660 | BackendArgs.push_back(BackendOption.c_str()); |
| 661 | BackendArgs.push_back(nullptr); |
| 662 | llvm::cl::ParseCommandLineOptions(BackendArgs.size() - 1, |
| 663 | BackendArgs.data()); |
| 664 | } |
| 665 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 666 | void EmitAssemblyHelper::CreateTargetMachine(bool MustCreateTM) { |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 667 | // Create the TargetMachine for generating code. |
| 668 | std::string Error; |
| 669 | std::string Triple = TheModule->getTargetTriple(); |
| 670 | const llvm::Target *TheTarget = TargetRegistry::lookupTarget(Triple, Error); |
| 671 | if (!TheTarget) { |
Nadav Rotem | dc06b2d | 2012-10-24 03:52:31 +0000 | [diff] [blame] | 672 | if (MustCreateTM) |
Chad Rosier | ecafbe6 | 2013-03-27 00:14:35 +0000 | [diff] [blame] | 673 | Diags.Report(diag::err_fe_unable_to_create_target) << Error; |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 674 | return; |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 675 | } |
| 676 | |
Rafael Espindola | aa0226e | 2017-08-03 02:16:28 +0000 | [diff] [blame] | 677 | Optional<llvm::CodeModel::Model> CM = getCodeModel(CodeGenOpts); |
Eric Christopher | 583a1f7 | 2015-09-26 01:25:08 +0000 | [diff] [blame] | 678 | std::string FeaturesStr = |
| 679 | llvm::join(TargetOpts.Features.begin(), TargetOpts.Features.end(), ","); |
Rafael Espindola | e1d7053 | 2018-01-18 00:20:03 +0000 | [diff] [blame^] | 680 | llvm::Reloc::Model RM = CodeGenOpts.RelocationModel; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 681 | CodeGenOpt::Level OptLevel = getCGOptLevel(CodeGenOpts); |
Evan Cheng | dd286bc | 2011-11-16 08:38:55 +0000 | [diff] [blame] | 682 | |
Nick Lewycky | 432add5 | 2011-12-02 22:17:00 +0000 | [diff] [blame] | 683 | llvm::TargetOptions Options; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 684 | initTargetOptions(Options, CodeGenOpts, TargetOpts, LangOpts, HSOpts); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 685 | TM.reset(TheTarget->createTargetMachine(Triple, TargetOpts.CPU, FeaturesStr, |
| 686 | Options, RM, CM, OptLevel)); |
Nadav Rotem | ec57ab3 | 2012-10-24 00:53:38 +0000 | [diff] [blame] | 687 | } |
| 688 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 689 | bool EmitAssemblyHelper::AddEmitPasses(legacy::PassManager &CodeGenPasses, |
| 690 | BackendAction Action, |
Rafael Espindola | 2f16bc1 | 2015-04-14 15:15:49 +0000 | [diff] [blame] | 691 | raw_pwrite_stream &OS) { |
Chad Rosier | b1cfc68 | 2012-02-29 20:14:59 +0000 | [diff] [blame] | 692 | // Add LibraryInfo. |
Daniel Dunbar | aa437df | 2012-10-19 20:10:10 +0000 | [diff] [blame] | 693 | llvm::Triple TargetTriple(TheModule->getTargetTriple()); |
Chandler Carruth | 57bb7c7 | 2015-01-24 02:25:21 +0000 | [diff] [blame] | 694 | std::unique_ptr<TargetLibraryInfoImpl> TLII( |
| 695 | createTLII(TargetTriple, CodeGenOpts)); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 696 | CodeGenPasses.add(new TargetLibraryInfoWrapperPass(*TLII)); |
Chad Rosier | b1cfc68 | 2012-02-29 20:14:59 +0000 | [diff] [blame] | 697 | |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 698 | // Normal mode, emit a .s or .o file by running the code generator. Note, |
| 699 | // this also adds codegenerator level optimization passes. |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 700 | TargetMachine::CodeGenFileType CGFT = getCodeGenFileType(Action); |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 701 | |
| 702 | // Add ObjC ARC final-cleanup optimizations. This is done as part of the |
| 703 | // "codegen" passes so that it isn't run multiple times when there is |
| 704 | // inlining happening. |
Steven Wu | 1d56be8 | 2015-05-02 00:56:15 +0000 | [diff] [blame] | 705 | if (CodeGenOpts.OptimizationLevel > 0) |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 706 | CodeGenPasses.add(createObjCARCContractPass()); |
Dan Gohman | fec0ff8 | 2011-07-05 22:02:36 +0000 | [diff] [blame] | 707 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 708 | if (TM->addPassesToEmitFile(CodeGenPasses, OS, CGFT, |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 709 | /*DisableVerify=*/!CodeGenOpts.VerifyModule)) { |
| 710 | Diags.Report(diag::err_fe_unable_to_interface_with_target); |
| 711 | return false; |
| 712 | } |
| 713 | |
| 714 | return true; |
| 715 | } |
| 716 | |
Rafael Espindola | 2f16bc1 | 2015-04-14 15:15:49 +0000 | [diff] [blame] | 717 | void EmitAssemblyHelper::EmitAssembly(BackendAction Action, |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 718 | std::unique_ptr<raw_pwrite_stream> OS) { |
Craig Topper | 8a13c41 | 2014-05-21 05:09:00 +0000 | [diff] [blame] | 719 | TimeRegion Region(llvm::TimePassesIsEnabled ? &CodeGenerationTime : nullptr); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 720 | |
David Blaikie | 6e2ec5f | 2017-04-19 20:08:21 +0000 | [diff] [blame] | 721 | setCommandLineOpts(CodeGenOpts); |
Yaxun Liu | b5e80c3 | 2016-04-12 20:22:32 +0000 | [diff] [blame] | 722 | |
Nadav Rotem | dc06b2d | 2012-10-24 03:52:31 +0000 | [diff] [blame] | 723 | bool UsesCodeGen = (Action != Backend_EmitNothing && |
| 724 | Action != Backend_EmitBC && |
| 725 | Action != Backend_EmitLL); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 726 | CreateTargetMachine(UsesCodeGen); |
Alp Toker | f4e2238 | 2013-12-20 20:26:53 +0000 | [diff] [blame] | 727 | |
Rafael Espindola | b633d20 | 2015-06-23 13:59:36 +0000 | [diff] [blame] | 728 | if (UsesCodeGen && !TM) |
| 729 | return; |
| 730 | if (TM) |
Mehdi Amini | ca3cf9e | 2015-07-24 16:04:29 +0000 | [diff] [blame] | 731 | TheModule->setDataLayout(TM->createDataLayout()); |
Teresa Johnson | 4b4f4b9 | 2016-01-08 17:04:29 +0000 | [diff] [blame] | 732 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 733 | legacy::PassManager PerModulePasses; |
| 734 | PerModulePasses.add( |
| 735 | createTargetTransformInfoWrapperPass(getTargetIRAnalysis())); |
| 736 | |
| 737 | legacy::FunctionPassManager PerFunctionPasses(TheModule); |
| 738 | PerFunctionPasses.add( |
| 739 | createTargetTransformInfoWrapperPass(getTargetIRAnalysis())); |
| 740 | |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 741 | CreatePasses(PerModulePasses, PerFunctionPasses); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 742 | |
| 743 | legacy::PassManager CodeGenPasses; |
| 744 | CodeGenPasses.add( |
| 745 | createTargetTransformInfoWrapperPass(getTargetIRAnalysis())); |
Nadav Rotem | ec57ab3 | 2012-10-24 00:53:38 +0000 | [diff] [blame] | 746 | |
Teresa Johnson | 488d1dc | 2017-03-23 19:47:49 +0000 | [diff] [blame] | 747 | std::unique_ptr<raw_fd_ostream> ThinLinkOS; |
| 748 | |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 749 | switch (Action) { |
| 750 | case Backend_EmitNothing: |
| 751 | break; |
| 752 | |
| 753 | case Backend_EmitBC: |
Teresa Johnson | 488d1dc | 2017-03-23 19:47:49 +0000 | [diff] [blame] | 754 | if (CodeGenOpts.EmitSummaryIndex) { |
| 755 | if (!CodeGenOpts.ThinLinkBitcodeFile.empty()) { |
| 756 | std::error_code EC; |
| 757 | ThinLinkOS.reset(new llvm::raw_fd_ostream( |
| 758 | CodeGenOpts.ThinLinkBitcodeFile, EC, |
| 759 | llvm::sys::fs::F_None)); |
| 760 | if (EC) { |
| 761 | Diags.Report(diag::err_fe_unable_to_open_output) << CodeGenOpts.ThinLinkBitcodeFile |
| 762 | << EC.message(); |
| 763 | return; |
| 764 | } |
| 765 | } |
| 766 | PerModulePasses.add( |
| 767 | createWriteThinLTOBitcodePass(*OS, ThinLinkOS.get())); |
| 768 | } |
Peter Collingbourne | b884716 | 2017-01-26 23:51:50 +0000 | [diff] [blame] | 769 | else |
| 770 | PerModulePasses.add( |
| 771 | createBitcodeWriterPass(*OS, CodeGenOpts.EmitLLVMUseLists)); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 772 | break; |
| 773 | |
| 774 | case Backend_EmitLL: |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 775 | PerModulePasses.add( |
Duncan P. N. Exon Smith | bb9cadf | 2015-04-15 02:45:28 +0000 | [diff] [blame] | 776 | createPrintModulePass(*OS, "", CodeGenOpts.EmitLLVMUseLists)); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 777 | break; |
| 778 | |
| 779 | default: |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 780 | if (!AddEmitPasses(CodeGenPasses, Action, *OS)) |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 781 | return; |
| 782 | } |
| 783 | |
Andrew Trick | 15e36e8 | 2011-04-05 18:56:55 +0000 | [diff] [blame] | 784 | // Before executing passes, print the final values of the LLVM options. |
| 785 | cl::PrintOptionValues(); |
| 786 | |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 787 | // Run passes. For now we do all passes at once, but eventually we |
| 788 | // would like to have the option of streaming code generation. |
| 789 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 790 | { |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 791 | PrettyStackTraceString CrashInfo("Per-function optimization"); |
| 792 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 793 | PerFunctionPasses.doInitialization(); |
Yaron Keren | 10d6d16 | 2015-06-05 09:40:53 +0000 | [diff] [blame] | 794 | for (Function &F : *TheModule) |
| 795 | if (!F.isDeclaration()) |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 796 | PerFunctionPasses.run(F); |
| 797 | PerFunctionPasses.doFinalization(); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 798 | } |
| 799 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 800 | { |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 801 | PrettyStackTraceString CrashInfo("Per-module optimization passes"); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 802 | PerModulePasses.run(*TheModule); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 803 | } |
| 804 | |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 805 | { |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 806 | PrettyStackTraceString CrashInfo("Code generation"); |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 807 | CodeGenPasses.run(*TheModule); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 808 | } |
| 809 | } |
| 810 | |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 811 | static PassBuilder::OptimizationLevel mapToLevel(const CodeGenOptions &Opts) { |
| 812 | switch (Opts.OptimizationLevel) { |
| 813 | default: |
| 814 | llvm_unreachable("Invalid optimization level!"); |
| 815 | |
| 816 | case 1: |
| 817 | return PassBuilder::O1; |
| 818 | |
| 819 | case 2: |
| 820 | switch (Opts.OptimizeSize) { |
| 821 | default: |
| 822 | llvm_unreachable("Invalide optimization level for size!"); |
| 823 | |
| 824 | case 0: |
| 825 | return PassBuilder::O2; |
| 826 | |
| 827 | case 1: |
| 828 | return PassBuilder::Os; |
| 829 | |
| 830 | case 2: |
| 831 | return PassBuilder::Oz; |
| 832 | } |
| 833 | |
| 834 | case 3: |
| 835 | return PassBuilder::O3; |
| 836 | } |
| 837 | } |
| 838 | |
| 839 | /// A clean version of `EmitAssembly` that uses the new pass manager. |
| 840 | /// |
| 841 | /// Not all features are currently supported in this system, but where |
| 842 | /// necessary it falls back to the legacy pass manager to at least provide |
| 843 | /// basic functionality. |
| 844 | /// |
| 845 | /// This API is planned to have its functionality finished and then to replace |
| 846 | /// `EmitAssembly` at some point in the future when the default switches. |
| 847 | void EmitAssemblyHelper::EmitAssemblyWithNewPassManager( |
| 848 | BackendAction Action, std::unique_ptr<raw_pwrite_stream> OS) { |
| 849 | TimeRegion Region(llvm::TimePassesIsEnabled ? &CodeGenerationTime : nullptr); |
David Blaikie | 6e2ec5f | 2017-04-19 20:08:21 +0000 | [diff] [blame] | 850 | setCommandLineOpts(CodeGenOpts); |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 851 | |
| 852 | // The new pass manager always makes a target machine available to passes |
| 853 | // during construction. |
| 854 | CreateTargetMachine(/*MustCreateTM*/ true); |
| 855 | if (!TM) |
| 856 | // This will already be diagnosed, just bail. |
| 857 | return; |
| 858 | TheModule->setDataLayout(TM->createDataLayout()); |
| 859 | |
Dehao Chen | c76a27e | 2017-07-27 15:29:53 +0000 | [diff] [blame] | 860 | Optional<PGOOptions> PGOOpt; |
Davide Italiano | 945de43 | 2017-02-13 16:07:05 +0000 | [diff] [blame] | 861 | |
Dehao Chen | c76a27e | 2017-07-27 15:29:53 +0000 | [diff] [blame] | 862 | if (CodeGenOpts.hasProfileIRInstr()) |
| 863 | // -fprofile-generate. |
| 864 | PGOOpt = PGOOptions(CodeGenOpts.InstrProfileOutput.empty() |
| 865 | ? DefaultProfileGenName |
| 866 | : CodeGenOpts.InstrProfileOutput, |
| 867 | "", "", true, CodeGenOpts.DebugInfoForProfiling); |
| 868 | else if (CodeGenOpts.hasProfileIRUse()) |
| 869 | // -fprofile-use. |
| 870 | PGOOpt = PGOOptions("", CodeGenOpts.ProfileInstrumentUsePath, "", false, |
| 871 | CodeGenOpts.DebugInfoForProfiling); |
| 872 | else if (!CodeGenOpts.SampleProfileFile.empty()) |
| 873 | // -fprofile-sample-use |
| 874 | PGOOpt = PGOOptions("", "", CodeGenOpts.SampleProfileFile, false, |
| 875 | CodeGenOpts.DebugInfoForProfiling); |
| 876 | else if (CodeGenOpts.DebugInfoForProfiling) |
| 877 | // -fdebug-info-for-profiling |
| 878 | PGOOpt = PGOOptions("", "", "", false, true); |
Davide Italiano | 945de43 | 2017-02-13 16:07:05 +0000 | [diff] [blame] | 879 | |
Dehao Chen | c76a27e | 2017-07-27 15:29:53 +0000 | [diff] [blame] | 880 | PassBuilder PB(TM.get(), PGOOpt); |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 881 | |
Craig Topper | 926b95c | 2017-11-14 08:48:28 +0000 | [diff] [blame] | 882 | LoopAnalysisManager LAM(CodeGenOpts.DebugPassManager); |
| 883 | FunctionAnalysisManager FAM(CodeGenOpts.DebugPassManager); |
| 884 | CGSCCAnalysisManager CGAM(CodeGenOpts.DebugPassManager); |
| 885 | ModuleAnalysisManager MAM(CodeGenOpts.DebugPassManager); |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 886 | |
| 887 | // Register the AA manager first so that our version is the one used. |
| 888 | FAM.registerPass([&] { return PB.buildDefaultAAPipeline(); }); |
| 889 | |
Chandler Carruth | 9fdd5fa | 2017-07-25 10:46:07 +0000 | [diff] [blame] | 890 | // Register the target library analysis directly and give it a customized |
| 891 | // preset TLI. |
| 892 | Triple TargetTriple(TheModule->getTargetTriple()); |
| 893 | std::unique_ptr<TargetLibraryInfoImpl> TLII( |
| 894 | createTLII(TargetTriple, CodeGenOpts)); |
| 895 | FAM.registerPass([&] { return TargetLibraryAnalysis(*TLII); }); |
| 896 | MAM.registerPass([&] { return TargetLibraryAnalysis(*TLII); }); |
| 897 | |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 898 | // Register all the basic analyses with the managers. |
| 899 | PB.registerModuleAnalyses(MAM); |
| 900 | PB.registerCGSCCAnalyses(CGAM); |
| 901 | PB.registerFunctionAnalyses(FAM); |
| 902 | PB.registerLoopAnalyses(LAM); |
| 903 | PB.crossRegisterProxies(LAM, FAM, CGAM, MAM); |
| 904 | |
Tim Shen | b13eebe | 2017-06-29 23:10:13 +0000 | [diff] [blame] | 905 | ModulePassManager MPM(CodeGenOpts.DebugPassManager); |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 906 | |
Chandler Carruth | 6d1b83e | 2016-12-27 00:13:09 +0000 | [diff] [blame] | 907 | if (!CodeGenOpts.DisableLLVMPasses) { |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 908 | bool IsThinLTO = CodeGenOpts.EmitSummaryIndex; |
| 909 | bool IsLTO = CodeGenOpts.PrepareForLTO; |
| 910 | |
Chandler Carruth | 6d1b83e | 2016-12-27 00:13:09 +0000 | [diff] [blame] | 911 | if (CodeGenOpts.OptimizationLevel == 0) { |
| 912 | // Build a minimal pipeline based on the semantics required by Clang, |
| 913 | // which is just that always inlining occurs. |
| 914 | MPM.addPass(AlwaysInlinerPass()); |
Chandler Carruth | a8bd4e3 | 2017-11-14 01:59:18 +0000 | [diff] [blame] | 915 | |
| 916 | // At -O0 we directly run necessary sanitizer passes. |
| 917 | if (LangOpts.Sanitize.has(SanitizerKind::LocalBounds)) |
| 918 | MPM.addPass(createModuleToFunctionPassAdaptor(BoundsCheckingPass())); |
| 919 | |
| 920 | // Lastly, add a semantically necessary pass for ThinLTO. |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 921 | if (IsThinLTO) |
| 922 | MPM.addPass(NameAnonGlobalPass()); |
Chandler Carruth | 6d1b83e | 2016-12-27 00:13:09 +0000 | [diff] [blame] | 923 | } else { |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 924 | // Map our optimization levels into one of the distinct levels used to |
| 925 | // configure the pipeline. |
Chandler Carruth | 6d1b83e | 2016-12-27 00:13:09 +0000 | [diff] [blame] | 926 | PassBuilder::OptimizationLevel Level = mapToLevel(CodeGenOpts); |
| 927 | |
Chandler Carruth | a8bd4e3 | 2017-11-14 01:59:18 +0000 | [diff] [blame] | 928 | // Register callbacks to schedule sanitizer passes at the appropriate part of |
| 929 | // the pipeline. |
| 930 | if (LangOpts.Sanitize.has(SanitizerKind::LocalBounds)) |
| 931 | PB.registerScalarOptimizerLateEPCallback( |
| 932 | [](FunctionPassManager &FPM, PassBuilder::OptimizationLevel Level) { |
| 933 | FPM.addPass(BoundsCheckingPass()); |
| 934 | }); |
| 935 | |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 936 | if (IsThinLTO) { |
Tim Shen | b13eebe | 2017-06-29 23:10:13 +0000 | [diff] [blame] | 937 | MPM = PB.buildThinLTOPreLinkDefaultPipeline( |
| 938 | Level, CodeGenOpts.DebugPassManager); |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 939 | MPM.addPass(NameAnonGlobalPass()); |
| 940 | } else if (IsLTO) { |
Tim Shen | b13eebe | 2017-06-29 23:10:13 +0000 | [diff] [blame] | 941 | MPM = PB.buildLTOPreLinkDefaultPipeline(Level, |
| 942 | CodeGenOpts.DebugPassManager); |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 943 | } else { |
Tim Shen | b13eebe | 2017-06-29 23:10:13 +0000 | [diff] [blame] | 944 | MPM = PB.buildPerModuleDefaultPipeline(Level, |
| 945 | CodeGenOpts.DebugPassManager); |
Tim Shen | 6647069 | 2017-06-29 23:08:38 +0000 | [diff] [blame] | 946 | } |
Chandler Carruth | 6d1b83e | 2016-12-27 00:13:09 +0000 | [diff] [blame] | 947 | } |
David Blaikie | 7a4f7f5 | 2018-01-09 22:03:47 +0000 | [diff] [blame] | 948 | if (Optional<GCOVOptions> Options = getGCOVOptions(CodeGenOpts)) { |
| 949 | MPM.addPass(GCOVProfilerPass(*Options)); |
| 950 | } |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 951 | } |
| 952 | |
| 953 | // FIXME: We still use the legacy pass manager to do code generation. We |
| 954 | // create that pass manager here and use it as needed below. |
| 955 | legacy::PassManager CodeGenPasses; |
| 956 | bool NeedCodeGen = false; |
Tim Shen | 50fedec | 2017-06-01 23:27:51 +0000 | [diff] [blame] | 957 | Optional<raw_fd_ostream> ThinLinkOS; |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 958 | |
| 959 | // Append any output we need to the pass manager. |
| 960 | switch (Action) { |
| 961 | case Backend_EmitNothing: |
| 962 | break; |
| 963 | |
| 964 | case Backend_EmitBC: |
Tim Shen | 50fedec | 2017-06-01 23:27:51 +0000 | [diff] [blame] | 965 | if (CodeGenOpts.EmitSummaryIndex) { |
| 966 | if (!CodeGenOpts.ThinLinkBitcodeFile.empty()) { |
| 967 | std::error_code EC; |
| 968 | ThinLinkOS.emplace(CodeGenOpts.ThinLinkBitcodeFile, EC, |
| 969 | llvm::sys::fs::F_None); |
| 970 | if (EC) { |
| 971 | Diags.Report(diag::err_fe_unable_to_open_output) |
| 972 | << CodeGenOpts.ThinLinkBitcodeFile << EC.message(); |
| 973 | return; |
| 974 | } |
| 975 | } |
| 976 | MPM.addPass( |
| 977 | ThinLTOBitcodeWriterPass(*OS, ThinLinkOS ? &*ThinLinkOS : nullptr)); |
| 978 | } else { |
| 979 | MPM.addPass(BitcodeWriterPass(*OS, CodeGenOpts.EmitLLVMUseLists, |
| 980 | CodeGenOpts.EmitSummaryIndex, |
| 981 | CodeGenOpts.EmitSummaryIndex)); |
| 982 | } |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 983 | break; |
| 984 | |
| 985 | case Backend_EmitLL: |
| 986 | MPM.addPass(PrintModulePass(*OS, "", CodeGenOpts.EmitLLVMUseLists)); |
| 987 | break; |
| 988 | |
| 989 | case Backend_EmitAssembly: |
| 990 | case Backend_EmitMCNull: |
| 991 | case Backend_EmitObj: |
| 992 | NeedCodeGen = true; |
| 993 | CodeGenPasses.add( |
| 994 | createTargetTransformInfoWrapperPass(getTargetIRAnalysis())); |
| 995 | if (!AddEmitPasses(CodeGenPasses, Action, *OS)) |
| 996 | // FIXME: Should we handle this error differently? |
| 997 | return; |
| 998 | break; |
| 999 | } |
| 1000 | |
| 1001 | // Before executing passes, print the final values of the LLVM options. |
| 1002 | cl::PrintOptionValues(); |
| 1003 | |
| 1004 | // Now that we have all of the passes ready, run them. |
| 1005 | { |
| 1006 | PrettyStackTraceString CrashInfo("Optimizer"); |
| 1007 | MPM.run(*TheModule, MAM); |
| 1008 | } |
| 1009 | |
| 1010 | // Now if needed, run the legacy PM for codegen. |
| 1011 | if (NeedCodeGen) { |
| 1012 | PrettyStackTraceString CrashInfo("Code generation"); |
| 1013 | CodeGenPasses.run(*TheModule); |
| 1014 | } |
| 1015 | } |
| 1016 | |
Peter Collingbourne | 47d2364 | 2017-01-24 19:54:37 +0000 | [diff] [blame] | 1017 | Expected<BitcodeModule> clang::FindThinLTOModule(MemoryBufferRef MBRef) { |
| 1018 | Expected<std::vector<BitcodeModule>> BMsOrErr = getBitcodeModuleList(MBRef); |
| 1019 | if (!BMsOrErr) |
| 1020 | return BMsOrErr.takeError(); |
| 1021 | |
Peter Collingbourne | dbd2fed | 2017-06-15 17:26:13 +0000 | [diff] [blame] | 1022 | // The bitcode file may contain multiple modules, we want the one that is |
| 1023 | // marked as being the ThinLTO module. |
Peter Collingbourne | 47d2364 | 2017-01-24 19:54:37 +0000 | [diff] [blame] | 1024 | for (BitcodeModule &BM : *BMsOrErr) { |
Peter Collingbourne | dbd2fed | 2017-06-15 17:26:13 +0000 | [diff] [blame] | 1025 | Expected<BitcodeLTOInfo> LTOInfo = BM.getLTOInfo(); |
| 1026 | if (LTOInfo && LTOInfo->IsThinLTO) |
Peter Collingbourne | 47d2364 | 2017-01-24 19:54:37 +0000 | [diff] [blame] | 1027 | return BM; |
| 1028 | } |
| 1029 | |
| 1030 | return make_error<StringError>("Could not find module summary", |
| 1031 | inconvertibleErrorCode()); |
| 1032 | } |
| 1033 | |
Teresa Johnson | cffeb54 | 2017-01-06 23:37:33 +0000 | [diff] [blame] | 1034 | static void runThinLTOBackend(ModuleSummaryIndex *CombinedIndex, Module *M, |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1035 | const HeaderSearchOptions &HeaderOpts, |
| 1036 | const CodeGenOptions &CGOpts, |
| 1037 | const clang::TargetOptions &TOpts, |
| 1038 | const LangOptions &LOpts, |
Dehao Chen | a1bd2d6 | 2017-01-13 00:51:55 +0000 | [diff] [blame] | 1039 | std::unique_ptr<raw_pwrite_stream> OS, |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1040 | std::string SampleProfile, |
| 1041 | BackendAction Action) { |
Dehao Chen | 5f83d0e | 2017-07-10 20:31:37 +0000 | [diff] [blame] | 1042 | StringMap<DenseMap<GlobalValue::GUID, GlobalValueSummary *>> |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1043 | ModuleToDefinedGVSummaries; |
| 1044 | CombinedIndex->collectDefinedGVSummariesPerModule(ModuleToDefinedGVSummaries); |
| 1045 | |
David Blaikie | 6e2ec5f | 2017-04-19 20:08:21 +0000 | [diff] [blame] | 1046 | setCommandLineOpts(CGOpts); |
| 1047 | |
Teresa Johnson | f9b17d4 | 2016-12-28 18:00:08 +0000 | [diff] [blame] | 1048 | // We can simply import the values mentioned in the combined index, since |
| 1049 | // we should only invoke this using the individual indexes written out |
| 1050 | // via a WriteIndexesThinBackend. |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1051 | FunctionImporter::ImportMapTy ImportList; |
Teresa Johnson | f9b17d4 | 2016-12-28 18:00:08 +0000 | [diff] [blame] | 1052 | for (auto &GlobalList : *CombinedIndex) { |
Peter Collingbourne | 9667b91 | 2017-05-04 18:03:25 +0000 | [diff] [blame] | 1053 | // Ignore entries for undefined references. |
| 1054 | if (GlobalList.second.SummaryList.empty()) |
| 1055 | continue; |
| 1056 | |
Teresa Johnson | f9b17d4 | 2016-12-28 18:00:08 +0000 | [diff] [blame] | 1057 | auto GUID = GlobalList.first; |
Peter Collingbourne | 9667b91 | 2017-05-04 18:03:25 +0000 | [diff] [blame] | 1058 | assert(GlobalList.second.SummaryList.size() == 1 && |
Teresa Johnson | f9b17d4 | 2016-12-28 18:00:08 +0000 | [diff] [blame] | 1059 | "Expected individual combined index to have one summary per GUID"); |
Peter Collingbourne | 9667b91 | 2017-05-04 18:03:25 +0000 | [diff] [blame] | 1060 | auto &Summary = GlobalList.second.SummaryList[0]; |
Teresa Johnson | f9b17d4 | 2016-12-28 18:00:08 +0000 | [diff] [blame] | 1061 | // Skip the summaries for the importing module. These are included to |
| 1062 | // e.g. record required linkage changes. |
| 1063 | if (Summary->modulePath() == M->getModuleIdentifier()) |
| 1064 | continue; |
| 1065 | // Doesn't matter what value we plug in to the map, just needs an entry |
| 1066 | // to provoke importing by thinBackend. |
| 1067 | ImportList[Summary->modulePath()][GUID] = 1; |
| 1068 | } |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1069 | |
| 1070 | std::vector<std::unique_ptr<llvm::MemoryBuffer>> OwnedImports; |
Peter Collingbourne | 1a0720e | 2016-12-14 01:17:59 +0000 | [diff] [blame] | 1071 | MapVector<llvm::StringRef, llvm::BitcodeModule> ModuleMap; |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1072 | |
| 1073 | for (auto &I : ImportList) { |
| 1074 | ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> MBOrErr = |
| 1075 | llvm::MemoryBuffer::getFile(I.first()); |
| 1076 | if (!MBOrErr) { |
| 1077 | errs() << "Error loading imported file '" << I.first() |
| 1078 | << "': " << MBOrErr.getError().message() << "\n"; |
| 1079 | return; |
| 1080 | } |
Peter Collingbourne | 1a0720e | 2016-12-14 01:17:59 +0000 | [diff] [blame] | 1081 | |
Peter Collingbourne | 47d2364 | 2017-01-24 19:54:37 +0000 | [diff] [blame] | 1082 | Expected<BitcodeModule> BMOrErr = FindThinLTOModule(**MBOrErr); |
| 1083 | if (!BMOrErr) { |
| 1084 | handleAllErrors(BMOrErr.takeError(), [&](ErrorInfoBase &EIB) { |
Peter Collingbourne | 1a0720e | 2016-12-14 01:17:59 +0000 | [diff] [blame] | 1085 | errs() << "Error loading imported file '" << I.first() |
| 1086 | << "': " << EIB.message() << '\n'; |
| 1087 | }); |
| 1088 | return; |
| 1089 | } |
Peter Collingbourne | 47d2364 | 2017-01-24 19:54:37 +0000 | [diff] [blame] | 1090 | ModuleMap.insert({I.first(), *BMOrErr}); |
Peter Collingbourne | 1a0720e | 2016-12-14 01:17:59 +0000 | [diff] [blame] | 1091 | |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1092 | OwnedImports.push_back(std::move(*MBOrErr)); |
| 1093 | } |
Peter Collingbourne | 2d3a26f | 2016-09-23 21:43:51 +0000 | [diff] [blame] | 1094 | auto AddStream = [&](size_t Task) { |
| 1095 | return llvm::make_unique<lto::NativeObjectStream>(std::move(OS)); |
Mehdi Amini | 406aa22 | 2016-08-17 06:23:08 +0000 | [diff] [blame] | 1096 | }; |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1097 | lto::Config Conf; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1098 | Conf.CPU = TOpts.CPU; |
| 1099 | Conf.CodeModel = getCodeModel(CGOpts); |
| 1100 | Conf.MAttrs = TOpts.Features; |
Rafael Espindola | e1d7053 | 2018-01-18 00:20:03 +0000 | [diff] [blame^] | 1101 | Conf.RelocModel = CGOpts.RelocationModel; |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1102 | Conf.CGOptLevel = getCGOptLevel(CGOpts); |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1103 | initTargetOptions(Conf.Options, CGOpts, TOpts, LOpts, HeaderOpts); |
Benjamin Kramer | f6021ec | 2017-03-21 21:35:04 +0000 | [diff] [blame] | 1104 | Conf.SampleProfile = std::move(SampleProfile); |
Tim Shen | 50fedec | 2017-06-01 23:27:51 +0000 | [diff] [blame] | 1105 | Conf.UseNewPM = CGOpts.ExperimentalNewPassManager; |
Teresa Johnson | 4cd016ab | 2017-11-13 15:38:33 +0000 | [diff] [blame] | 1106 | Conf.DebugPassManager = CGOpts.DebugPassManager; |
Teresa Johnson | b637cb0 | 2017-03-31 22:35:47 +0000 | [diff] [blame] | 1107 | switch (Action) { |
| 1108 | case Backend_EmitNothing: |
| 1109 | Conf.PreCodeGenModuleHook = [](size_t Task, const Module &Mod) { |
| 1110 | return false; |
| 1111 | }; |
| 1112 | break; |
| 1113 | case Backend_EmitLL: |
| 1114 | Conf.PreCodeGenModuleHook = [&](size_t Task, const Module &Mod) { |
| 1115 | M->print(*OS, nullptr, CGOpts.EmitLLVMUseLists); |
| 1116 | return false; |
| 1117 | }; |
| 1118 | break; |
| 1119 | case Backend_EmitBC: |
| 1120 | Conf.PreCodeGenModuleHook = [&](size_t Task, const Module &Mod) { |
| 1121 | WriteBitcodeToFile(M, *OS, CGOpts.EmitLLVMUseLists); |
| 1122 | return false; |
| 1123 | }; |
| 1124 | break; |
| 1125 | default: |
| 1126 | Conf.CGFileType = getCodeGenFileType(Action); |
| 1127 | break; |
| 1128 | } |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1129 | if (Error E = thinBackend( |
Peter Collingbourne | 2d3a26f | 2016-09-23 21:43:51 +0000 | [diff] [blame] | 1130 | Conf, 0, AddStream, *M, *CombinedIndex, ImportList, |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1131 | ModuleToDefinedGVSummaries[M->getModuleIdentifier()], ModuleMap)) { |
| 1132 | handleAllErrors(std::move(E), [&](ErrorInfoBase &EIB) { |
| 1133 | errs() << "Error running ThinLTO backend: " << EIB.message() << '\n'; |
| 1134 | }); |
| 1135 | } |
| 1136 | } |
| 1137 | |
David Blaikie | 9c902b5 | 2011-09-25 23:23:43 +0000 | [diff] [blame] | 1138 | void clang::EmitBackendOutput(DiagnosticsEngine &Diags, |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 1139 | const HeaderSearchOptions &HeaderOpts, |
David Blaikie | 9c902b5 | 2011-09-25 23:23:43 +0000 | [diff] [blame] | 1140 | const CodeGenOptions &CGOpts, |
Nick Lewycky | 432add5 | 2011-12-02 22:17:00 +0000 | [diff] [blame] | 1141 | const clang::TargetOptions &TOpts, |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 1142 | const LangOptions &LOpts, |
| 1143 | const llvm::DataLayout &TDesc, Module *M, |
| 1144 | BackendAction Action, |
Peter Collingbourne | 03f8907 | 2016-07-15 00:55:40 +0000 | [diff] [blame] | 1145 | std::unique_ptr<raw_pwrite_stream> OS) { |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1146 | if (!CGOpts.ThinLTOIndexFile.empty()) { |
Teresa Johnson | cffeb54 | 2017-01-06 23:37:33 +0000 | [diff] [blame] | 1147 | // If we are performing a ThinLTO importing compile, load the function index |
| 1148 | // into memory and pass it into runThinLTOBackend, which will run the |
| 1149 | // function importer and invoke LTO passes. |
| 1150 | Expected<std::unique_ptr<ModuleSummaryIndex>> IndexOrErr = |
Teresa Johnson | 517729f | 2017-05-12 19:32:17 +0000 | [diff] [blame] | 1151 | llvm::getModuleSummaryIndexForFile(CGOpts.ThinLTOIndexFile, |
| 1152 | /*IgnoreEmptyThinLTOIndexFile*/true); |
Teresa Johnson | cffeb54 | 2017-01-06 23:37:33 +0000 | [diff] [blame] | 1153 | if (!IndexOrErr) { |
| 1154 | logAllUnhandledErrors(IndexOrErr.takeError(), errs(), |
| 1155 | "Error loading index file '" + |
| 1156 | CGOpts.ThinLTOIndexFile + "': "); |
| 1157 | return; |
| 1158 | } |
| 1159 | std::unique_ptr<ModuleSummaryIndex> CombinedIndex = std::move(*IndexOrErr); |
| 1160 | // A null CombinedIndex means we should skip ThinLTO compilation |
| 1161 | // (LLVM will optionally ignore empty index files, returning null instead |
| 1162 | // of an error). |
| 1163 | bool DoThinLTOBackend = CombinedIndex != nullptr; |
| 1164 | if (DoThinLTOBackend) { |
Teresa Johnson | 5ed6c10 | 2017-03-31 02:05:15 +0000 | [diff] [blame] | 1165 | runThinLTOBackend(CombinedIndex.get(), M, HeaderOpts, CGOpts, TOpts, |
| 1166 | LOpts, std::move(OS), CGOpts.SampleProfileFile, Action); |
Teresa Johnson | cffeb54 | 2017-01-06 23:37:33 +0000 | [diff] [blame] | 1167 | return; |
| 1168 | } |
Teresa Johnson | 9e3f474 | 2016-08-12 18:12:08 +0000 | [diff] [blame] | 1169 | } |
| 1170 | |
Saleem Abdulrasool | 888e289 | 2017-01-05 16:02:32 +0000 | [diff] [blame] | 1171 | EmitAssemblyHelper AsmHelper(Diags, HeaderOpts, CGOpts, TOpts, LOpts, M); |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 1172 | |
Chandler Carruth | 50f9e89 | 2016-12-23 20:44:01 +0000 | [diff] [blame] | 1173 | if (CGOpts.ExperimentalNewPassManager) |
| 1174 | AsmHelper.EmitAssemblyWithNewPassManager(Action, std::move(OS)); |
| 1175 | else |
| 1176 | AsmHelper.EmitAssembly(Action, std::move(OS)); |
Alp Toker | e83b906 | 2014-01-02 15:08:04 +0000 | [diff] [blame] | 1177 | |
James Y Knight | b214cbc | 2016-03-04 19:00:41 +0000 | [diff] [blame] | 1178 | // Verify clang's TargetInfo DataLayout against the LLVM TargetMachine's |
| 1179 | // DataLayout. |
| 1180 | if (AsmHelper.TM) { |
Mehdi Amini | ca3cf9e | 2015-07-24 16:04:29 +0000 | [diff] [blame] | 1181 | std::string DLDesc = M->getDataLayout().getStringRepresentation(); |
James Y Knight | b214cbc | 2016-03-04 19:00:41 +0000 | [diff] [blame] | 1182 | if (DLDesc != TDesc.getStringRepresentation()) { |
Alp Toker | e83b906 | 2014-01-02 15:08:04 +0000 | [diff] [blame] | 1183 | unsigned DiagID = Diags.getCustomDiagID( |
| 1184 | DiagnosticsEngine::Error, "backend data layout '%0' does not match " |
| 1185 | "expected target description '%1'"); |
James Y Knight | b214cbc | 2016-03-04 19:00:41 +0000 | [diff] [blame] | 1186 | Diags.Report(DiagID) << DLDesc << TDesc.getStringRepresentation(); |
Alp Toker | e83b906 | 2014-01-02 15:08:04 +0000 | [diff] [blame] | 1187 | } |
| 1188 | } |
Daniel Dunbar | f976d1b | 2010-06-07 23:20:08 +0000 | [diff] [blame] | 1189 | } |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1190 | |
| 1191 | static const char* getSectionNameForBitcode(const Triple &T) { |
| 1192 | switch (T.getObjectFormat()) { |
| 1193 | case Triple::MachO: |
| 1194 | return "__LLVM,__bitcode"; |
| 1195 | case Triple::COFF: |
| 1196 | case Triple::ELF: |
Dan Gohman | 839f215 | 2017-01-17 21:46:38 +0000 | [diff] [blame] | 1197 | case Triple::Wasm: |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1198 | case Triple::UnknownObjectFormat: |
| 1199 | return ".llvmbc"; |
| 1200 | } |
Simon Pilgrim | bff9dae | 2016-05-11 21:55:37 +0000 | [diff] [blame] | 1201 | llvm_unreachable("Unimplemented ObjectFormatType"); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1202 | } |
| 1203 | |
| 1204 | static const char* getSectionNameForCommandline(const Triple &T) { |
| 1205 | switch (T.getObjectFormat()) { |
| 1206 | case Triple::MachO: |
| 1207 | return "__LLVM,__cmdline"; |
| 1208 | case Triple::COFF: |
| 1209 | case Triple::ELF: |
Dan Gohman | 839f215 | 2017-01-17 21:46:38 +0000 | [diff] [blame] | 1210 | case Triple::Wasm: |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1211 | case Triple::UnknownObjectFormat: |
| 1212 | return ".llvmcmd"; |
| 1213 | } |
Simon Pilgrim | bff9dae | 2016-05-11 21:55:37 +0000 | [diff] [blame] | 1214 | llvm_unreachable("Unimplemented ObjectFormatType"); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1215 | } |
| 1216 | |
| 1217 | // With -fembed-bitcode, save a copy of the llvm IR as data in the |
| 1218 | // __LLVM,__bitcode section. |
| 1219 | void clang::EmbedBitcode(llvm::Module *M, const CodeGenOptions &CGOpts, |
| 1220 | llvm::MemoryBufferRef Buf) { |
| 1221 | if (CGOpts.getEmbedBitcode() == CodeGenOptions::Embed_Off) |
| 1222 | return; |
| 1223 | |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1224 | // Save llvm.compiler.used and remote it. |
| 1225 | SmallVector<Constant*, 2> UsedArray; |
| 1226 | SmallSet<GlobalValue*, 4> UsedGlobals; |
| 1227 | Type *UsedElementType = Type::getInt8Ty(M->getContext())->getPointerTo(0); |
| 1228 | GlobalVariable *Used = collectUsedGlobalVariables(*M, UsedGlobals, true); |
| 1229 | for (auto *GV : UsedGlobals) { |
| 1230 | if (GV->getName() != "llvm.embedded.module" && |
| 1231 | GV->getName() != "llvm.cmdline") |
| 1232 | UsedArray.push_back( |
| 1233 | ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); |
| 1234 | } |
| 1235 | if (Used) |
| 1236 | Used->eraseFromParent(); |
| 1237 | |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1238 | // Embed the bitcode for the llvm module. |
| 1239 | std::string Data; |
| 1240 | ArrayRef<uint8_t> ModuleData; |
| 1241 | Triple T(M->getTargetTriple()); |
| 1242 | // Create a constant that contains the bitcode. |
| 1243 | // In case of embedding a marker, ignore the input Buf and use the empty |
| 1244 | // ArrayRef. It is also legal to create a bitcode marker even Buf is empty. |
| 1245 | if (CGOpts.getEmbedBitcode() != CodeGenOptions::Embed_Marker) { |
| 1246 | if (!isBitcode((const unsigned char *)Buf.getBufferStart(), |
| 1247 | (const unsigned char *)Buf.getBufferEnd())) { |
| 1248 | // If the input is LLVM Assembly, bitcode is produced by serializing |
| 1249 | // the module. Use-lists order need to be perserved in this case. |
| 1250 | llvm::raw_string_ostream OS(Data); |
| 1251 | llvm::WriteBitcodeToFile(M, OS, /* ShouldPreserveUseListOrder */ true); |
| 1252 | ModuleData = |
| 1253 | ArrayRef<uint8_t>((const uint8_t *)OS.str().data(), OS.str().size()); |
| 1254 | } else |
| 1255 | // If the input is LLVM bitcode, write the input byte stream directly. |
| 1256 | ModuleData = ArrayRef<uint8_t>((const uint8_t *)Buf.getBufferStart(), |
| 1257 | Buf.getBufferSize()); |
| 1258 | } |
| 1259 | llvm::Constant *ModuleConstant = |
| 1260 | llvm::ConstantDataArray::get(M->getContext(), ModuleData); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1261 | llvm::GlobalVariable *GV = new llvm::GlobalVariable( |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1262 | *M, ModuleConstant->getType(), true, llvm::GlobalValue::PrivateLinkage, |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1263 | ModuleConstant); |
| 1264 | GV->setSection(getSectionNameForBitcode(T)); |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1265 | UsedArray.push_back( |
| 1266 | ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1267 | if (llvm::GlobalVariable *Old = |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1268 | M->getGlobalVariable("llvm.embedded.module", true)) { |
| 1269 | assert(Old->hasOneUse() && |
| 1270 | "llvm.embedded.module can only be used once in llvm.compiler.used"); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1271 | GV->takeName(Old); |
| 1272 | Old->eraseFromParent(); |
| 1273 | } else { |
| 1274 | GV->setName("llvm.embedded.module"); |
| 1275 | } |
| 1276 | |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1277 | // Skip if only bitcode needs to be embedded. |
| 1278 | if (CGOpts.getEmbedBitcode() != CodeGenOptions::Embed_Bitcode) { |
| 1279 | // Embed command-line options. |
Saleem Abdulrasool | b1d50c1 | 2016-05-20 03:58:12 +0000 | [diff] [blame] | 1280 | ArrayRef<uint8_t> CmdData(const_cast<uint8_t *>(CGOpts.CmdArgs.data()), |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1281 | CGOpts.CmdArgs.size()); |
| 1282 | llvm::Constant *CmdConstant = |
| 1283 | llvm::ConstantDataArray::get(M->getContext(), CmdData); |
| 1284 | GV = new llvm::GlobalVariable(*M, CmdConstant->getType(), true, |
| 1285 | llvm::GlobalValue::PrivateLinkage, |
| 1286 | CmdConstant); |
| 1287 | GV->setSection(getSectionNameForCommandline(T)); |
| 1288 | UsedArray.push_back( |
| 1289 | ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); |
| 1290 | if (llvm::GlobalVariable *Old = |
| 1291 | M->getGlobalVariable("llvm.cmdline", true)) { |
| 1292 | assert(Old->hasOneUse() && |
| 1293 | "llvm.cmdline can only be used once in llvm.compiler.used"); |
| 1294 | GV->takeName(Old); |
| 1295 | Old->eraseFromParent(); |
| 1296 | } else { |
| 1297 | GV->setName("llvm.cmdline"); |
| 1298 | } |
| 1299 | } |
| 1300 | |
| 1301 | if (UsedArray.empty()) |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1302 | return; |
| 1303 | |
Steven Wu | 2c059a5 | 2016-05-16 18:54:58 +0000 | [diff] [blame] | 1304 | // Recreate llvm.compiler.used. |
| 1305 | ArrayType *ATy = ArrayType::get(UsedElementType, UsedArray.size()); |
| 1306 | auto *NewUsed = new GlobalVariable( |
| 1307 | *M, ATy, false, llvm::GlobalValue::AppendingLinkage, |
| 1308 | llvm::ConstantArray::get(ATy, UsedArray), "llvm.compiler.used"); |
| 1309 | NewUsed->setSection("llvm.metadata"); |
Steven Wu | 27fb522 | 2016-05-11 16:26:03 +0000 | [diff] [blame] | 1310 | } |