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