lib/Target/Hexagon/HexagonRemoveSZExtArgs.cpp

   1 //===- HexagonRemoveExtendArgs.cpp - Remove unnecessary argument sign extends //
   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 // Pass that removes sign extends for function parameters. These parameters
  11 // are already sign extended by the caller per Hexagon's ABI
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "Hexagon.h"
  16 #include "HexagonTargetMachine.h"
  17 #include "llvm/CodeGen/MachineFunctionAnalysis.h"
  18 #include "llvm/CodeGen/StackProtector.h"
  19 #include "llvm/IR/Function.h"
  20 #include "llvm/IR/Instructions.h"
  21 #include "llvm/Pass.h"
  22 #include "llvm/Transforms/Scalar.h"
  23
  24 using namespace llvm;
  25
  26 namespace llvm {
  27   FunctionPass *createHexagonRemoveExtendArgs(const HexagonTargetMachine &TM);
  28   void initializeHexagonRemoveExtendArgsPass(PassRegistry&);
  29 }
  30
  31 namespace {
  32   struct HexagonRemoveExtendArgs : public FunctionPass {
  33   public:
  34     static char ID;
  35     HexagonRemoveExtendArgs() : FunctionPass(ID) {
  36       initializeHexagonRemoveExtendArgsPass(*PassRegistry::getPassRegistry());
  37     }
  38     bool runOnFunction(Function &F) override;
  39
  40     const char *getPassName() const override {
  41       return "Remove sign extends";
  42     }
  43
  44     void getAnalysisUsage(AnalysisUsage &AU) const override {
  45       AU.addRequired<MachineFunctionAnalysis>();
  46       AU.addPreserved<MachineFunctionAnalysis>();
  47       AU.addPreserved<StackProtector>();
  48       FunctionPass::getAnalysisUsage(AU);
  49     }
  50   };
  51 }
  52
  53 char HexagonRemoveExtendArgs::ID = 0;
  54
  55 INITIALIZE_PASS(HexagonRemoveExtendArgs, "reargs",
  56                 "Remove Sign and Zero Extends for Args", false, false)
  57
  58 bool HexagonRemoveExtendArgs::runOnFunction(Function &F) {
  59   unsigned Idx = 1;
  60   for (Function::arg_iterator AI = F.arg_begin(), AE = F.arg_end(); AI != AE;
  61        ++AI, ++Idx) {
  62     if (F.getAttributes().hasAttribute(Idx, Attribute::SExt)) {
  63       Argument* Arg = AI;
  64       if (!isa<PointerType>(Arg->getType())) {
  65         for (auto UI = Arg->user_begin(); UI != Arg->user_end();) {
  66           if (isa<SExtInst>(*UI)) {
  67             Instruction* I = cast<Instruction>(*UI);
  68             SExtInst* SI = new SExtInst(Arg, I->getType());
  69             assert (EVT::getEVT(SI->getType()) ==
  70                     (EVT::getEVT(I->getType())));
  71             ++UI;
  72             I->replaceAllUsesWith(SI);
  73             Instruction* First = F.getEntryBlock().begin();
  74             SI->insertBefore(First);
  75             I->eraseFromParent();
  76           } else {
  77             ++UI;
  78           }
  79         }
  80       }
  81     }
  82   }
  83   return true;
  84 }
  85
  86
  87
  88 FunctionPass*
  89 llvm::createHexagonRemoveExtendArgs(const HexagonTargetMachine &TM) {
  90   return new HexagonRemoveExtendArgs();
  91 }