blob: 1847d17713776d28f5258177a378a3dc67a7b1c3 [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 =
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000139 new ScheduleDAGMILive(C,
140 llvm::make_unique<GCNMaxOccupancySchedStrategy>(C));
Matthias Braun115efcd2016-11-28 20:11:54 +0000141 DAG->addMutation(createLoadClusterDAGMutation(DAG->TII, DAG->TRI));
142 DAG->addMutation(createStoreClusterDAGMutation(DAG->TII, DAG->TRI));
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000143 return DAG;
144}
145
Tom Stellard45bb48e2015-06-13 03:28:10 +0000146static MachineSchedRegistry
Nicolai Haehnle02c32912016-01-13 16:10:10 +0000147R600SchedRegistry("r600", "Run R600's custom scheduler",
148 createR600MachineScheduler);
149
150static MachineSchedRegistry
151SISchedRegistry("si", "Run SI's custom scheduler",
152 createSIMachineScheduler);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000153
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000154static MachineSchedRegistry
155GCNMaxOccupancySchedRegistry("gcn-max-occupancy",
156 "Run GCN scheduler to maximize occupancy",
157 createGCNMaxOccupancyMachineScheduler);
158
Matt Arsenaultec30eb52016-05-31 16:57:45 +0000159static StringRef computeDataLayout(const Triple &TT) {
160 if (TT.getArch() == Triple::r600) {
161 // 32-bit pointers.
162 return "e-p:32:32-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128"
163 "-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64";
Tom Stellard45bb48e2015-06-13 03:28:10 +0000164 }
165
Matt Arsenaultec30eb52016-05-31 16:57:45 +0000166 // 32-bit private, local, and region pointers. 64-bit global, constant and
167 // flat.
168 return "e-p:32:32-p1:64:64-p2:64:64-p3:32:32-p4:64:64-p5:32:32"
169 "-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128"
170 "-v192:256-v256:256-v512:512-v1024:1024-v2048:2048-n32:64";
Tom Stellard45bb48e2015-06-13 03:28:10 +0000171}
172
Matt Arsenaultb22828f2016-01-27 02:17:49 +0000173LLVM_READNONE
174static StringRef getGPUOrDefault(const Triple &TT, StringRef GPU) {
175 if (!GPU.empty())
176 return GPU;
177
178 // HSA only supports CI+, so change the default GPU to a CI for HSA.
179 if (TT.getArch() == Triple::amdgcn)
180 return (TT.getOS() == Triple::AMDHSA) ? "kaveri" : "tahiti";
181
Matt Arsenault8e001942016-06-02 18:37:16 +0000182 return "r600";
Matt Arsenaultb22828f2016-01-27 02:17:49 +0000183}
184
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000185static Reloc::Model getEffectiveRelocModel(Optional<Reloc::Model> RM) {
Tom Stellard418beb72016-07-13 14:23:33 +0000186 // The AMDGPU toolchain only supports generating shared objects, so we
187 // must always use PIC.
188 return Reloc::PIC_;
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000189}
190
Tom Stellard45bb48e2015-06-13 03:28:10 +0000191AMDGPUTargetMachine::AMDGPUTargetMachine(const Target &T, const Triple &TT,
192 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000193 TargetOptions Options,
194 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000195 CodeModel::Model CM,
196 CodeGenOpt::Level OptLevel)
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000197 : LLVMTargetMachine(T, computeDataLayout(TT), TT, getGPUOrDefault(TT, CPU),
198 FS, Options, getEffectiveRelocModel(RM), CM, OptLevel),
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000199 TLOF(createTLOF(getTargetTriple())) {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000200 initAsmInfo();
201}
202
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000203AMDGPUTargetMachine::~AMDGPUTargetMachine() = default;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000204
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000205StringRef AMDGPUTargetMachine::getGPUName(const Function &F) const {
206 Attribute GPUAttr = F.getFnAttribute("target-cpu");
207 return GPUAttr.hasAttribute(Attribute::None) ?
208 getTargetCPU() : GPUAttr.getValueAsString();
209}
210
211StringRef AMDGPUTargetMachine::getFeatureString(const Function &F) const {
212 Attribute FSAttr = F.getFnAttribute("target-features");
213
214 return FSAttr.hasAttribute(Attribute::None) ?
215 getTargetFeatureString() :
216 FSAttr.getValueAsString();
217}
218
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000219void AMDGPUTargetMachine::adjustPassManager(PassManagerBuilder &Builder) {
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000220 bool Internalize = InternalizeSymbols &&
221 (getOptLevel() > CodeGenOpt::None) &&
222 (getTargetTriple().getArch() == Triple::amdgcn);
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000223 Builder.addExtension(
Stanislav Mekhanoshinf6c1feb2017-01-27 16:38:10 +0000224 PassManagerBuilder::EP_ModuleOptimizerEarly,
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000225 [Internalize](const PassManagerBuilder &, legacy::PassManagerBase &PM) {
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000226 PM.add(createAMDGPUUnifyMetadataPass());
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000227 if (Internalize) {
228 PM.add(createInternalizePass([=](const GlobalValue &GV) -> bool {
229 if (const Function *F = dyn_cast<Function>(&GV)) {
230 if (F->isDeclaration())
231 return true;
232 switch (F->getCallingConv()) {
233 default:
234 return false;
235 case CallingConv::AMDGPU_VS:
236 case CallingConv::AMDGPU_GS:
237 case CallingConv::AMDGPU_PS:
238 case CallingConv::AMDGPU_CS:
239 case CallingConv::AMDGPU_KERNEL:
240 case CallingConv::SPIR_KERNEL:
241 return true;
242 }
243 }
244 return !GV.use_empty();
245 }));
246 PM.add(createGlobalDCEPass());
247 }
248 });
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +0000249}
250
Tom Stellard45bb48e2015-06-13 03:28:10 +0000251//===----------------------------------------------------------------------===//
252// R600 Target Machine (R600 -> Cayman)
253//===----------------------------------------------------------------------===//
254
255R600TargetMachine::R600TargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000256 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000257 TargetOptions Options,
258 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000259 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenaultad55ee52016-12-06 01:02:51 +0000260 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {
261 setRequiresStructuredCFG(true);
262}
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000263
264const R600Subtarget *R600TargetMachine::getSubtargetImpl(
265 const Function &F) const {
266 StringRef GPU = getGPUName(F);
267 StringRef FS = getFeatureString(F);
268
269 SmallString<128> SubtargetKey(GPU);
270 SubtargetKey.append(FS);
271
272 auto &I = SubtargetMap[SubtargetKey];
273 if (!I) {
274 // This needs to be done before we create a new subtarget since any
275 // creation will depend on the TM and the code generation flags on the
276 // function that reside in TargetOptions.
277 resetTargetOptions(F);
278 I = llvm::make_unique<R600Subtarget>(TargetTriple, GPU, FS, *this);
279 }
280
281 return I.get();
282}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000283
284//===----------------------------------------------------------------------===//
285// GCN Target Machine (SI+)
286//===----------------------------------------------------------------------===//
287
Matt Arsenault55dff272016-06-28 00:11:26 +0000288#ifdef LLVM_BUILD_GLOBAL_ISEL
289namespace {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000290
Matt Arsenault55dff272016-06-28 00:11:26 +0000291struct SIGISelActualAccessor : public GISelAccessor {
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000292 std::unique_ptr<AMDGPUCallLowering> CallLoweringInfo;
Tom Stellardca166212017-01-30 21:56:46 +0000293 std::unique_ptr<InstructionSelector> InstSelector;
294 std::unique_ptr<LegalizerInfo> Legalizer;
295 std::unique_ptr<RegisterBankInfo> RegBankInfo;
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000296 const AMDGPUCallLowering *getCallLowering() const override {
Matt Arsenault55dff272016-06-28 00:11:26 +0000297 return CallLoweringInfo.get();
298 }
Tom Stellardca166212017-01-30 21:56:46 +0000299 const InstructionSelector *getInstructionSelector() const override {
300 return InstSelector.get();
301 }
302 const LegalizerInfo *getLegalizerInfo() const override {
303 return Legalizer.get();
304 }
305 const RegisterBankInfo *getRegBankInfo() const override {
306 return RegBankInfo.get();
307 }
Matt Arsenault55dff272016-06-28 00:11:26 +0000308};
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000309
310} // end anonymous namespace
Matt Arsenault55dff272016-06-28 00:11:26 +0000311#endif
312
Tom Stellard45bb48e2015-06-13 03:28:10 +0000313GCNTargetMachine::GCNTargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000314 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000315 TargetOptions Options,
316 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000317 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000318 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {}
319
320const SISubtarget *GCNTargetMachine::getSubtargetImpl(const Function &F) const {
321 StringRef GPU = getGPUName(F);
322 StringRef FS = getFeatureString(F);
323
324 SmallString<128> SubtargetKey(GPU);
325 SubtargetKey.append(FS);
326
327 auto &I = SubtargetMap[SubtargetKey];
328 if (!I) {
329 // This needs to be done before we create a new subtarget since any
330 // creation will depend on the TM and the code generation flags on the
331 // function that reside in TargetOptions.
332 resetTargetOptions(F);
333 I = llvm::make_unique<SISubtarget>(TargetTriple, GPU, FS, *this);
334
335#ifndef LLVM_BUILD_GLOBAL_ISEL
336 GISelAccessor *GISel = new GISelAccessor();
337#else
338 SIGISelActualAccessor *GISel = new SIGISelActualAccessor();
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000339 GISel->CallLoweringInfo.reset(
340 new AMDGPUCallLowering(*I->getTargetLowering()));
Tom Stellardca166212017-01-30 21:56:46 +0000341 GISel->Legalizer.reset(new AMDGPULegalizerInfo());
342
343 GISel->RegBankInfo.reset(new AMDGPURegisterBankInfo(*I->getRegisterInfo()));
344 GISel->InstSelector.reset(new AMDGPUInstructionSelector(*I,
345 *static_cast<AMDGPURegisterBankInfo*>(GISel->RegBankInfo.get())));
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000346#endif
347
348 I->setGISelAccessor(*GISel);
349 }
350
Alexander Timofeev18009562016-12-08 17:28:47 +0000351 I->setScalarizeGlobalBehavior(ScalarizeGlobal);
352
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000353 return I.get();
354}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000355
356//===----------------------------------------------------------------------===//
357// AMDGPU Pass Setup
358//===----------------------------------------------------------------------===//
359
360namespace {
Tom Stellardcc7067a62016-03-03 03:53:29 +0000361
Tom Stellard45bb48e2015-06-13 03:28:10 +0000362class AMDGPUPassConfig : public TargetPassConfig {
363public:
364 AMDGPUPassConfig(TargetMachine *TM, PassManagerBase &PM)
Matt Arsenault0a109002015-09-25 17:41:20 +0000365 : TargetPassConfig(TM, PM) {
Matt Arsenault0a109002015-09-25 17:41:20 +0000366 // Exceptions and StackMaps are not supported, so these passes will never do
367 // anything.
368 disablePass(&StackMapLivenessID);
369 disablePass(&FuncletLayoutID);
370 }
Tom Stellard45bb48e2015-06-13 03:28:10 +0000371
372 AMDGPUTargetMachine &getAMDGPUTargetMachine() const {
373 return getTM<AMDGPUTargetMachine>();
374 }
375
Matthias Braun115efcd2016-11-28 20:11:54 +0000376 ScheduleDAGInstrs *
377 createMachineScheduler(MachineSchedContext *C) const override {
378 ScheduleDAGMILive *DAG = createGenericSchedLive(C);
379 DAG->addMutation(createLoadClusterDAGMutation(DAG->TII, DAG->TRI));
380 DAG->addMutation(createStoreClusterDAGMutation(DAG->TII, DAG->TRI));
381 return DAG;
382 }
383
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000384 void addEarlyCSEOrGVNPass();
385 void addStraightLineScalarOptimizationPasses();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000386 void addIRPasses() override;
Matt Arsenault908b9e22016-07-01 03:33:52 +0000387 void addCodeGenPrepare() override;
Matt Arsenault0a109002015-09-25 17:41:20 +0000388 bool addPreISel() override;
389 bool addInstSelector() override;
390 bool addGCPasses() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000391};
392
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000393class R600PassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000394public:
395 R600PassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000396 : AMDGPUPassConfig(TM, PM) {}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000397
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000398 ScheduleDAGInstrs *createMachineScheduler(
399 MachineSchedContext *C) const override {
400 return createR600MachineScheduler(C);
401 }
402
Tom Stellard45bb48e2015-06-13 03:28:10 +0000403 bool addPreISel() override;
404 void addPreRegAlloc() override;
405 void addPreSched2() override;
406 void addPreEmitPass() override;
407};
408
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000409class GCNPassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000410public:
411 GCNPassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000412 : AMDGPUPassConfig(TM, PM) {}
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000413
414 GCNTargetMachine &getGCNTargetMachine() const {
415 return getTM<GCNTargetMachine>();
416 }
417
418 ScheduleDAGInstrs *
Matt Arsenault03d85842016-06-27 20:32:13 +0000419 createMachineScheduler(MachineSchedContext *C) const override;
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000420
Tom Stellard45bb48e2015-06-13 03:28:10 +0000421 bool addPreISel() override;
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000422 void addMachineSSAOptimization() override;
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000423 bool addILPOpts() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000424 bool addInstSelector() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000425#ifdef LLVM_BUILD_GLOBAL_ISEL
426 bool addIRTranslator() override;
Tim Northover33b07d62016-07-22 20:03:43 +0000427 bool addLegalizeMachineIR() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000428 bool addRegBankSelect() override;
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000429 bool addGlobalInstructionSelect() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000430#endif
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000431 void addFastRegAlloc(FunctionPass *RegAllocPass) override;
432 void addOptimizedRegAlloc(FunctionPass *RegAllocPass) override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000433 void addPreRegAlloc() override;
Matt Arsenaulte6740752016-09-29 01:44:16 +0000434 void addPostRegAlloc() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000435 void addPreSched2() override;
436 void addPreEmitPass() override;
437};
438
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000439} // end anonymous namespace
Tom Stellard45bb48e2015-06-13 03:28:10 +0000440
441TargetIRAnalysis AMDGPUTargetMachine::getTargetIRAnalysis() {
Eric Christophera4e5d3c2015-09-16 23:38:13 +0000442 return TargetIRAnalysis([this](const Function &F) {
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000443 return TargetTransformInfo(AMDGPUTTIImpl(this, F));
Mehdi Amini5010ebf2015-07-09 02:08:42 +0000444 });
Tom Stellard45bb48e2015-06-13 03:28:10 +0000445}
446
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000447void AMDGPUPassConfig::addEarlyCSEOrGVNPass() {
448 if (getOptLevel() == CodeGenOpt::Aggressive)
449 addPass(createGVNPass());
450 else
451 addPass(createEarlyCSEPass());
452}
453
454void AMDGPUPassConfig::addStraightLineScalarOptimizationPasses() {
455 addPass(createSeparateConstOffsetFromGEPPass());
456 addPass(createSpeculativeExecutionPass());
457 // ReassociateGEPs exposes more opportunites for SLSR. See
458 // the example in reassociate-geps-and-slsr.ll.
459 addPass(createStraightLineStrengthReducePass());
460 // SeparateConstOffsetFromGEP and SLSR creates common expressions which GVN or
461 // EarlyCSE can reuse.
462 addEarlyCSEOrGVNPass();
463 // Run NaryReassociate after EarlyCSE/GVN to be more effective.
464 addPass(createNaryReassociatePass());
465 // NaryReassociate on GEPs creates redundant common expressions, so run
466 // EarlyCSE after it.
467 addPass(createEarlyCSEPass());
468}
469
Tom Stellard45bb48e2015-06-13 03:28:10 +0000470void AMDGPUPassConfig::addIRPasses() {
Matt Arsenaultbde80342016-05-18 15:41:07 +0000471 // There is no reason to run these.
472 disablePass(&StackMapLivenessID);
473 disablePass(&FuncletLayoutID);
474 disablePass(&PatchableFunctionID);
475
Matt Arsenault0699ef32017-02-09 22:00:42 +0000476 addPass(createAMDGPULowerIntrinsicsPass());
477
Tom Stellard45bb48e2015-06-13 03:28:10 +0000478 // Function calls are not supported, so make sure we inline everything.
479 addPass(createAMDGPUAlwaysInlinePass());
Chandler Carruth67fc52f2016-08-17 02:56:20 +0000480 addPass(createAlwaysInlinerLegacyPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000481 // We need to add the barrier noop pass, otherwise adding the function
482 // inlining pass will cause all of the PassConfigs passes to be run
483 // one function at a time, which means if we have a nodule with two
484 // functions, then we will generate code for the first function
485 // without ever running any passes on the second.
486 addPass(createBarrierNoopPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000487
Matt Arsenault0c329382017-01-30 18:40:29 +0000488 const AMDGPUTargetMachine &TM = getAMDGPUTargetMachine();
489
490 if (TM.getTargetTriple().getArch() == Triple::amdgcn) {
491 // TODO: May want to move later or split into an early and late one.
492
493 addPass(createAMDGPUCodeGenPreparePass(
494 static_cast<const GCNTargetMachine *>(&TM)));
495 }
496
Tom Stellardfd253952015-08-07 23:19:30 +0000497 // Handle uses of OpenCL image2d_t, image3d_t and sampler_t arguments.
498 addPass(createAMDGPUOpenCLImageTypeLoweringPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000499
Matt Arsenault03d85842016-06-27 20:32:13 +0000500 if (TM.getOptLevel() > CodeGenOpt::None) {
Matt Arsenault417e0072017-02-08 06:16:04 +0000501 addPass(createInferAddressSpacesPass());
Matt Arsenaulte0132462016-01-30 05:19:45 +0000502 addPass(createAMDGPUPromoteAlloca(&TM));
Matt Arsenault03d85842016-06-27 20:32:13 +0000503
504 if (EnableSROA)
505 addPass(createSROAPass());
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000506
Konstantin Zhuravlyov4658e5f2016-09-30 16:39:24 +0000507 addStraightLineScalarOptimizationPasses();
508 }
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000509
510 TargetPassConfig::addIRPasses();
511
512 // EarlyCSE is not always strong enough to clean up what LSR produces. For
513 // example, GVN can combine
514 //
515 // %0 = add %a, %b
516 // %1 = add %b, %a
517 //
518 // and
519 //
520 // %0 = shl nsw %a, 2
521 // %1 = shl %a, 2
522 //
523 // but EarlyCSE can do neither of them.
524 if (getOptLevel() != CodeGenOpt::None)
525 addEarlyCSEOrGVNPass();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000526}
527
Matt Arsenault908b9e22016-07-01 03:33:52 +0000528void AMDGPUPassConfig::addCodeGenPrepare() {
529 TargetPassConfig::addCodeGenPrepare();
530
531 if (EnableLoadStoreVectorizer)
532 addPass(createLoadStoreVectorizerPass());
533}
534
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000535bool AMDGPUPassConfig::addPreISel() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000536 addPass(createFlattenCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000537 return false;
538}
539
540bool AMDGPUPassConfig::addInstSelector() {
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +0000541 addPass(createAMDGPUISelDag(getAMDGPUTargetMachine(), getOptLevel()));
Tom Stellard45bb48e2015-06-13 03:28:10 +0000542 return false;
543}
544
Matt Arsenault0a109002015-09-25 17:41:20 +0000545bool AMDGPUPassConfig::addGCPasses() {
546 // Do nothing. GC is not supported.
547 return false;
548}
549
Tom Stellard45bb48e2015-06-13 03:28:10 +0000550//===----------------------------------------------------------------------===//
551// R600 Pass Setup
552//===----------------------------------------------------------------------===//
553
554bool R600PassConfig::addPreISel() {
555 AMDGPUPassConfig::addPreISel();
Matt Arsenaultc5816112016-06-24 06:30:22 +0000556
557 if (EnableR600StructurizeCFG)
Tom Stellardbc4497b2016-02-12 23:45:29 +0000558 addPass(createStructurizeCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000559 return false;
560}
561
562void R600PassConfig::addPreRegAlloc() {
563 addPass(createR600VectorRegMerger(*TM));
564}
565
566void R600PassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000567 addPass(createR600EmitClauseMarkers(), false);
Matt Arsenault03d85842016-06-27 20:32:13 +0000568 if (EnableR600IfConvert)
Tom Stellard45bb48e2015-06-13 03:28:10 +0000569 addPass(&IfConverterID, false);
570 addPass(createR600ClauseMergePass(*TM), false);
571}
572
573void R600PassConfig::addPreEmitPass() {
574 addPass(createAMDGPUCFGStructurizerPass(), false);
575 addPass(createR600ExpandSpecialInstrsPass(*TM), false);
576 addPass(&FinalizeMachineBundlesID, false);
577 addPass(createR600Packetizer(*TM), false);
578 addPass(createR600ControlFlowFinalizer(*TM), false);
579}
580
581TargetPassConfig *R600TargetMachine::createPassConfig(PassManagerBase &PM) {
582 return new R600PassConfig(this, PM);
583}
584
585//===----------------------------------------------------------------------===//
586// GCN Pass Setup
587//===----------------------------------------------------------------------===//
588
Matt Arsenault03d85842016-06-27 20:32:13 +0000589ScheduleDAGInstrs *GCNPassConfig::createMachineScheduler(
590 MachineSchedContext *C) const {
591 const SISubtarget &ST = C->MF->getSubtarget<SISubtarget>();
592 if (ST.enableSIScheduler())
593 return createSIMachineScheduler(C);
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000594 return createGCNMaxOccupancyMachineScheduler(C);
Matt Arsenault03d85842016-06-27 20:32:13 +0000595}
596
Tom Stellard45bb48e2015-06-13 03:28:10 +0000597bool GCNPassConfig::addPreISel() {
598 AMDGPUPassConfig::addPreISel();
Matt Arsenault39319482015-11-06 18:01:57 +0000599
600 // FIXME: We need to run a pass to propagate the attributes when calls are
601 // supported.
602 addPass(&AMDGPUAnnotateKernelFeaturesID);
Tom Stellardbc4497b2016-02-12 23:45:29 +0000603 addPass(createStructurizeCFGPass(true)); // true -> SkipUniformRegions
Tom Stellard45bb48e2015-06-13 03:28:10 +0000604 addPass(createSinkingPass());
605 addPass(createSITypeRewriter());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000606 addPass(createAMDGPUAnnotateUniformValues());
Tom Stellardbc4497b2016-02-12 23:45:29 +0000607 addPass(createSIAnnotateControlFlowPass());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000608
Tom Stellard45bb48e2015-06-13 03:28:10 +0000609 return false;
610}
611
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000612void GCNPassConfig::addMachineSSAOptimization() {
613 TargetPassConfig::addMachineSSAOptimization();
614
615 // We want to fold operands after PeepholeOptimizer has run (or as part of
616 // it), because it will eliminate extra copies making it easier to fold the
617 // real source operand. We want to eliminate dead instructions after, so that
618 // we see fewer uses of the copies. We then need to clean up the dead
619 // instructions leftover after the operands are folded as well.
620 //
621 // XXX - Can we get away without running DeadMachineInstructionElim again?
622 addPass(&SIFoldOperandsID);
623 addPass(&DeadMachineInstructionElimID);
Tom Stellardc2ff0eb2016-08-29 19:15:22 +0000624 addPass(&SILoadStoreOptimizerID);
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000625}
626
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000627bool GCNPassConfig::addILPOpts() {
628 if (EnableEarlyIfConversion)
629 addPass(&EarlyIfConverterID);
630
631 TargetPassConfig::addILPOpts();
632 return false;
633}
634
Tom Stellard45bb48e2015-06-13 03:28:10 +0000635bool GCNPassConfig::addInstSelector() {
636 AMDGPUPassConfig::addInstSelector();
637 addPass(createSILowerI1CopiesPass());
Matt Arsenault782c03b2015-11-03 22:30:13 +0000638 addPass(&SIFixSGPRCopiesID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000639 return false;
640}
641
Tom Stellard000c5af2016-04-14 19:09:28 +0000642#ifdef LLVM_BUILD_GLOBAL_ISEL
643bool GCNPassConfig::addIRTranslator() {
644 addPass(new IRTranslator());
645 return false;
646}
647
Tim Northover33b07d62016-07-22 20:03:43 +0000648bool GCNPassConfig::addLegalizeMachineIR() {
Tom Stellardca166212017-01-30 21:56:46 +0000649 addPass(new Legalizer());
Tim Northover33b07d62016-07-22 20:03:43 +0000650 return false;
651}
652
Tom Stellard000c5af2016-04-14 19:09:28 +0000653bool GCNPassConfig::addRegBankSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000654 addPass(new RegBankSelect());
Tom Stellard000c5af2016-04-14 19:09:28 +0000655 return false;
656}
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000657
658bool GCNPassConfig::addGlobalInstructionSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000659 addPass(new InstructionSelect());
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000660 return false;
661}
Tom Stellardca166212017-01-30 21:56:46 +0000662
Tom Stellard000c5af2016-04-14 19:09:28 +0000663#endif
664
Tom Stellard45bb48e2015-06-13 03:28:10 +0000665void GCNPassConfig::addPreRegAlloc() {
Matt Arsenault4a07bf62016-06-22 20:26:24 +0000666 addPass(createSIShrinkInstructionsPass());
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000667 addPass(createSIWholeQuadModePass());
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000668}
669
670void GCNPassConfig::addFastRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000671 // FIXME: We have to disable the verifier here because of PHIElimination +
672 // TwoAddressInstructions disabling it.
Matt Arsenaulte6740752016-09-29 01:44:16 +0000673
674 // This must be run immediately after phi elimination and before
675 // TwoAddressInstructions, otherwise the processing of the tied operand of
676 // SI_ELSE will introduce a copy of the tied operand source after the else.
677 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000678
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000679 TargetPassConfig::addFastRegAlloc(RegAllocPass);
680}
681
682void GCNPassConfig::addOptimizedRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000683 // This needs to be run directly before register allocation because earlier
684 // passes might recompute live intervals.
685 insertPass(&MachineSchedulerID, &SIFixControlFlowLiveIntervalsID);
686
Matt Arsenaulte6740752016-09-29 01:44:16 +0000687 // This must be run immediately after phi elimination and before
688 // TwoAddressInstructions, otherwise the processing of the tied operand of
689 // SI_ELSE will introduce a copy of the tied operand source after the else.
690 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000691
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000692 TargetPassConfig::addOptimizedRegAlloc(RegAllocPass);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000693}
694
Matt Arsenaulte6740752016-09-29 01:44:16 +0000695void GCNPassConfig::addPostRegAlloc() {
Stanislav Mekhanoshin22a56f22017-01-24 17:46:17 +0000696 addPass(&SIFixVGPRCopiesID);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000697 addPass(&SIOptimizeExecMaskingID);
698 TargetPassConfig::addPostRegAlloc();
699}
700
Tom Stellard45bb48e2015-06-13 03:28:10 +0000701void GCNPassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000702}
703
704void GCNPassConfig::addPreEmitPass() {
Tom Stellardcb6ba622016-04-30 00:23:06 +0000705 // The hazard recognizer that runs as part of the post-ra scheduler does not
Matt Arsenault254a6452016-06-28 16:59:53 +0000706 // guarantee to be able handle all hazards correctly. This is because if there
707 // are multiple scheduling regions in a basic block, the regions are scheduled
708 // bottom up, so when we begin to schedule a region we don't know what
709 // instructions were emitted directly before it.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000710 //
Matt Arsenault254a6452016-06-28 16:59:53 +0000711 // Here we add a stand-alone hazard recognizer pass which can handle all
712 // cases.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000713 addPass(&PostRAHazardRecognizerID);
714
Matt Arsenaulte2bd9a32016-06-09 23:19:14 +0000715 addPass(createSIInsertWaitsPass());
Matt Arsenaultcf2744f2016-04-29 20:23:42 +0000716 addPass(createSIShrinkInstructionsPass());
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000717 addPass(&SIInsertSkipsPassID);
Matt Arsenault9babdf42016-06-22 20:15:28 +0000718 addPass(createSIDebuggerInsertNopsPass());
Matt Arsenault6bc43d82016-10-06 16:20:41 +0000719 addPass(&BranchRelaxationPassID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000720}
721
722TargetPassConfig *GCNTargetMachine::createPassConfig(PassManagerBase &PM) {
723 return new GCNPassConfig(this, PM);
724}