[AArch64] Add support for TargetTransformInfo Analysis.
[oota-llvm.git] / lib / Target / AArch64 / AArch64TargetMachine.cpp
1 //===-- AArch64TargetMachine.cpp - Define TargetMachine for AArch64 -------===//
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 // This file contains the implementation of the AArch64TargetMachine
11 // methods. Principally just setting up the passes needed to generate correct
12 // code on this architecture.
13 //
14 //===----------------------------------------------------------------------===//
15
16 #include "AArch64.h"
17 #include "AArch64TargetMachine.h"
18 #include "MCTargetDesc/AArch64MCTargetDesc.h"
19 #include "llvm/CodeGen/Passes.h"
20 #include "llvm/PassManager.h"
21 #include "llvm/Support/TargetRegistry.h"
22
23 using namespace llvm;
24
25 extern "C" void LLVMInitializeAArch64Target() {
26   RegisterTargetMachine<AArch64TargetMachine> X(TheAArch64Target);
27 }
28
29 AArch64TargetMachine::AArch64TargetMachine(const Target &T, StringRef TT,
30                                            StringRef CPU, StringRef FS,
31                                            const TargetOptions &Options,
32                                            Reloc::Model RM, CodeModel::Model CM,
33                                            CodeGenOpt::Level OL)
34   : LLVMTargetMachine(T, TT, CPU, FS, Options, RM, CM, OL),
35     Subtarget(TT, CPU, FS),
36     InstrInfo(Subtarget),
37     DL("e-m:e-i64:64-i128:128-n32:64-S128"),
38     TLInfo(*this),
39     TSInfo(*this),
40     FrameLowering(Subtarget) {
41   initAsmInfo();
42 }
43
44 void AArch64TargetMachine::addAnalysisPasses(PassManagerBase &PM) {
45   // Add first the target-independent BasicTTI pass, then our AArch64 pass. This
46   // allows the AArch64 pass to delegate to the target independent layer when
47   // appropriate.
48   PM.add(createBasicTargetTransformInfoPass(this));
49   PM.add(createAArch64TargetTransformInfoPass(this));
50 }
51
52 namespace {
53 /// AArch64 Code Generator Pass Configuration Options.
54 class AArch64PassConfig : public TargetPassConfig {
55 public:
56   AArch64PassConfig(AArch64TargetMachine *TM, PassManagerBase &PM)
57     : TargetPassConfig(TM, PM) {}
58
59   AArch64TargetMachine &getAArch64TargetMachine() const {
60     return getTM<AArch64TargetMachine>();
61   }
62
63   const AArch64Subtarget &getAArch64Subtarget() const {
64     return *getAArch64TargetMachine().getSubtargetImpl();
65   }
66
67   virtual bool addInstSelector();
68   virtual bool addPreEmitPass();
69 };
70 } // namespace
71
72 TargetPassConfig *AArch64TargetMachine::createPassConfig(PassManagerBase &PM) {
73   return new AArch64PassConfig(this, PM);
74 }
75
76 bool AArch64PassConfig::addPreEmitPass() {
77   addPass(&UnpackMachineBundlesID);
78   addPass(createAArch64BranchFixupPass());
79   return true;
80 }
81
82 bool AArch64PassConfig::addInstSelector() {
83   addPass(createAArch64ISelDAG(getAArch64TargetMachine(), getOptLevel()));
84
85   // For ELF, cleanup any local-dynamic TLS accesses.
86   if (getAArch64Subtarget().isTargetELF() && getOptLevel() != CodeGenOpt::None)
87     addPass(createAArch64CleanupLocalDynamicTLSPass());
88
89   return false;
90 }