From ebcee99c0213806f7a47aa6d562d9699c430611e Mon Sep 17 00:00:00 2001 From: Chad Rosier Date: Thu, 20 Feb 2014 16:00:08 +0000 Subject: [PATCH] [AArch64] Add support for TargetTransformInfo Analysis. llvm-svn: 201793 --- lib/Target/AArch64/AArch64.h | 4 + lib/Target/AArch64/AArch64TargetMachine.cpp | 8 ++ lib/Target/AArch64/AArch64TargetMachine.h | 2 + .../AArch64/AArch64TargetTransformInfo.cpp | 116 ++++++++++++++++++ lib/Target/AArch64/CMakeLists.txt | 1 + 5 files changed, 131 insertions(+) create mode 100644 lib/Target/AArch64/AArch64TargetTransformInfo.cpp diff --git a/lib/Target/AArch64/AArch64.h b/lib/Target/AArch64/AArch64.h index 4de4faa5818..0297de120cb 100644 --- a/lib/Target/AArch64/AArch64.h +++ b/lib/Target/AArch64/AArch64.h @@ -33,6 +33,10 @@ FunctionPass *createAArch64CleanupLocalDynamicTLSPass(); FunctionPass *createAArch64BranchFixupPass(); +/// \brief Creates an AArch64-specific Target Transformation Info pass. +ImmutablePass *createAArch64TargetTransformInfoPass( + const AArch64TargetMachine *TM); + void LowerAArch64MachineInstrToMCInst(const MachineInstr *MI, MCInst &OutMI, AArch64AsmPrinter &AP); diff --git a/lib/Target/AArch64/AArch64TargetMachine.cpp b/lib/Target/AArch64/AArch64TargetMachine.cpp index 60bd3b9665a..77bdf002f74 100644 --- a/lib/Target/AArch64/AArch64TargetMachine.cpp +++ b/lib/Target/AArch64/AArch64TargetMachine.cpp @@ -41,6 +41,14 @@ AArch64TargetMachine::AArch64TargetMachine(const Target &T, StringRef TT, initAsmInfo(); } +void AArch64TargetMachine::addAnalysisPasses(PassManagerBase &PM) { + // Add first the target-independent BasicTTI pass, then our AArch64 pass. This + // allows the AArch64 pass to delegate to the target independent layer when + // appropriate. + PM.add(createBasicTargetTransformInfoPass(this)); + PM.add(createAArch64TargetTransformInfoPass(this)); +} + namespace { /// AArch64 Code Generator Pass Configuration Options. class AArch64PassConfig : public TargetPassConfig { diff --git a/lib/Target/AArch64/AArch64TargetMachine.h b/lib/Target/AArch64/AArch64TargetMachine.h index c1f47c2e537..14c3f6c6ded 100644 --- a/lib/Target/AArch64/AArch64TargetMachine.h +++ b/lib/Target/AArch64/AArch64TargetMachine.h @@ -62,6 +62,8 @@ public: return &InstrInfo.getRegisterInfo(); } TargetPassConfig *createPassConfig(PassManagerBase &PM); + + virtual void addAnalysisPasses(PassManagerBase &PM); }; } diff --git a/lib/Target/AArch64/AArch64TargetTransformInfo.cpp b/lib/Target/AArch64/AArch64TargetTransformInfo.cpp new file mode 100644 index 00000000000..ce442c28032 --- /dev/null +++ b/lib/Target/AArch64/AArch64TargetTransformInfo.cpp @@ -0,0 +1,116 @@ +//===- AArch64TargetTransformInfo.cpp - AArch64 specific TTI pass ---------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +/// \file +/// This file implements a TargetTransformInfo analysis pass specific to the +/// AArch64 target machine. It uses the target's detailed information to provide +/// more precise answers to certain TTI queries, while letting the target +/// independent and default TTI implementations handle the rest. +/// +//===----------------------------------------------------------------------===// + +#define DEBUG_TYPE "aarch64tti" +#include "AArch64.h" +#include "AArch64TargetMachine.h" +#include "llvm/Analysis/TargetTransformInfo.h" +#include "llvm/Support/Debug.h" +#include "llvm/Target/CostTable.h" +#include "llvm/Target/TargetLowering.h" +using namespace llvm; + +// Declare the pass initialization routine locally as target-specific passes +// don't have a target-wide initialization entry point, and so we rely on the +// pass constructor initialization. +namespace llvm { +void initializeAArch64TTIPass(PassRegistry &); +} + +namespace { + +class AArch64TTI LLVM_FINAL : public ImmutablePass, public TargetTransformInfo { + const AArch64TargetMachine *TM; + const AArch64Subtarget *ST; + const AArch64TargetLowering *TLI; + + /// Estimate the overhead of scalarizing an instruction. Insert and Extract + /// are set if the result needs to be inserted and/or extracted from vectors. + unsigned getScalarizationOverhead(Type *Ty, bool Insert, bool Extract) const; + +public: + AArch64TTI() : ImmutablePass(ID), TM(0), ST(0), TLI(0) { + llvm_unreachable("This pass cannot be directly constructed"); + } + + AArch64TTI(const AArch64TargetMachine *TM) + : ImmutablePass(ID), TM(TM), ST(TM->getSubtargetImpl()), + TLI(TM->getTargetLowering()) { + initializeAArch64TTIPass(*PassRegistry::getPassRegistry()); + } + + virtual void initializePass() LLVM_OVERRIDE { + pushTTIStack(this); + } + + virtual void finalizePass() { + popTTIStack(); + } + + virtual void getAnalysisUsage(AnalysisUsage &AU) const LLVM_OVERRIDE { + TargetTransformInfo::getAnalysisUsage(AU); + } + + /// Pass identification. + static char ID; + + /// Provide necessary pointer adjustments for the two base classes. + virtual void *getAdjustedAnalysisPointer(const void *ID) LLVM_OVERRIDE { + if (ID == &TargetTransformInfo::ID) + return (TargetTransformInfo*)this; + return this; + } + + /// \name Scalar TTI Implementations + /// @{ + + /// @} + + + /// \name Vector TTI Implementations + /// @{ + + unsigned getNumberOfRegisters(bool Vector) const { + if (Vector) { + if (ST->hasNEON()) + return 32; + return 0; + } + return 32; + } + + unsigned getRegisterBitWidth(bool Vector) const { + if (Vector) { + if (ST->hasNEON()) + return 128; + return 0; + } + return 64; + } + + /// @} +}; + +} // end anonymous namespace + +INITIALIZE_AG_PASS(AArch64TTI, TargetTransformInfo, "aarch64tti", + "AArch64 Target Transform Info", true, true, false) +char AArch64TTI::ID = 0; + +ImmutablePass * +llvm::createAArch64TargetTransformInfoPass(const AArch64TargetMachine *TM) { + return new AArch64TTI(TM); +} diff --git a/lib/Target/AArch64/CMakeLists.txt b/lib/Target/AArch64/CMakeLists.txt index 1eb83f5d03b..dfc10afcdcf 100644 --- a/lib/Target/AArch64/CMakeLists.txt +++ b/lib/Target/AArch64/CMakeLists.txt @@ -26,6 +26,7 @@ add_llvm_target(AArch64CodeGen AArch64Subtarget.cpp AArch64TargetMachine.cpp AArch64TargetObjectFile.cpp + AArch64TargetTransformInfo.cpp ) add_subdirectory(AsmParser)