blob: 497bf6b54add06578b9e605140598398f77b7aba [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 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
Matt Arsenault0699ef32017-02-09 22:00:42 +0000475 addPass(createAMDGPULowerIntrinsicsPass());
476
Tom Stellard45bb48e2015-06-13 03:28:10 +0000477 // Function calls are not supported, so make sure we inline everything.
478 addPass(createAMDGPUAlwaysInlinePass());
Chandler Carruth67fc52f2016-08-17 02:56:20 +0000479 addPass(createAlwaysInlinerLegacyPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000480 // We need to add the barrier noop pass, otherwise adding the function
481 // inlining pass will cause all of the PassConfigs passes to be run
482 // one function at a time, which means if we have a nodule with two
483 // functions, then we will generate code for the first function
484 // without ever running any passes on the second.
485 addPass(createBarrierNoopPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000486
Matt Arsenault0c329382017-01-30 18:40:29 +0000487 const AMDGPUTargetMachine &TM = getAMDGPUTargetMachine();
488
489 if (TM.getTargetTriple().getArch() == Triple::amdgcn) {
490 // TODO: May want to move later or split into an early and late one.
491
492 addPass(createAMDGPUCodeGenPreparePass(
493 static_cast<const GCNTargetMachine *>(&TM)));
494 }
495
Tom Stellardfd253952015-08-07 23:19:30 +0000496 // Handle uses of OpenCL image2d_t, image3d_t and sampler_t arguments.
497 addPass(createAMDGPUOpenCLImageTypeLoweringPass());
Matt Arsenault39319482015-11-06 18:01:57 +0000498
Matt Arsenault03d85842016-06-27 20:32:13 +0000499 if (TM.getOptLevel() > CodeGenOpt::None) {
Matt Arsenault417e0072017-02-08 06:16:04 +0000500 addPass(createInferAddressSpacesPass());
Matt Arsenaulte0132462016-01-30 05:19:45 +0000501 addPass(createAMDGPUPromoteAlloca(&TM));
Matt Arsenault03d85842016-06-27 20:32:13 +0000502
503 if (EnableSROA)
504 addPass(createSROAPass());
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000505
Konstantin Zhuravlyov4658e5f2016-09-30 16:39:24 +0000506 addStraightLineScalarOptimizationPasses();
507 }
Matt Arsenaultf42c6922016-06-15 00:11:01 +0000508
509 TargetPassConfig::addIRPasses();
510
511 // EarlyCSE is not always strong enough to clean up what LSR produces. For
512 // example, GVN can combine
513 //
514 // %0 = add %a, %b
515 // %1 = add %b, %a
516 //
517 // and
518 //
519 // %0 = shl nsw %a, 2
520 // %1 = shl %a, 2
521 //
522 // but EarlyCSE can do neither of them.
523 if (getOptLevel() != CodeGenOpt::None)
524 addEarlyCSEOrGVNPass();
Tom Stellard45bb48e2015-06-13 03:28:10 +0000525}
526
Matt Arsenault908b9e22016-07-01 03:33:52 +0000527void AMDGPUPassConfig::addCodeGenPrepare() {
528 TargetPassConfig::addCodeGenPrepare();
529
530 if (EnableLoadStoreVectorizer)
531 addPass(createLoadStoreVectorizerPass());
532}
533
Matt Arsenault43e92fe2016-06-24 06:30:11 +0000534bool AMDGPUPassConfig::addPreISel() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000535 addPass(createFlattenCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000536 return false;
537}
538
539bool AMDGPUPassConfig::addInstSelector() {
Konstantin Zhuravlyov60a83732016-10-03 18:47:26 +0000540 addPass(createAMDGPUISelDag(getAMDGPUTargetMachine(), getOptLevel()));
Tom Stellard45bb48e2015-06-13 03:28:10 +0000541 return false;
542}
543
Matt Arsenault0a109002015-09-25 17:41:20 +0000544bool AMDGPUPassConfig::addGCPasses() {
545 // Do nothing. GC is not supported.
546 return false;
547}
548
Tom Stellard45bb48e2015-06-13 03:28:10 +0000549//===----------------------------------------------------------------------===//
550// R600 Pass Setup
551//===----------------------------------------------------------------------===//
552
553bool R600PassConfig::addPreISel() {
554 AMDGPUPassConfig::addPreISel();
Matt Arsenaultc5816112016-06-24 06:30:22 +0000555
556 if (EnableR600StructurizeCFG)
Tom Stellardbc4497b2016-02-12 23:45:29 +0000557 addPass(createStructurizeCFGPass());
Tom Stellard45bb48e2015-06-13 03:28:10 +0000558 return false;
559}
560
561void R600PassConfig::addPreRegAlloc() {
562 addPass(createR600VectorRegMerger(*TM));
563}
564
565void R600PassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000566 addPass(createR600EmitClauseMarkers(), false);
Matt Arsenault03d85842016-06-27 20:32:13 +0000567 if (EnableR600IfConvert)
Tom Stellard45bb48e2015-06-13 03:28:10 +0000568 addPass(&IfConverterID, false);
569 addPass(createR600ClauseMergePass(*TM), false);
570}
571
572void R600PassConfig::addPreEmitPass() {
573 addPass(createAMDGPUCFGStructurizerPass(), false);
574 addPass(createR600ExpandSpecialInstrsPass(*TM), false);
575 addPass(&FinalizeMachineBundlesID, false);
576 addPass(createR600Packetizer(*TM), false);
577 addPass(createR600ControlFlowFinalizer(*TM), false);
578}
579
580TargetPassConfig *R600TargetMachine::createPassConfig(PassManagerBase &PM) {
581 return new R600PassConfig(this, PM);
582}
583
584//===----------------------------------------------------------------------===//
585// GCN Pass Setup
586//===----------------------------------------------------------------------===//
587
Matt Arsenault03d85842016-06-27 20:32:13 +0000588ScheduleDAGInstrs *GCNPassConfig::createMachineScheduler(
589 MachineSchedContext *C) const {
590 const SISubtarget &ST = C->MF->getSubtarget<SISubtarget>();
591 if (ST.enableSIScheduler())
592 return createSIMachineScheduler(C);
Tom Stellard0d23ebe2016-08-29 19:42:52 +0000593 return createGCNMaxOccupancyMachineScheduler(C);
Matt Arsenault03d85842016-06-27 20:32:13 +0000594}
595
Tom Stellard45bb48e2015-06-13 03:28:10 +0000596bool GCNPassConfig::addPreISel() {
597 AMDGPUPassConfig::addPreISel();
Matt Arsenault39319482015-11-06 18:01:57 +0000598
599 // FIXME: We need to run a pass to propagate the attributes when calls are
600 // supported.
601 addPass(&AMDGPUAnnotateKernelFeaturesID);
Tom Stellardbc4497b2016-02-12 23:45:29 +0000602 addPass(createStructurizeCFGPass(true)); // true -> SkipUniformRegions
Tom Stellard45bb48e2015-06-13 03:28:10 +0000603 addPass(createSinkingPass());
604 addPass(createSITypeRewriter());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000605 addPass(createAMDGPUAnnotateUniformValues());
Tom Stellardbc4497b2016-02-12 23:45:29 +0000606 addPass(createSIAnnotateControlFlowPass());
Tom Stellarda6f24c62015-12-15 20:55:55 +0000607
Tom Stellard45bb48e2015-06-13 03:28:10 +0000608 return false;
609}
610
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000611void GCNPassConfig::addMachineSSAOptimization() {
612 TargetPassConfig::addMachineSSAOptimization();
613
614 // We want to fold operands after PeepholeOptimizer has run (or as part of
615 // it), because it will eliminate extra copies making it easier to fold the
616 // real source operand. We want to eliminate dead instructions after, so that
617 // we see fewer uses of the copies. We then need to clean up the dead
618 // instructions leftover after the operands are folded as well.
619 //
620 // XXX - Can we get away without running DeadMachineInstructionElim again?
621 addPass(&SIFoldOperandsID);
622 addPass(&DeadMachineInstructionElimID);
Tom Stellardc2ff0eb2016-08-29 19:15:22 +0000623 addPass(&SILoadStoreOptimizerID);
Matt Arsenault3d1c1de2016-04-14 21:58:24 +0000624}
625
Matt Arsenault9f5e0ef2017-01-25 04:25:02 +0000626bool GCNPassConfig::addILPOpts() {
627 if (EnableEarlyIfConversion)
628 addPass(&EarlyIfConverterID);
629
630 TargetPassConfig::addILPOpts();
631 return false;
632}
633
Tom Stellard45bb48e2015-06-13 03:28:10 +0000634bool GCNPassConfig::addInstSelector() {
635 AMDGPUPassConfig::addInstSelector();
636 addPass(createSILowerI1CopiesPass());
Matt Arsenault782c03b2015-11-03 22:30:13 +0000637 addPass(&SIFixSGPRCopiesID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000638 return false;
639}
640
Tom Stellard000c5af2016-04-14 19:09:28 +0000641#ifdef LLVM_BUILD_GLOBAL_ISEL
642bool GCNPassConfig::addIRTranslator() {
643 addPass(new IRTranslator());
644 return false;
645}
646
Tim Northover33b07d62016-07-22 20:03:43 +0000647bool GCNPassConfig::addLegalizeMachineIR() {
Tom Stellardca166212017-01-30 21:56:46 +0000648 addPass(new Legalizer());
Tim Northover33b07d62016-07-22 20:03:43 +0000649 return false;
650}
651
Tom Stellard000c5af2016-04-14 19:09:28 +0000652bool GCNPassConfig::addRegBankSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000653 addPass(new RegBankSelect());
Tom Stellard000c5af2016-04-14 19:09:28 +0000654 return false;
655}
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000656
657bool GCNPassConfig::addGlobalInstructionSelect() {
Tom Stellardca166212017-01-30 21:56:46 +0000658 addPass(new InstructionSelect());
Ahmed Bougacha6756a2c2016-07-27 14:31:55 +0000659 return false;
660}
Tom Stellardca166212017-01-30 21:56:46 +0000661
Tom Stellard000c5af2016-04-14 19:09:28 +0000662#endif
663
Tom Stellard45bb48e2015-06-13 03:28:10 +0000664void GCNPassConfig::addPreRegAlloc() {
Matt Arsenault4a07bf62016-06-22 20:26:24 +0000665 addPass(createSIShrinkInstructionsPass());
Nicolai Haehnle213e87f2016-03-21 20:28:33 +0000666 addPass(createSIWholeQuadModePass());
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000667}
668
669void GCNPassConfig::addFastRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000670 // FIXME: We have to disable the verifier here because of PHIElimination +
671 // TwoAddressInstructions disabling it.
Matt Arsenaulte6740752016-09-29 01:44:16 +0000672
673 // This must be run immediately after phi elimination and before
674 // TwoAddressInstructions, otherwise the processing of the tied operand of
675 // SI_ELSE will introduce a copy of the tied operand source after the else.
676 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000677
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000678 TargetPassConfig::addFastRegAlloc(RegAllocPass);
679}
680
681void GCNPassConfig::addOptimizedRegAlloc(FunctionPass *RegAllocPass) {
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000682 // This needs to be run directly before register allocation because earlier
683 // passes might recompute live intervals.
684 insertPass(&MachineSchedulerID, &SIFixControlFlowLiveIntervalsID);
685
Matt Arsenaulte6740752016-09-29 01:44:16 +0000686 // This must be run immediately after phi elimination and before
687 // TwoAddressInstructions, otherwise the processing of the tied operand of
688 // SI_ELSE will introduce a copy of the tied operand source after the else.
689 insertPass(&PHIEliminationID, &SILowerControlFlowID, false);
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000690
Matt Arsenaultb87fc222015-10-01 22:10:03 +0000691 TargetPassConfig::addOptimizedRegAlloc(RegAllocPass);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000692}
693
Matt Arsenaulte6740752016-09-29 01:44:16 +0000694void GCNPassConfig::addPostRegAlloc() {
Stanislav Mekhanoshin22a56f22017-01-24 17:46:17 +0000695 addPass(&SIFixVGPRCopiesID);
Matt Arsenaulte6740752016-09-29 01:44:16 +0000696 addPass(&SIOptimizeExecMaskingID);
697 TargetPassConfig::addPostRegAlloc();
698}
699
Tom Stellard45bb48e2015-06-13 03:28:10 +0000700void GCNPassConfig::addPreSched2() {
Tom Stellard45bb48e2015-06-13 03:28:10 +0000701}
702
703void GCNPassConfig::addPreEmitPass() {
Tom Stellardcb6ba622016-04-30 00:23:06 +0000704 // The hazard recognizer that runs as part of the post-ra scheduler does not
Matt Arsenault254a6452016-06-28 16:59:53 +0000705 // guarantee to be able handle all hazards correctly. This is because if there
706 // are multiple scheduling regions in a basic block, the regions are scheduled
707 // bottom up, so when we begin to schedule a region we don't know what
708 // instructions were emitted directly before it.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000709 //
Matt Arsenault254a6452016-06-28 16:59:53 +0000710 // Here we add a stand-alone hazard recognizer pass which can handle all
711 // cases.
Tom Stellardcb6ba622016-04-30 00:23:06 +0000712 addPass(&PostRAHazardRecognizerID);
713
Matt Arsenaulte2bd9a32016-06-09 23:19:14 +0000714 addPass(createSIInsertWaitsPass());
Matt Arsenaultcf2744f2016-04-29 20:23:42 +0000715 addPass(createSIShrinkInstructionsPass());
Matt Arsenault78fc9da2016-08-22 19:33:16 +0000716 addPass(&SIInsertSkipsPassID);
Matt Arsenault9babdf42016-06-22 20:15:28 +0000717 addPass(createSIDebuggerInsertNopsPass());
Matt Arsenault6bc43d82016-10-06 16:20:41 +0000718 addPass(&BranchRelaxationPassID);
Tom Stellard45bb48e2015-06-13 03:28:10 +0000719}
720
721TargetPassConfig *GCNTargetMachine::createPassConfig(PassManagerBase &PM) {
722 return new GCNPassConfig(this, PM);
723}