blob: 48442bcf2f1df6c2f78dad45e916dc4c2cfcf621 [file] [log] [blame]
Tom Stellard45bb48e2015-06-13 03:28:10 +00001//===-- AMDGPUTargetMachine.cpp - TargetMachine for hw codegen targets-----===//
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//
10/// \file
11/// \brief The AMDGPU target machine contains all of the hardware specific
12/// information needed to emit code for R600 and SI GPUs.
13//
14//===----------------------------------------------------------------------===//
15
16#include "AMDGPUTargetMachine.h"
17#include "AMDGPU.h"
Matt Arsenaulteb9025d2016-06-28 17:42:09 +000018#include "AMDGPUCallLowering.h"
Tom Stellardca166212017-01-30 21:56:46 +000019#include "AMDGPUInstructionSelector.h"
20#include "AMDGPULegalizerInfo.h"
21#ifdef LLVM_BUILD_GLOBAL_ISEL
22#include "AMDGPURegisterBankInfo.h"
23#endif
Matt Arsenaulteb9025d2016-06-28 17:42:09 +000024#include "AMDGPUTargetObjectFile.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000025#include "AMDGPUTargetTransformInfo.h"
Tom Stellard0d23ebe2016-08-29 19:42:52 +000026#include "GCNSchedStrategy.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000027#include "R600MachineScheduler.h"
Matt Arsenault2ffe8fd2016-08-11 19:18:50 +000028#include "SIMachineScheduler.h"
Tom Stellardca166212017-01-30 21:56:46 +000029#include "llvm/CodeGen/GlobalISel/InstructionSelect.h"
Tom Stellard000c5af2016-04-14 19:09:28 +000030#include "llvm/CodeGen/GlobalISel/IRTranslator.h"
Tom Stellardca166212017-01-30 21:56:46 +000031#include "llvm/CodeGen/GlobalISel/Legalizer.h"
32#include "llvm/CodeGen/GlobalISel/RegBankSelect.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000033#include "llvm/CodeGen/Passes.h"
Matthias Braun31d19d42016-05-10 03:21:59 +000034#include "llvm/CodeGen/TargetPassConfig.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000035#include "llvm/Support/TargetRegistry.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000036#include "llvm/Transforms/IPO.h"
Chandler Carruth67fc52f2016-08-17 02:56:20 +000037#include "llvm/Transforms/IPO/AlwaysInliner.h"
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +000038#include "llvm/Transforms/IPO/PassManagerBuilder.h"
Tom Stellard45bb48e2015-06-13 03:28:10 +000039#include "llvm/Transforms/Scalar.h"
Matt Arsenaultf42c6922016-06-15 00:11:01 +000040#include "llvm/Transforms/Scalar/GVN.h"
Matt Arsenault908b9e22016-07-01 03:33:52 +000041#include "llvm/Transforms/Vectorize.h"
Eugene Zelenko6a9226d2016-12-12 22:23:53 +000042#include "llvm/IR/Attributes.h"
43#include "llvm/IR/Function.h"
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +000044#include "llvm/IR/LegacyPassManager.h"
Eugene Zelenko6a9226d2016-12-12 22:23:53 +000045#include "llvm/Pass.h"
46#include "llvm/Support/CommandLine.h"
47#include "llvm/Support/Compiler.h"
48#include "llvm/Target/TargetLoweringObjectFile.h"
49#include <memory>
Tom Stellard45bb48e2015-06-13 03:28:10 +000050
51using namespace llvm;
52
Matt Arsenaultc5816112016-06-24 06:30:22 +000053static cl::opt<bool> EnableR600StructurizeCFG(
54 "r600-ir-structurize",
55 cl::desc("Use StructurizeCFG IR pass"),
56 cl::init(true));
57
Matt Arsenault03d85842016-06-27 20:32:13 +000058static cl::opt<bool> EnableSROA(
59 "amdgpu-sroa",
60 cl::desc("Run SROA after promote alloca pass"),
61 cl::ReallyHidden,
62 cl::init(true));
63
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +000064static cl::opt<bool>
65EnableEarlyIfConversion("amdgpu-early-ifcvt", cl::Hidden,
66 cl::desc("Run early if-conversion"),
67 cl::init(false));
68
Matt Arsenault03d85842016-06-27 20:32:13 +000069static cl::opt<bool> EnableR600IfConvert(
70 "r600-if-convert",
71 cl::desc("Use if conversion pass"),
72 cl::ReallyHidden,
73 cl::init(true));
74
Matt Arsenault908b9e22016-07-01 03:33:52 +000075// Option to disable vectorizer for tests.
76static cl::opt<bool> EnableLoadStoreVectorizer(
77 "amdgpu-load-store-vectorizer",
78 cl::desc("Enable load store vectorizer"),
Matt Arsenault0efdd062016-09-09 22:29:28 +000079 cl::init(true),
Matt Arsenault908b9e22016-07-01 03:33:52 +000080 cl::Hidden);
81
Alexander Timofeev18009562016-12-08 17:28:47 +000082// Option to to control global loads scalarization
83static cl::opt<bool> ScalarizeGlobal(
84 "amdgpu-scalarize-global-loads",
85 cl::desc("Enable global load scalarization"),
86 cl::init(false),
87 cl::Hidden);
88
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +000089// Option to run internalize pass.
90static cl::opt<bool> InternalizeSymbols(
91 "amdgpu-internalize-symbols",
92 cl::desc("Enable elimination of non-kernel functions and unused globals"),
93 cl::init(false),
94 cl::Hidden);
95
Tom Stellard45bb48e2015-06-13 03:28:10 +000096extern "C" void LLVMInitializeAMDGPUTarget() {
97 // Register the target
Mehdi Aminif42454b2016-10-09 23:00:34 +000098 RegisterTargetMachine<R600TargetMachine> X(getTheAMDGPUTarget());
99 RegisterTargetMachine<GCNTargetMachine> Y(getTheGCNTarget());
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000100
101 PassRegistry *PR = PassRegistry::getPassRegistry();
Matt Arsenault8c0ef8b2015-10-12 17:43:59 +0000102 initializeSILowerI1CopiesPass(*PR);
Matt Arsenault782c03b2015-11-03 22:30:13 +0000103 initializeSIFixSGPRCopiesPass(*PR);
Stanislav Mekhanoshin22a56f22017-01-24 17:46:17 +0000104 initializeSIFixVGPRCopiesPass(*PR);
Matt Arsenault8c0ef8b2015-10-12 17:43:59 +0000105 initializeSIFoldOperandsPass(*PR);
Matt Arsenaultc3a01ec2016-06-09 23:18:47 +0000106 initializeSIShrinkInstructionsPass(*PR);
Matt Arsenault187276f2015-10-07 00:42:53 +0000107 initializeSIFixControlFlowLiveIntervalsPass(*PR);
108 initializeSILoadStoreOptimizerPass(*PR);
Matt Arsenault39319482015-11-06 18:01:57 +0000109 initializeAMDGPUAnnotateKernelFeaturesPass(*PR);
Tom Stellarda6f24c62015-12-15 20:55:55 +0000110 initializeAMDGPUAnnotateUniformValuesPass(*PR);
Matt Arsenault0699ef32017-02-09 22:00:42 +0000111 initializeAMDGPULowerIntrinsicsPass(*PR);
Matt Arsenaulte0132462016-01-30 05:19:45 +0000112 initializeAMDGPUPromoteAllocaPass(*PR);
Matt Arsenault86de4862016-06-24 07:07:55 +0000113 initializeAMDGPUCodeGenPreparePass(*PR);
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +0000114 initializeAMDGPUUnifyMetadataPass(*PR);
Tom Stellard77a17772016-01-20 15:48:27 +0000115 initializeSIAnnotateControlFlowPass(*PR);
Tom Stellard6e1967e2016-02-05 17:42:38 +0000116 initializeSIInsertWaitsPass(*PR);
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000117 initializeSIWholeQuadModePass(*PR);
Matt Arsenault55d49cf2016-02-12 02:16:10 +0000118 initializeSILowerControlFlowPass(*PR);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000119 initializeSIInsertSkipsPass(*PR);
Matt Arsenaultd3e4c642016-06-02 00:04:22 +0000120 initializeSIDebuggerInsertNopsPass(*PR);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000121 initializeSIOptimizeExecMaskingPass(*PR);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000122}
123
Tom Stellarde135ffd2015-09-25 21:41:28 +0000124static std::unique_ptr<TargetLoweringObjectFile> createTLOF(const Triple &TT) {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000125 return llvm::make_unique<AMDGPUTargetObjectFile>();
Tom Stellarde135ffd2015-09-25 21:41:28 +0000126}
127
Tom Stellard45bb48e2015-06-13 03:28:10 +0000128static ScheduleDAGInstrs *createR600MachineScheduler(MachineSchedContext *C) {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000129 return new ScheduleDAGMILive(C, llvm::make_unique<R600SchedStrategy>());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000130}
131
Matt Arsenault2ffe8fd2016-08-11 19:18:50 +0000132static ScheduleDAGInstrs *createSIMachineScheduler(MachineSchedContext *C) {
133 return new SIScheduleDAGMI(C);
134}
135
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000136static ScheduleDAGInstrs *
137createGCNMaxOccupancyMachineScheduler(MachineSchedContext *C) {
138 ScheduleDAGMILive *DAG =
Stanislav Mekhanoshin582a5232017-02-15 17:19:50 +0000139 new GCNScheduleDAGMILive(C, make_unique<GCNMaxOccupancySchedStrategy>(C));
Matthias Braun115efcd2016-11-28 20:11:54 +0000140 DAG->addMutation(createLoadClusterDAGMutation(DAG->TII, DAG->TRI));
141 DAG->addMutation(createStoreClusterDAGMutation(DAG->TII, DAG->TRI));
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000142 return DAG;
143}
144
Tom Stellard45bb48e2015-06-13 03:28:10 +0000145static MachineSchedRegistry
Nicolai Haehnle02c32912016-01-13 16:10:10 +0000146R600SchedRegistry("r600", "Run R600's custom scheduler",
147 createR600MachineScheduler);
148
149static MachineSchedRegistry
150SISchedRegistry("si", "Run SI's custom scheduler",
151 createSIMachineScheduler);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000152
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000153static MachineSchedRegistry
154GCNMaxOccupancySchedRegistry("gcn-max-occupancy",
155 "Run GCN scheduler to maximize occupancy",
156 createGCNMaxOccupancyMachineScheduler);
157
Matt Arsenaultec30eb52016-05-31 16:57:45 +0000158static StringRef computeDataLayout(const Triple &TT) {
159 if (TT.getArch() == Triple::r600) {
160 // 32-bit pointers.
161 return "e-p:32:32-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128"
162 "-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64";
Tom Stellard45bb48e2015-06-13 03:28:10 +0000163 }
164
Matt Arsenaultec30eb52016-05-31 16:57:45 +0000165 // 32-bit private, local, and region pointers. 64-bit global, constant and
166 // flat.
167 return "e-p:32:32-p1:64:64-p2:64:64-p3:32:32-p4:64:64-p5:32:32"
168 "-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128"
169 "-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64";
Tom Stellard45bb48e2015-06-13 03:28:10 +0000170}
171
Matt Arsenaultb22828f2016-01-27 02:17:49 +0000172LLVM_READNONE
173static StringRef getGPUOrDefault(const Triple &TT, StringRef GPU) {
174 if (!GPU.empty())
175 return GPU;
176
177 // HSA only supports CI+, so change the default GPU to a CI for HSA.
178 if (TT.getArch() == Triple::amdgcn)
179 return (TT.getOS() == Triple::AMDHSA) ? "kaveri" : "tahiti";
180
Matt Arsenault8e001942016-06-02 18:37:16 +0000181 return "r600";
Matt Arsenaultb22828f2016-01-27 02:17:49 +0000182}
183
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000184static Reloc::Model getEffectiveRelocModel(Optional<Reloc::Model> RM) {
Tom Stellard418beb72016-07-13 14:23:33 +0000185 // The AMDGPU toolchain only supports generating shared objects, so we
186 // must always use PIC.
187 return Reloc::PIC_;
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000188}
189
Tom Stellard45bb48e2015-06-13 03:28:10 +0000190AMDGPUTargetMachine::AMDGPUTargetMachine(const Target &T, const Triple &TT,
191 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000192 TargetOptions Options,
193 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000194 CodeModel::Model CM,
195 CodeGenOpt::Level OptLevel)
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000196 : LLVMTargetMachine(T, computeDataLayout(TT), TT, getGPUOrDefault(TT, CPU),
197 FS, Options, getEffectiveRelocModel(RM), CM, OptLevel),
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000198 TLOF(createTLOF(getTargetTriple())) {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000199 initAsmInfo();
200}
201
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000202AMDGPUTargetMachine::~AMDGPUTargetMachine() = default;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000203
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000204StringRef AMDGPUTargetMachine::getGPUName(const Function &F) const {
205 Attribute GPUAttr = F.getFnAttribute("target-cpu");
206 return GPUAttr.hasAttribute(Attribute::None) ?
207 getTargetCPU() : GPUAttr.getValueAsString();
208}
209
210StringRef AMDGPUTargetMachine::getFeatureString(const Function &F) const {
211 Attribute FSAttr = F.getFnAttribute("target-features");
212
213 return FSAttr.hasAttribute(Attribute::None) ?
214 getTargetFeatureString() :
215 FSAttr.getValueAsString();
216}
217
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000218void AMDGPUTargetMachine::adjustPassManager(PassManagerBuilder &Builder) {
Stanislav Mekhanoshinee2dd782017-03-17 17:13:41 +0000219 Builder.DivergentTarget = true;
220
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000221 bool Internalize = InternalizeSymbols &&
222 (getOptLevel() > CodeGenOpt::None) &&
223 (getTargetTriple().getArch() == Triple::amdgcn);
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000224 Builder.addExtension(
Stanislav Mekhanoshinf6c1feb2017-01-27 16:38:10 +0000225 PassManagerBuilder::EP_ModuleOptimizerEarly,
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000226 [Internalize](const PassManagerBuilder &, legacy::PassManagerBase &PM) {
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000227 PM.add(createAMDGPUUnifyMetadataPass());
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000228 if (Internalize) {
229 PM.add(createInternalizePass([=](const GlobalValue &GV) -> bool {
230 if (const Function *F = dyn_cast<Function>(&GV)) {
231 if (F->isDeclaration())
232 return true;
233 switch (F->getCallingConv()) {
234 default:
235 return false;
236 case CallingConv::AMDGPU_VS:
237 case CallingConv::AMDGPU_GS:
238 case CallingConv::AMDGPU_PS:
239 case CallingConv::AMDGPU_CS:
240 case CallingConv::AMDGPU_KERNEL:
241 case CallingConv::SPIR_KERNEL:
242 return true;
243 }
244 }
245 return !GV.use_empty();
246 }));
247 PM.add(createGlobalDCEPass());
248 }
Stanislav Mekhanoshinf8050792017-03-16 16:11:46 +0000249 PM.add(createAMDGPUAlwaysInlinePass());
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000250 });
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +0000251}
252
Tom Stellard45bb48e2015-06-13 03:28:10 +0000253//===----------------------------------------------------------------------===//
254// R600 Target Machine (R600 -> Cayman)
255//===----------------------------------------------------------------------===//
256
257R600TargetMachine::R600TargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000258 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000259 TargetOptions Options,
260 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000261 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenaultad55ee52016-12-06 01:02:51 +0000262 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {
263 setRequiresStructuredCFG(true);
264}
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000265
266const R600Subtarget *R600TargetMachine::getSubtargetImpl(
267 const Function &F) const {
268 StringRef GPU = getGPUName(F);
269 StringRef FS = getFeatureString(F);
270
271 SmallString<128> SubtargetKey(GPU);
272 SubtargetKey.append(FS);
273
274 auto &I = SubtargetMap[SubtargetKey];
275 if (!I) {
276 // This needs to be done before we create a new subtarget since any
277 // creation will depend on the TM and the code generation flags on the
278 // function that reside in TargetOptions.
279 resetTargetOptions(F);
280 I = llvm::make_unique<R600Subtarget>(TargetTriple, GPU, FS, *this);
281 }
282
283 return I.get();
284}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000285
286//===----------------------------------------------------------------------===//
287// GCN Target Machine (SI+)
288//===----------------------------------------------------------------------===//
289
Matt Arsenault55dff272016-06-28 00:11:26 +0000290#ifdef LLVM_BUILD_GLOBAL_ISEL
291namespace {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000292
Matt Arsenault55dff272016-06-28 00:11:26 +0000293struct SIGISelActualAccessor : public GISelAccessor {
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000294 std::unique_ptr<AMDGPUCallLowering> CallLoweringInfo;
Tom Stellardca166212017-01-30 21:56:46 +0000295 std::unique_ptr<InstructionSelector> InstSelector;
296 std::unique_ptr<LegalizerInfo> Legalizer;
297 std::unique_ptr<RegisterBankInfo> RegBankInfo;
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000298 const AMDGPUCallLowering *getCallLowering() const override {
Matt Arsenault55dff272016-06-28 00:11:26 +0000299 return CallLoweringInfo.get();
300 }
Tom Stellardca166212017-01-30 21:56:46 +0000301 const InstructionSelector *getInstructionSelector() const override {
302 return InstSelector.get();
303 }
304 const LegalizerInfo *getLegalizerInfo() const override {
305 return Legalizer.get();
306 }
307 const RegisterBankInfo *getRegBankInfo() const override {
308 return RegBankInfo.get();
309 }
Matt Arsenault55dff272016-06-28 00:11:26 +0000310};
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000311
312} // end anonymous namespace
Matt Arsenault55dff272016-06-28 00:11:26 +0000313#endif
314
Tom Stellard45bb48e2015-06-13 03:28:10 +0000315GCNTargetMachine::GCNTargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000316 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000317 TargetOptions Options,
318 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000319 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000320 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {}
321
322const SISubtarget *GCNTargetMachine::getSubtargetImpl(const Function &F) const {
323 StringRef GPU = getGPUName(F);
324 StringRef FS = getFeatureString(F);
325
326 SmallString<128> SubtargetKey(GPU);
327 SubtargetKey.append(FS);
328
329 auto &I = SubtargetMap[SubtargetKey];
330 if (!I) {
331 // This needs to be done before we create a new subtarget since any
332 // creation will depend on the TM and the code generation flags on the
333 // function that reside in TargetOptions.
334 resetTargetOptions(F);
335 I = llvm::make_unique<SISubtarget>(TargetTriple, GPU, FS, *this);
336
337#ifndef LLVM_BUILD_GLOBAL_ISEL
338 GISelAccessor *GISel = new GISelAccessor();
339#else
340 SIGISelActualAccessor *GISel = new SIGISelActualAccessor();
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000341 GISel->CallLoweringInfo.reset(
342 new AMDGPUCallLowering(*I->getTargetLowering()));
Tom Stellardca166212017-01-30 21:56:46 +0000343 GISel->Legalizer.reset(new AMDGPULegalizerInfo());
344
345 GISel->RegBankInfo.reset(new AMDGPURegisterBankInfo(*I->getRegisterInfo()));
346 GISel->InstSelector.reset(new AMDGPUInstructionSelector(*I,
347 *static_cast<AMDGPURegisterBankInfo*>(GISel->RegBankInfo.get())));
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000348#endif
349
350 I->setGISelAccessor(*GISel);
351 }
352
Alexander Timofeev18009562016-12-08 17:28:47 +0000353 I->setScalarizeGlobalBehavior(ScalarizeGlobal);
354
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000355 return I.get();
356}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000357
358//===----------------------------------------------------------------------===//
359// AMDGPU Pass Setup
360//===----------------------------------------------------------------------===//
361
362namespace {
Tom Stellardcc7067a62016-03-03 03:53:29 +0000363
Tom Stellard45bb48e2015-06-13 03:28:10 +0000364class AMDGPUPassConfig : public TargetPassConfig {
365public:
366 AMDGPUPassConfig(TargetMachine *TM, PassManagerBase &PM)
Matt Arsenault0a109002015-09-25 17:41:20 +0000367 : TargetPassConfig(TM, PM) {
Matt Arsenault0a109002015-09-25 17:41:20 +0000368 // Exceptions and StackMaps are not supported, so these passes will never do
369 // anything.
370 disablePass(&StackMapLivenessID);
371 disablePass(&FuncletLayoutID);
372 }
Tom Stellard45bb48e2015-06-13 03:28:10 +0000373
374 AMDGPUTargetMachine &getAMDGPUTargetMachine() const {
375 return getTM<AMDGPUTargetMachine>();
376 }
377
Matthias Braun115efcd2016-11-28 20:11:54 +0000378 ScheduleDAGInstrs *
379 createMachineScheduler(MachineSchedContext *C) const override {
380 ScheduleDAGMILive *DAG = createGenericSchedLive(C);
381 DAG->addMutation(createLoadClusterDAGMutation(DAG->TII, DAG->TRI));
382 DAG->addMutation(createStoreClusterDAGMutation(DAG->TII, DAG->TRI));
383 return DAG;
384 }
385
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000386 void addEarlyCSEOrGVNPass();
387 void addStraightLineScalarOptimizationPasses();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000388 void addIRPasses() override;
Matt Arsenault908b9e22016-07-01 03:33:52 +0000389 void addCodeGenPrepare() override;
Matt Arsenault0a109002015-09-25 17:41:20 +0000390 bool addPreISel() override;
391 bool addInstSelector() override;
392 bool addGCPasses() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000393};
394
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000395class R600PassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000396public:
397 R600PassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000398 : AMDGPUPassConfig(TM, PM) {}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000399
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000400 ScheduleDAGInstrs *createMachineScheduler(
401 MachineSchedContext *C) const override {
402 return createR600MachineScheduler(C);
403 }
404
Tom Stellard45bb48e2015-06-13 03:28:10 +0000405 bool addPreISel() override;
406 void addPreRegAlloc() override;
407 void addPreSched2() override;
408 void addPreEmitPass() override;
409};
410
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000411class GCNPassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000412public:
413 GCNPassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000414 : AMDGPUPassConfig(TM, PM) {}
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000415
416 GCNTargetMachine &getGCNTargetMachine() const {
417 return getTM<GCNTargetMachine>();
418 }
419
420 ScheduleDAGInstrs *
Matt Arsenault03d85842016-06-27 20:32:13 +0000421 createMachineScheduler(MachineSchedContext *C) const override;
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000422
Tom Stellard45bb48e2015-06-13 03:28:10 +0000423 bool addPreISel() override;
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000424 void addMachineSSAOptimization() override;
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000425 bool addILPOpts() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000426 bool addInstSelector() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000427#ifdef LLVM_BUILD_GLOBAL_ISEL
428 bool addIRTranslator() override;
Tim Northover33b07d62016-07-22 20:03:43 +0000429 bool addLegalizeMachineIR() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000430 bool addRegBankSelect() override;
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000431 bool addGlobalInstructionSelect() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000432#endif
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000433 void addFastRegAlloc(FunctionPass *RegAllocPass) override;
434 void addOptimizedRegAlloc(FunctionPass *RegAllocPass) override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000435 void addPreRegAlloc() override;
Matt Arsenaulte6740752016-09-29 01:44:16 +0000436 void addPostRegAlloc() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000437 void addPreSched2() override;
438 void addPreEmitPass() override;
439};
440
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000441} // end anonymous namespace
Tom Stellard45bb48e2015-06-13 03:28:10 +0000442
443TargetIRAnalysis AMDGPUTargetMachine::getTargetIRAnalysis() {
Eric Christophera4e5d3c2015-09-16 23:38:13 +0000444 return TargetIRAnalysis([this](const Function &F) {
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000445 return TargetTransformInfo(AMDGPUTTIImpl(this, F));
Mehdi Amini5010ebf2015-07-09 02:08:42 +0000446 });
Tom Stellard45bb48e2015-06-13 03:28:10 +0000447}
448
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000449void AMDGPUPassConfig::addEarlyCSEOrGVNPass() {
450 if (getOptLevel() == CodeGenOpt::Aggressive)
451 addPass(createGVNPass());
452 else
453 addPass(createEarlyCSEPass());
454}
455
456void AMDGPUPassConfig::addStraightLineScalarOptimizationPasses() {
457 addPass(createSeparateConstOffsetFromGEPPass());
458 addPass(createSpeculativeExecutionPass());
459 // ReassociateGEPs exposes more opportunites for SLSR. See
460 // the example in reassociate-geps-and-slsr.ll.
461 addPass(createStraightLineStrengthReducePass());
462 // SeparateConstOffsetFromGEP and SLSR creates common expressions which GVN or
463 // EarlyCSE can reuse.
464 addEarlyCSEOrGVNPass();
465 // Run NaryReassociate after EarlyCSE/GVN to be more effective.
466 addPass(createNaryReassociatePass());
467 // NaryReassociate on GEPs creates redundant common expressions, so run
468 // EarlyCSE after it.
469 addPass(createEarlyCSEPass());
470}
471
Tom Stellard45bb48e2015-06-13 03:28:10 +0000472void AMDGPUPassConfig::addIRPasses() {
Matt Arsenaultbde80342016-05-18 15:41:07 +0000473 // There is no reason to run these.
474 disablePass(&StackMapLivenessID);
475 disablePass(&FuncletLayoutID);
476 disablePass(&PatchableFunctionID);
477
Matt Arsenault0699ef32017-02-09 22:00:42 +0000478 addPass(createAMDGPULowerIntrinsicsPass());
479
Tom Stellard45bb48e2015-06-13 03:28:10 +0000480 // Function calls are not supported, so make sure we inline everything.
481 addPass(createAMDGPUAlwaysInlinePass());
Chandler Carruth67fc52f2016-08-17 02:56:20 +0000482 addPass(createAlwaysInlinerLegacyPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000483 // We need to add the barrier noop pass, otherwise adding the function
484 // inlining pass will cause all of the PassConfigs passes to be run
485 // one function at a time, which means if we have a nodule with two
486 // functions, then we will generate code for the first function
487 // without ever running any passes on the second.
488 addPass(createBarrierNoopPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000489
Matt Arsenault0c329382017-01-30 18:40:29 +0000490 const AMDGPUTargetMachine &TM = getAMDGPUTargetMachine();
491
492 if (TM.getTargetTriple().getArch() == Triple::amdgcn) {
493 // TODO: May want to move later or split into an early and late one.
494
495 addPass(createAMDGPUCodeGenPreparePass(
496 static_cast<const GCNTargetMachine *>(&TM)));
497 }
498
Tom Stellardfd253952015-08-07 23:19:30 +0000499 // Handle uses of OpenCL image2d_t, image3d_t and sampler_t arguments.
500 addPass(createAMDGPUOpenCLImageTypeLoweringPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000501
Matt Arsenault03d85842016-06-27 20:32:13 +0000502 if (TM.getOptLevel() > CodeGenOpt::None) {
Matt Arsenault417e0072017-02-08 06:16:04 +0000503 addPass(createInferAddressSpacesPass());
Matt Arsenaulte0132462016-01-30 05:19:45 +0000504 addPass(createAMDGPUPromoteAlloca(&TM));
Matt Arsenault03d85842016-06-27 20:32:13 +0000505
506 if (EnableSROA)
507 addPass(createSROAPass());
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000508
Konstantin Zhuravlyov4658e5f2016-09-30 16:39:24 +0000509 addStraightLineScalarOptimizationPasses();
510 }
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000511
512 TargetPassConfig::addIRPasses();
513
514 // EarlyCSE is not always strong enough to clean up what LSR produces. For
515 // example, GVN can combine
516 //
517 // %0 = add %a, %b
518 // %1 = add %b, %a
519 //
520 // and
521 //
522 // %0 = shl nsw %a, 2
523 // %1 = shl %a, 2
524 //
525 // but EarlyCSE can do neither of them.
526 if (getOptLevel() != CodeGenOpt::None)
527 addEarlyCSEOrGVNPass();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000528}
529
Matt Arsenault908b9e22016-07-01 03:33:52 +0000530void AMDGPUPassConfig::addCodeGenPrepare() {
531 TargetPassConfig::addCodeGenPrepare();
532
533 if (EnableLoadStoreVectorizer)
534 addPass(createLoadStoreVectorizerPass());
535}
536
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000537bool AMDGPUPassConfig::addPreISel() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000538 addPass(createFlattenCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000539 return false;
540}
541
542bool AMDGPUPassConfig::addInstSelector() {
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +0000543 addPass(createAMDGPUISelDag(getAMDGPUTargetMachine(), getOptLevel()));
Tom Stellard45bb48e2015-06-13 03:28:10 +0000544 return false;
545}
546
Matt Arsenault0a109002015-09-25 17:41:20 +0000547bool AMDGPUPassConfig::addGCPasses() {
548 // Do nothing. GC is not supported.
549 return false;
550}
551
Tom Stellard45bb48e2015-06-13 03:28:10 +0000552//===----------------------------------------------------------------------===//
553// R600 Pass Setup
554//===----------------------------------------------------------------------===//
555
556bool R600PassConfig::addPreISel() {
557 AMDGPUPassConfig::addPreISel();
Matt Arsenaultc5816112016-06-24 06:30:22 +0000558
559 if (EnableR600StructurizeCFG)
Tom Stellardbc4497b2016-02-12 23:45:29 +0000560 addPass(createStructurizeCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000561 return false;
562}
563
564void R600PassConfig::addPreRegAlloc() {
565 addPass(createR600VectorRegMerger(*TM));
566}
567
568void R600PassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000569 addPass(createR600EmitClauseMarkers(), false);
Matt Arsenault03d85842016-06-27 20:32:13 +0000570 if (EnableR600IfConvert)
Tom Stellard45bb48e2015-06-13 03:28:10 +0000571 addPass(&IfConverterID, false);
572 addPass(createR600ClauseMergePass(*TM), false);
573}
574
575void R600PassConfig::addPreEmitPass() {
576 addPass(createAMDGPUCFGStructurizerPass(), false);
577 addPass(createR600ExpandSpecialInstrsPass(*TM), false);
578 addPass(&FinalizeMachineBundlesID, false);
579 addPass(createR600Packetizer(*TM), false);
580 addPass(createR600ControlFlowFinalizer(*TM), false);
581}
582
583TargetPassConfig *R600TargetMachine::createPassConfig(PassManagerBase &PM) {
584 return new R600PassConfig(this, PM);
585}
586
587//===----------------------------------------------------------------------===//
588// GCN Pass Setup
589//===----------------------------------------------------------------------===//
590
Matt Arsenault03d85842016-06-27 20:32:13 +0000591ScheduleDAGInstrs *GCNPassConfig::createMachineScheduler(
592 MachineSchedContext *C) const {
593 const SISubtarget &ST = C->MF->getSubtarget<SISubtarget>();
594 if (ST.enableSIScheduler())
595 return createSIMachineScheduler(C);
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000596 return createGCNMaxOccupancyMachineScheduler(C);
Matt Arsenault03d85842016-06-27 20:32:13 +0000597}
598
Tom Stellard45bb48e2015-06-13 03:28:10 +0000599bool GCNPassConfig::addPreISel() {
600 AMDGPUPassConfig::addPreISel();
Matt Arsenault39319482015-11-06 18:01:57 +0000601
602 // FIXME: We need to run a pass to propagate the attributes when calls are
603 // supported.
Matt Arsenaulte823d922017-02-18 18:29:53 +0000604 const AMDGPUTargetMachine &TM = getAMDGPUTargetMachine();
605 addPass(createAMDGPUAnnotateKernelFeaturesPass(&TM));
Tom Stellardbc4497b2016-02-12 23:45:29 +0000606 addPass(createStructurizeCFGPass(true)); // true -> SkipUniformRegions
Tom Stellard45bb48e2015-06-13 03:28:10 +0000607 addPass(createSinkingPass());
608 addPass(createSITypeRewriter());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000609 addPass(createAMDGPUAnnotateUniformValues());
Tom Stellardbc4497b2016-02-12 23:45:29 +0000610 addPass(createSIAnnotateControlFlowPass());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000611
Tom Stellard45bb48e2015-06-13 03:28:10 +0000612 return false;
613}
614
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000615void GCNPassConfig::addMachineSSAOptimization() {
616 TargetPassConfig::addMachineSSAOptimization();
617
618 // We want to fold operands after PeepholeOptimizer has run (or as part of
619 // it), because it will eliminate extra copies making it easier to fold the
620 // real source operand. We want to eliminate dead instructions after, so that
621 // we see fewer uses of the copies. We then need to clean up the dead
622 // instructions leftover after the operands are folded as well.
623 //
624 // XXX - Can we get away without running DeadMachineInstructionElim again?
625 addPass(&SIFoldOperandsID);
626 addPass(&DeadMachineInstructionElimID);
Tom Stellardc2ff0eb2016-08-29 19:15:22 +0000627 addPass(&SILoadStoreOptimizerID);
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000628}
629
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000630bool GCNPassConfig::addILPOpts() {
631 if (EnableEarlyIfConversion)
632 addPass(&EarlyIfConverterID);
633
634 TargetPassConfig::addILPOpts();
635 return false;
636}
637
Tom Stellard45bb48e2015-06-13 03:28:10 +0000638bool GCNPassConfig::addInstSelector() {
639 AMDGPUPassConfig::addInstSelector();
640 addPass(createSILowerI1CopiesPass());
Matt Arsenault782c03b2015-11-03 22:30:13 +0000641 addPass(&SIFixSGPRCopiesID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000642 return false;
643}
644
Tom Stellard000c5af2016-04-14 19:09:28 +0000645#ifdef LLVM_BUILD_GLOBAL_ISEL
646bool GCNPassConfig::addIRTranslator() {
647 addPass(new IRTranslator());
648 return false;
649}
650
Tim Northover33b07d62016-07-22 20:03:43 +0000651bool GCNPassConfig::addLegalizeMachineIR() {
Tom Stellardca166212017-01-30 21:56:46 +0000652 addPass(new Legalizer());
Tim Northover33b07d62016-07-22 20:03:43 +0000653 return false;
654}
655
Tom Stellard000c5af2016-04-14 19:09:28 +0000656bool GCNPassConfig::addRegBankSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000657 addPass(new RegBankSelect());
Tom Stellard000c5af2016-04-14 19:09:28 +0000658 return false;
659}
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000660
661bool GCNPassConfig::addGlobalInstructionSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000662 addPass(new InstructionSelect());
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000663 return false;
664}
Tom Stellardca166212017-01-30 21:56:46 +0000665
Tom Stellard000c5af2016-04-14 19:09:28 +0000666#endif
667
Tom Stellard45bb48e2015-06-13 03:28:10 +0000668void GCNPassConfig::addPreRegAlloc() {
Matt Arsenault4a07bf62016-06-22 20:26:24 +0000669 addPass(createSIShrinkInstructionsPass());
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000670 addPass(createSIWholeQuadModePass());
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000671}
672
673void GCNPassConfig::addFastRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000674 // FIXME: We have to disable the verifier here because of PHIElimination +
675 // TwoAddressInstructions disabling it.
Matt Arsenaulte6740752016-09-29 01:44:16 +0000676
677 // This must be run immediately after phi elimination and before
678 // TwoAddressInstructions, otherwise the processing of the tied operand of
679 // SI_ELSE will introduce a copy of the tied operand source after the else.
680 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000681
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000682 TargetPassConfig::addFastRegAlloc(RegAllocPass);
683}
684
685void GCNPassConfig::addOptimizedRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000686 // This needs to be run directly before register allocation because earlier
687 // passes might recompute live intervals.
688 insertPass(&MachineSchedulerID, &SIFixControlFlowLiveIntervalsID);
689
Matt Arsenaulte6740752016-09-29 01:44:16 +0000690 // This must be run immediately after phi elimination and before
691 // TwoAddressInstructions, otherwise the processing of the tied operand of
692 // SI_ELSE will introduce a copy of the tied operand source after the else.
693 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000694
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000695 TargetPassConfig::addOptimizedRegAlloc(RegAllocPass);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000696}
697
Matt Arsenaulte6740752016-09-29 01:44:16 +0000698void GCNPassConfig::addPostRegAlloc() {
Stanislav Mekhanoshin22a56f22017-01-24 17:46:17 +0000699 addPass(&SIFixVGPRCopiesID);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000700 addPass(&SIOptimizeExecMaskingID);
701 TargetPassConfig::addPostRegAlloc();
702}
703
Tom Stellard45bb48e2015-06-13 03:28:10 +0000704void GCNPassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000705}
706
707void GCNPassConfig::addPreEmitPass() {
Tom Stellardcb6ba622016-04-30 00:23:06 +0000708 // The hazard recognizer that runs as part of the post-ra scheduler does not
Matt Arsenault254a6452016-06-28 16:59:53 +0000709 // guarantee to be able handle all hazards correctly. This is because if there
710 // are multiple scheduling regions in a basic block, the regions are scheduled
711 // bottom up, so when we begin to schedule a region we don't know what
712 // instructions were emitted directly before it.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000713 //
Matt Arsenault254a6452016-06-28 16:59:53 +0000714 // Here we add a stand-alone hazard recognizer pass which can handle all
715 // cases.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000716 addPass(&PostRAHazardRecognizerID);
717
Matt Arsenaulte2bd9a32016-06-09 23:19:14 +0000718 addPass(createSIInsertWaitsPass());
Matt Arsenaultcf2744f2016-04-29 20:23:42 +0000719 addPass(createSIShrinkInstructionsPass());
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000720 addPass(&SIInsertSkipsPassID);
Matt Arsenault9babdf42016-06-22 20:15:28 +0000721 addPass(createSIDebuggerInsertNopsPass());
Matt Arsenault6bc43d82016-10-06 16:20:41 +0000722 addPass(&BranchRelaxationPassID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000723}
724
725TargetPassConfig *GCNTargetMachine::createPassConfig(PassManagerBase &PM) {
726 return new GCNPassConfig(this, PM);
727}