MipsOs16.cpp revision 251607
1258882Semaste//===---- MipsOs16.cpp for Mips Option -Os16 --------===// 2258882Semaste// 3258882Semaste// The LLVM Compiler Infrastructure 4258882Semaste// 5258882Semaste// This file is distributed under the University of Illinois Open Source 6258882Semaste// License. See LICENSE.TXT for details. 7258882Semaste// 8258882Semaste//===----------------------------------------------------------------------===// 9258882Semaste// 10258882Semaste// This file defines an optimization phase for the MIPS target. 11258882Semaste// 12258882Semaste//===----------------------------------------------------------------------===// 13258882Semaste 14258882Semaste#define DEBUG_TYPE "mips-os16" 15258882Semaste#include "MipsOs16.h" 16258882Semaste#include "llvm/IR/Module.h" 17258882Semaste#include "llvm/Support/Debug.h" 18258882Semaste#include "llvm/Support/raw_ostream.h" 19258882Semaste 20258882Semastenamespace { 21258882Semaste 22258882Semaste // Figure out if we need float point based on the function signature. 23258882Semaste // We need to move variables in and/or out of floating point 24258882Semaste // registers because of the ABI 25269024Semaste // 26269024Semaste bool needsFPFromSig(Function &F) { 27269024Semaste Type* RetType = F.getReturnType(); 28269024Semaste switch (RetType->getTypeID()) { 29269024Semaste case Type::FloatTyID: 30269024Semaste case Type::DoubleTyID: 31269024Semaste return true; 32269024Semaste default: 33269024Semaste ; 34269024Semaste } 35258882Semaste if (F.arg_size() >=1) { 36258882Semaste Argument &Arg = F.getArgumentList().front(); 37258882Semaste switch (Arg.getType()->getTypeID()) { 38258882Semaste case Type::FloatTyID: 39258882Semaste case Type::DoubleTyID: 40258882Semaste return true; 41258882Semaste default: 42258882Semaste ; 43258882Semaste } 44258882Semaste } 45258882Semaste return false; 46258882Semaste } 47258882Semaste 48258882Semaste // Figure out if the function will need floating point operations 49258882Semaste // 50258882Semaste bool needsFP(Function &F) { 51258882Semaste if (needsFPFromSig(F)) 52258882Semaste return true; 53258882Semaste for (Function::const_iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 54258882Semaste for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); 55258882Semaste I != E; ++I) { 56258882Semaste const Instruction &Inst = *I; 57258882Semaste switch (Inst.getOpcode()) { 58258882Semaste case Instruction::FAdd: 59258882Semaste case Instruction::FSub: 60258882Semaste case Instruction::FMul: 61258882Semaste case Instruction::FDiv: 62258882Semaste case Instruction::FRem: 63258882Semaste case Instruction::FPToUI: 64258882Semaste case Instruction::FPToSI: 65258882Semaste case Instruction::UIToFP: 66258882Semaste case Instruction::SIToFP: 67258882Semaste case Instruction::FPTrunc: 68258882Semaste case Instruction::FPExt: 69258882Semaste case Instruction::FCmp: 70258882Semaste return true; 71258882Semaste default: 72258882Semaste ; 73258882Semaste } 74258882Semaste if (const CallInst *CI = dyn_cast<CallInst>(I)) { 75258882Semaste DEBUG(dbgs() << "Working on call" << "\n"); 76258882Semaste Function &F_ = *CI->getCalledFunction(); 77258882Semaste if (needsFPFromSig(F_)) 78258882Semaste return true; 79258882Semaste } 80258882Semaste } 81258882Semaste return false; 82258882Semaste } 83258882Semaste} 84258882Semastenamespace llvm { 85258882Semaste 86258882Semaste 87258882Semastebool MipsOs16::runOnModule(Module &M) { 88258882Semaste DEBUG(errs() << "Run on Module MipsOs16\n"); 89258882Semaste bool modified = false; 90258882Semaste for (Module::iterator F = M.begin(), E = M.end(); F != E; ++F) { 91258882Semaste if (F->isDeclaration()) continue; 92258882Semaste DEBUG(dbgs() << "Working on " << F->getName() << "\n"); 93258882Semaste if (needsFP(*F)) { 94258882Semaste DEBUG(dbgs() << " need to compile as nomips16 \n"); 95258882Semaste F->addFnAttr("nomips16"); 96258882Semaste } 97258882Semaste else { 98258882Semaste F->addFnAttr("mips16"); 99258882Semaste DEBUG(dbgs() << " no need to compile as nomips16 \n"); 100258882Semaste } 101258882Semaste } 102258882Semaste return modified; 103258882Semaste} 104258882Semaste 105258882Semastechar MipsOs16::ID = 0; 106258882Semaste 107258882Semaste} 108258882Semaste 109258882SemasteModulePass *llvm::createMipsOs16(MipsTargetMachine &TM) { 110258882Semaste return new MipsOs16; 111258882Semaste} 112258882Semaste 113258882Semaste 114258882Semaste