blob: 345dc6b4fbf3e246c159f83b2ecb7a086339a3b5 [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 Arsenaulte0132462016-01-30 05:19:45 +0000111 initializeAMDGPUPromoteAllocaPass(*PR);
Matt Arsenault86de4862016-06-24 07:07:55 +0000112 initializeAMDGPUCodeGenPreparePass(*PR);
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +0000113 initializeAMDGPUUnifyMetadataPass(*PR);
Tom Stellard77a17772016-01-20 15:48:27 +0000114 initializeSIAnnotateControlFlowPass(*PR);
Tom Stellard6e1967e2016-02-05 17:42:38 +0000115 initializeSIInsertWaitsPass(*PR);
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000116 initializeSIWholeQuadModePass(*PR);
Matt Arsenault55d49cf2016-02-12 02:16:10 +0000117 initializeSILowerControlFlowPass(*PR);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000118 initializeSIInsertSkipsPass(*PR);
Matt Arsenaultd3e4c642016-06-02 00:04:22 +0000119 initializeSIDebuggerInsertNopsPass(*PR);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000120 initializeSIOptimizeExecMaskingPass(*PR);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000121}
122
Tom Stellarde135ffd2015-09-25 21:41:28 +0000123static std::unique_ptr<TargetLoweringObjectFile> createTLOF(const Triple &TT) {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000124 return llvm::make_unique<AMDGPUTargetObjectFile>();
Tom Stellarde135ffd2015-09-25 21:41:28 +0000125}
126
Tom Stellard45bb48e2015-06-13 03:28:10 +0000127static ScheduleDAGInstrs *createR600MachineScheduler(MachineSchedContext *C) {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000128 return new ScheduleDAGMILive(C, llvm::make_unique<R600SchedStrategy>());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000129}
130
Matt Arsenault2ffe8fd2016-08-11 19:18:50 +0000131static ScheduleDAGInstrs *createSIMachineScheduler(MachineSchedContext *C) {
132 return new SIScheduleDAGMI(C);
133}
134
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000135static ScheduleDAGInstrs *
136createGCNMaxOccupancyMachineScheduler(MachineSchedContext *C) {
137 ScheduleDAGMILive *DAG =
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000138 new ScheduleDAGMILive(C,
139 llvm::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 Mekhanoshina3b72792017-01-30 21:05:18 +0000219 bool Internalize = InternalizeSymbols &&
220 (getOptLevel() > CodeGenOpt::None) &&
221 (getTargetTriple().getArch() == Triple::amdgcn);
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000222 Builder.addExtension(
Stanislav Mekhanoshinf6c1feb2017-01-27 16:38:10 +0000223 PassManagerBuilder::EP_ModuleOptimizerEarly,
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000224 [Internalize](const PassManagerBuilder &, legacy::PassManagerBase &PM) {
Stanislav Mekhanoshin81598112017-01-26 16:49:08 +0000225 PM.add(createAMDGPUUnifyMetadataPass());
Stanislav Mekhanoshina3b72792017-01-30 21:05:18 +0000226 if (Internalize) {
227 PM.add(createInternalizePass([=](const GlobalValue &GV) -> bool {
228 if (const Function *F = dyn_cast<Function>(&GV)) {
229 if (F->isDeclaration())
230 return true;
231 switch (F->getCallingConv()) {
232 default:
233 return false;
234 case CallingConv::AMDGPU_VS:
235 case CallingConv::AMDGPU_GS:
236 case CallingConv::AMDGPU_PS:
237 case CallingConv::AMDGPU_CS:
238 case CallingConv::AMDGPU_KERNEL:
239 case CallingConv::SPIR_KERNEL:
240 return true;
241 }
242 }
243 return !GV.use_empty();
244 }));
245 PM.add(createGlobalDCEPass());
246 }
247 });
Stanislav Mekhanoshin50ea93a2016-12-08 19:46:04 +0000248}
249
Tom Stellard45bb48e2015-06-13 03:28:10 +0000250//===----------------------------------------------------------------------===//
251// R600 Target Machine (R600 -> Cayman)
252//===----------------------------------------------------------------------===//
253
254R600TargetMachine::R600TargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000255 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000256 TargetOptions Options,
257 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000258 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenaultad55ee52016-12-06 01:02:51 +0000259 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {
260 setRequiresStructuredCFG(true);
261}
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000262
263const R600Subtarget *R600TargetMachine::getSubtargetImpl(
264 const Function &F) const {
265 StringRef GPU = getGPUName(F);
266 StringRef FS = getFeatureString(F);
267
268 SmallString<128> SubtargetKey(GPU);
269 SubtargetKey.append(FS);
270
271 auto &I = SubtargetMap[SubtargetKey];
272 if (!I) {
273 // This needs to be done before we create a new subtarget since any
274 // creation will depend on the TM and the code generation flags on the
275 // function that reside in TargetOptions.
276 resetTargetOptions(F);
277 I = llvm::make_unique<R600Subtarget>(TargetTriple, GPU, FS, *this);
278 }
279
280 return I.get();
281}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000282
283//===----------------------------------------------------------------------===//
284// GCN Target Machine (SI+)
285//===----------------------------------------------------------------------===//
286
Matt Arsenault55dff272016-06-28 00:11:26 +0000287#ifdef LLVM_BUILD_GLOBAL_ISEL
288namespace {
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000289
Matt Arsenault55dff272016-06-28 00:11:26 +0000290struct SIGISelActualAccessor : public GISelAccessor {
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000291 std::unique_ptr<AMDGPUCallLowering> CallLoweringInfo;
Tom Stellardca166212017-01-30 21:56:46 +0000292 std::unique_ptr<InstructionSelector> InstSelector;
293 std::unique_ptr<LegalizerInfo> Legalizer;
294 std::unique_ptr<RegisterBankInfo> RegBankInfo;
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000295 const AMDGPUCallLowering *getCallLowering() const override {
Matt Arsenault55dff272016-06-28 00:11:26 +0000296 return CallLoweringInfo.get();
297 }
Tom Stellardca166212017-01-30 21:56:46 +0000298 const InstructionSelector *getInstructionSelector() const override {
299 return InstSelector.get();
300 }
301 const LegalizerInfo *getLegalizerInfo() const override {
302 return Legalizer.get();
303 }
304 const RegisterBankInfo *getRegBankInfo() const override {
305 return RegBankInfo.get();
306 }
Matt Arsenault55dff272016-06-28 00:11:26 +0000307};
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000308
309} // end anonymous namespace
Matt Arsenault55dff272016-06-28 00:11:26 +0000310#endif
311
Tom Stellard45bb48e2015-06-13 03:28:10 +0000312GCNTargetMachine::GCNTargetMachine(const Target &T, const Triple &TT,
Tom Stellard5dde1d22016-02-05 18:29:17 +0000313 StringRef CPU, StringRef FS,
Rafael Espindola8c34dd82016-05-18 22:04:49 +0000314 TargetOptions Options,
315 Optional<Reloc::Model> RM,
Tom Stellard45bb48e2015-06-13 03:28:10 +0000316 CodeModel::Model CM, CodeGenOpt::Level OL)
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000317 : AMDGPUTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL) {}
318
319const SISubtarget *GCNTargetMachine::getSubtargetImpl(const Function &F) const {
320 StringRef GPU = getGPUName(F);
321 StringRef FS = getFeatureString(F);
322
323 SmallString<128> SubtargetKey(GPU);
324 SubtargetKey.append(FS);
325
326 auto &I = SubtargetMap[SubtargetKey];
327 if (!I) {
328 // This needs to be done before we create a new subtarget since any
329 // creation will depend on the TM and the code generation flags on the
330 // function that reside in TargetOptions.
331 resetTargetOptions(F);
332 I = llvm::make_unique<SISubtarget>(TargetTriple, GPU, FS, *this);
333
334#ifndef LLVM_BUILD_GLOBAL_ISEL
335 GISelAccessor *GISel = new GISelAccessor();
336#else
337 SIGISelActualAccessor *GISel = new SIGISelActualAccessor();
Matt Arsenaulteb9025d2016-06-28 17:42:09 +0000338 GISel->CallLoweringInfo.reset(
339 new AMDGPUCallLowering(*I->getTargetLowering()));
Tom Stellardca166212017-01-30 21:56:46 +0000340 GISel->Legalizer.reset(new AMDGPULegalizerInfo());
341
342 GISel->RegBankInfo.reset(new AMDGPURegisterBankInfo(*I->getRegisterInfo()));
343 GISel->InstSelector.reset(new AMDGPUInstructionSelector(*I,
344 *static_cast<AMDGPURegisterBankInfo*>(GISel->RegBankInfo.get())));
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000345#endif
346
347 I->setGISelAccessor(*GISel);
348 }
349
Alexander Timofeev18009562016-12-08 17:28:47 +0000350 I->setScalarizeGlobalBehavior(ScalarizeGlobal);
351
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000352 return I.get();
353}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000354
355//===----------------------------------------------------------------------===//
356// AMDGPU Pass Setup
357//===----------------------------------------------------------------------===//
358
359namespace {
Tom Stellardcc7067a62016-03-03 03:53:29 +0000360
Tom Stellard45bb48e2015-06-13 03:28:10 +0000361class AMDGPUPassConfig : public TargetPassConfig {
362public:
363 AMDGPUPassConfig(TargetMachine *TM, PassManagerBase &PM)
Matt Arsenault0a109002015-09-25 17:41:20 +0000364 : TargetPassConfig(TM, PM) {
Matt Arsenault0a109002015-09-25 17:41:20 +0000365 // Exceptions and StackMaps are not supported, so these passes will never do
366 // anything.
367 disablePass(&StackMapLivenessID);
368 disablePass(&FuncletLayoutID);
369 }
Tom Stellard45bb48e2015-06-13 03:28:10 +0000370
371 AMDGPUTargetMachine &getAMDGPUTargetMachine() const {
372 return getTM<AMDGPUTargetMachine>();
373 }
374
Matthias Braun115efcd2016-11-28 20:11:54 +0000375 ScheduleDAGInstrs *
376 createMachineScheduler(MachineSchedContext *C) const override {
377 ScheduleDAGMILive *DAG = createGenericSchedLive(C);
378 DAG->addMutation(createLoadClusterDAGMutation(DAG->TII, DAG->TRI));
379 DAG->addMutation(createStoreClusterDAGMutation(DAG->TII, DAG->TRI));
380 return DAG;
381 }
382
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000383 void addEarlyCSEOrGVNPass();
384 void addStraightLineScalarOptimizationPasses();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000385 void addIRPasses() override;
Matt Arsenault908b9e22016-07-01 03:33:52 +0000386 void addCodeGenPrepare() override;
Matt Arsenault0a109002015-09-25 17:41:20 +0000387 bool addPreISel() override;
388 bool addInstSelector() override;
389 bool addGCPasses() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000390};
391
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000392class R600PassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000393public:
394 R600PassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000395 : AMDGPUPassConfig(TM, PM) {}
Tom Stellard45bb48e2015-06-13 03:28:10 +0000396
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000397 ScheduleDAGInstrs *createMachineScheduler(
398 MachineSchedContext *C) const override {
399 return createR600MachineScheduler(C);
400 }
401
Tom Stellard45bb48e2015-06-13 03:28:10 +0000402 bool addPreISel() override;
403 void addPreRegAlloc() override;
404 void addPreSched2() override;
405 void addPreEmitPass() override;
406};
407
Matt Arsenault6b6a2c32016-03-11 08:00:27 +0000408class GCNPassConfig final : public AMDGPUPassConfig {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000409public:
410 GCNPassConfig(TargetMachine *TM, PassManagerBase &PM)
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000411 : AMDGPUPassConfig(TM, PM) {}
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000412
413 GCNTargetMachine &getGCNTargetMachine() const {
414 return getTM<GCNTargetMachine>();
415 }
416
417 ScheduleDAGInstrs *
Matt Arsenault03d85842016-06-27 20:32:13 +0000418 createMachineScheduler(MachineSchedContext *C) const override;
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000419
Tom Stellard45bb48e2015-06-13 03:28:10 +0000420 bool addPreISel() override;
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000421 void addMachineSSAOptimization() override;
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000422 bool addILPOpts() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000423 bool addInstSelector() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000424#ifdef LLVM_BUILD_GLOBAL_ISEL
425 bool addIRTranslator() override;
Tim Northover33b07d62016-07-22 20:03:43 +0000426 bool addLegalizeMachineIR() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000427 bool addRegBankSelect() override;
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000428 bool addGlobalInstructionSelect() override;
Tom Stellard000c5af2016-04-14 19:09:28 +0000429#endif
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000430 void addFastRegAlloc(FunctionPass *RegAllocPass) override;
431 void addOptimizedRegAlloc(FunctionPass *RegAllocPass) override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000432 void addPreRegAlloc() override;
Matt Arsenaulte6740752016-09-29 01:44:16 +0000433 void addPostRegAlloc() override;
Tom Stellard45bb48e2015-06-13 03:28:10 +0000434 void addPreSched2() override;
435 void addPreEmitPass() override;
436};
437
Eugene Zelenko6a9226d2016-12-12 22:23:53 +0000438} // end anonymous namespace
Tom Stellard45bb48e2015-06-13 03:28:10 +0000439
440TargetIRAnalysis AMDGPUTargetMachine::getTargetIRAnalysis() {
Eric Christophera4e5d3c2015-09-16 23:38:13 +0000441 return TargetIRAnalysis([this](const Function &F) {
Matt Arsenault59c0ffa2016-06-27 20:48:03 +0000442 return TargetTransformInfo(AMDGPUTTIImpl(this, F));
Mehdi Amini5010ebf2015-07-09 02:08:42 +0000443 });
Tom Stellard45bb48e2015-06-13 03:28:10 +0000444}
445
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000446void AMDGPUPassConfig::addEarlyCSEOrGVNPass() {
447 if (getOptLevel() == CodeGenOpt::Aggressive)
448 addPass(createGVNPass());
449 else
450 addPass(createEarlyCSEPass());
451}
452
453void AMDGPUPassConfig::addStraightLineScalarOptimizationPasses() {
454 addPass(createSeparateConstOffsetFromGEPPass());
455 addPass(createSpeculativeExecutionPass());
456 // ReassociateGEPs exposes more opportunites for SLSR. See
457 // the example in reassociate-geps-and-slsr.ll.
458 addPass(createStraightLineStrengthReducePass());
459 // SeparateConstOffsetFromGEP and SLSR creates common expressions which GVN or
460 // EarlyCSE can reuse.
461 addEarlyCSEOrGVNPass();
462 // Run NaryReassociate after EarlyCSE/GVN to be more effective.
463 addPass(createNaryReassociatePass());
464 // NaryReassociate on GEPs creates redundant common expressions, so run
465 // EarlyCSE after it.
466 addPass(createEarlyCSEPass());
467}
468
Tom Stellard45bb48e2015-06-13 03:28:10 +0000469void AMDGPUPassConfig::addIRPasses() {
Matt Arsenaultbde80342016-05-18 15:41:07 +0000470 // There is no reason to run these.
471 disablePass(&StackMapLivenessID);
472 disablePass(&FuncletLayoutID);
473 disablePass(&PatchableFunctionID);
474
Tom Stellard45bb48e2015-06-13 03:28:10 +0000475 // Function calls are not supported, so make sure we inline everything.
476 addPass(createAMDGPUAlwaysInlinePass());
Chandler Carruth67fc52f2016-08-17 02:56:20 +0000477 addPass(createAlwaysInlinerLegacyPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000478 // We need to add the barrier noop pass, otherwise adding the function
479 // inlining pass will cause all of the PassConfigs passes to be run
480 // one function at a time, which means if we have a nodule with two
481 // functions, then we will generate code for the first function
482 // without ever running any passes on the second.
483 addPass(createBarrierNoopPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000484
Matt Arsenault0c329382017-01-30 18:40:29 +0000485 const AMDGPUTargetMachine &TM = getAMDGPUTargetMachine();
486
487 if (TM.getTargetTriple().getArch() == Triple::amdgcn) {
488 // TODO: May want to move later or split into an early and late one.
489
490 addPass(createAMDGPUCodeGenPreparePass(
491 static_cast<const GCNTargetMachine *>(&TM)));
492 }
493
Tom Stellardfd253952015-08-07 23:19:30 +0000494 // Handle uses of OpenCL image2d_t, image3d_t and sampler_t arguments.
495 addPass(createAMDGPUOpenCLImageTypeLoweringPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000496
Matt Arsenault03d85842016-06-27 20:32:13 +0000497 if (TM.getOptLevel() > CodeGenOpt::None) {
Matt Arsenaulte0132462016-01-30 05:19:45 +0000498 addPass(createAMDGPUPromoteAlloca(&TM));
Matt Arsenault03d85842016-06-27 20:32:13 +0000499
500 if (EnableSROA)
501 addPass(createSROAPass());
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000502
Konstantin Zhuravlyov4658e5f2016-09-30 16:39:24 +0000503 addStraightLineScalarOptimizationPasses();
504 }
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000505
506 TargetPassConfig::addIRPasses();
507
508 // EarlyCSE is not always strong enough to clean up what LSR produces. For
509 // example, GVN can combine
510 //
511 // %0 = add %a, %b
512 // %1 = add %b, %a
513 //
514 // and
515 //
516 // %0 = shl nsw %a, 2
517 // %1 = shl %a, 2
518 //
519 // but EarlyCSE can do neither of them.
520 if (getOptLevel() != CodeGenOpt::None)
521 addEarlyCSEOrGVNPass();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000522}
523
Matt Arsenault908b9e22016-07-01 03:33:52 +0000524void AMDGPUPassConfig::addCodeGenPrepare() {
525 TargetPassConfig::addCodeGenPrepare();
526
527 if (EnableLoadStoreVectorizer)
528 addPass(createLoadStoreVectorizerPass());
529}
530
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000531bool AMDGPUPassConfig::addPreISel() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000532 addPass(createFlattenCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000533 return false;
534}
535
536bool AMDGPUPassConfig::addInstSelector() {
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +0000537 addPass(createAMDGPUISelDag(getAMDGPUTargetMachine(), getOptLevel()));
Tom Stellard45bb48e2015-06-13 03:28:10 +0000538 return false;
539}
540
Matt Arsenault0a109002015-09-25 17:41:20 +0000541bool AMDGPUPassConfig::addGCPasses() {
542 // Do nothing. GC is not supported.
543 return false;
544}
545
Tom Stellard45bb48e2015-06-13 03:28:10 +0000546//===----------------------------------------------------------------------===//
547// R600 Pass Setup
548//===----------------------------------------------------------------------===//
549
550bool R600PassConfig::addPreISel() {
551 AMDGPUPassConfig::addPreISel();
Matt Arsenaultc5816112016-06-24 06:30:22 +0000552
553 if (EnableR600StructurizeCFG)
Tom Stellardbc4497b2016-02-12 23:45:29 +0000554 addPass(createStructurizeCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000555 return false;
556}
557
558void R600PassConfig::addPreRegAlloc() {
559 addPass(createR600VectorRegMerger(*TM));
560}
561
562void R600PassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000563 addPass(createR600EmitClauseMarkers(), false);
Matt Arsenault03d85842016-06-27 20:32:13 +0000564 if (EnableR600IfConvert)
Tom Stellard45bb48e2015-06-13 03:28:10 +0000565 addPass(&IfConverterID, false);
566 addPass(createR600ClauseMergePass(*TM), false);
567}
568
569void R600PassConfig::addPreEmitPass() {
570 addPass(createAMDGPUCFGStructurizerPass(), false);
571 addPass(createR600ExpandSpecialInstrsPass(*TM), false);
572 addPass(&FinalizeMachineBundlesID, false);
573 addPass(createR600Packetizer(*TM), false);
574 addPass(createR600ControlFlowFinalizer(*TM), false);
575}
576
577TargetPassConfig *R600TargetMachine::createPassConfig(PassManagerBase &PM) {
578 return new R600PassConfig(this, PM);
579}
580
581//===----------------------------------------------------------------------===//
582// GCN Pass Setup
583//===----------------------------------------------------------------------===//
584
Matt Arsenault03d85842016-06-27 20:32:13 +0000585ScheduleDAGInstrs *GCNPassConfig::createMachineScheduler(
586 MachineSchedContext *C) const {
587 const SISubtarget &ST = C->MF->getSubtarget<SISubtarget>();
588 if (ST.enableSIScheduler())
589 return createSIMachineScheduler(C);
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000590 return createGCNMaxOccupancyMachineScheduler(C);
Matt Arsenault03d85842016-06-27 20:32:13 +0000591}
592
Tom Stellard45bb48e2015-06-13 03:28:10 +0000593bool GCNPassConfig::addPreISel() {
594 AMDGPUPassConfig::addPreISel();
Matt Arsenault39319482015-11-06 18:01:57 +0000595
596 // FIXME: We need to run a pass to propagate the attributes when calls are
597 // supported.
598 addPass(&AMDGPUAnnotateKernelFeaturesID);
Tom Stellardbc4497b2016-02-12 23:45:29 +0000599 addPass(createStructurizeCFGPass(true)); // true -> SkipUniformRegions
Tom Stellard45bb48e2015-06-13 03:28:10 +0000600 addPass(createSinkingPass());
601 addPass(createSITypeRewriter());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000602 addPass(createAMDGPUAnnotateUniformValues());
Tom Stellardbc4497b2016-02-12 23:45:29 +0000603 addPass(createSIAnnotateControlFlowPass());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000604
Tom Stellard45bb48e2015-06-13 03:28:10 +0000605 return false;
606}
607
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000608void GCNPassConfig::addMachineSSAOptimization() {
609 TargetPassConfig::addMachineSSAOptimization();
610
611 // We want to fold operands after PeepholeOptimizer has run (or as part of
612 // it), because it will eliminate extra copies making it easier to fold the
613 // real source operand. We want to eliminate dead instructions after, so that
614 // we see fewer uses of the copies. We then need to clean up the dead
615 // instructions leftover after the operands are folded as well.
616 //
617 // XXX - Can we get away without running DeadMachineInstructionElim again?
618 addPass(&SIFoldOperandsID);
619 addPass(&DeadMachineInstructionElimID);
Tom Stellardc2ff0eb2016-08-29 19:15:22 +0000620 addPass(&SILoadStoreOptimizerID);
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000621}
622
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000623bool GCNPassConfig::addILPOpts() {
624 if (EnableEarlyIfConversion)
625 addPass(&EarlyIfConverterID);
626
627 TargetPassConfig::addILPOpts();
628 return false;
629}
630
Tom Stellard45bb48e2015-06-13 03:28:10 +0000631bool GCNPassConfig::addInstSelector() {
632 AMDGPUPassConfig::addInstSelector();
633 addPass(createSILowerI1CopiesPass());
Matt Arsenault782c03b2015-11-03 22:30:13 +0000634 addPass(&SIFixSGPRCopiesID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000635 return false;
636}
637
Tom Stellard000c5af2016-04-14 19:09:28 +0000638#ifdef LLVM_BUILD_GLOBAL_ISEL
639bool GCNPassConfig::addIRTranslator() {
640 addPass(new IRTranslator());
641 return false;
642}
643
Tim Northover33b07d62016-07-22 20:03:43 +0000644bool GCNPassConfig::addLegalizeMachineIR() {
Tom Stellardca166212017-01-30 21:56:46 +0000645 addPass(new Legalizer());
Tim Northover33b07d62016-07-22 20:03:43 +0000646 return false;
647}
648
Tom Stellard000c5af2016-04-14 19:09:28 +0000649bool GCNPassConfig::addRegBankSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000650 addPass(new RegBankSelect());
Tom Stellard000c5af2016-04-14 19:09:28 +0000651 return false;
652}
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000653
654bool GCNPassConfig::addGlobalInstructionSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000655 addPass(new InstructionSelect());
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000656 return false;
657}
Tom Stellardca166212017-01-30 21:56:46 +0000658
Tom Stellard000c5af2016-04-14 19:09:28 +0000659#endif
660
Tom Stellard45bb48e2015-06-13 03:28:10 +0000661void GCNPassConfig::addPreRegAlloc() {
Matt Arsenault4a07bf62016-06-22 20:26:24 +0000662 addPass(createSIShrinkInstructionsPass());
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000663 addPass(createSIWholeQuadModePass());
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000664}
665
666void GCNPassConfig::addFastRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000667 // FIXME: We have to disable the verifier here because of PHIElimination +
668 // TwoAddressInstructions disabling it.
Matt Arsenaulte6740752016-09-29 01:44:16 +0000669
670 // This must be run immediately after phi elimination and before
671 // TwoAddressInstructions, otherwise the processing of the tied operand of
672 // SI_ELSE will introduce a copy of the tied operand source after the else.
673 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000674
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000675 TargetPassConfig::addFastRegAlloc(RegAllocPass);
676}
677
678void GCNPassConfig::addOptimizedRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000679 // This needs to be run directly before register allocation because earlier
680 // passes might recompute live intervals.
681 insertPass(&MachineSchedulerID, &SIFixControlFlowLiveIntervalsID);
682
Matt Arsenaulte6740752016-09-29 01:44:16 +0000683 // This must be run immediately after phi elimination and before
684 // TwoAddressInstructions, otherwise the processing of the tied operand of
685 // SI_ELSE will introduce a copy of the tied operand source after the else.
686 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000687
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000688 TargetPassConfig::addOptimizedRegAlloc(RegAllocPass);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000689}
690
Matt Arsenaulte6740752016-09-29 01:44:16 +0000691void GCNPassConfig::addPostRegAlloc() {
Stanislav Mekhanoshin22a56f22017-01-24 17:46:17 +0000692 addPass(&SIFixVGPRCopiesID);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000693 addPass(&SIOptimizeExecMaskingID);
694 TargetPassConfig::addPostRegAlloc();
695}
696
Tom Stellard45bb48e2015-06-13 03:28:10 +0000697void GCNPassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000698}
699
700void GCNPassConfig::addPreEmitPass() {
Tom Stellardcb6ba622016-04-30 00:23:06 +0000701 // The hazard recognizer that runs as part of the post-ra scheduler does not
Matt Arsenault254a6452016-06-28 16:59:53 +0000702 // guarantee to be able handle all hazards correctly. This is because if there
703 // are multiple scheduling regions in a basic block, the regions are scheduled
704 // bottom up, so when we begin to schedule a region we don't know what
705 // instructions were emitted directly before it.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000706 //
Matt Arsenault254a6452016-06-28 16:59:53 +0000707 // Here we add a stand-alone hazard recognizer pass which can handle all
708 // cases.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000709 addPass(&PostRAHazardRecognizerID);
710
Matt Arsenaulte2bd9a32016-06-09 23:19:14 +0000711 addPass(createSIInsertWaitsPass());
Matt Arsenaultcf2744f2016-04-29 20:23:42 +0000712 addPass(createSIShrinkInstructionsPass());
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000713 addPass(&SIInsertSkipsPassID);
Matt Arsenault9babdf42016-06-22 20:15:28 +0000714 addPass(createSIDebuggerInsertNopsPass());
Matt Arsenault6bc43d82016-10-06 16:20:41 +0000715 addPass(&BranchRelaxationPassID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000716}
717
718TargetPassConfig *GCNTargetMachine::createPassConfig(PassManagerBase &PM) {
719 return new GCNPassConfig(this, PM);
720}