diff options
author | Owen Anderson <resistor@mac.com> | 2008-02-18 09:24:53 +0000 |
---|---|---|
committer | Owen Anderson <resistor@mac.com> | 2008-02-18 09:24:53 +0000 |
commit | 5aa4f2a0857563b4ad9115c614afed9501aa58f4 (patch) | |
tree | d1def04f316ef138fc6afe4cb5a91349cfe71054 /lib/Transforms/Scalar | |
parent | 874a892c9969fbbe50a5017868c2f82923632b29 (diff) | |
download | external_llvm-5aa4f2a0857563b4ad9115c614afed9501aa58f4.zip external_llvm-5aa4f2a0857563b4ad9115c614afed9501aa58f4.tar.gz external_llvm-5aa4f2a0857563b4ad9115c614afed9501aa58f4.tar.bz2 |
Add support to GVN for performing sret return slot optimization. This means that, if an sret function tail calls
another sret function, it should pass its own sret parameter to the tail callee, allowing it to fill in the correct
return value. llvm-gcc does not emit this by default. Instead, it allocates space in the caller for the sret of
the tail call and then uses memcpy to copy the result into the caller's sret parameter. This optimization detects
and optimizes that case.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@47265 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/Transforms/Scalar')
-rw-r--r-- | lib/Transforms/Scalar/GVN.cpp | 68 |
1 files changed, 66 insertions, 2 deletions
diff --git a/lib/Transforms/Scalar/GVN.cpp b/lib/Transforms/Scalar/GVN.cpp index 41a23ed..a4f78fe 100644 --- a/lib/Transforms/Scalar/GVN.cpp +++ b/lib/Transforms/Scalar/GVN.cpp @@ -21,6 +21,7 @@ #include "llvm/Function.h" #include "llvm/IntrinsicInst.h" #include "llvm/Instructions.h" +#include "llvm/ParameterAttributes.h" #include "llvm/Value.h" #include "llvm/ADT/BitVector.h" #include "llvm/ADT/DenseMap.h" @@ -738,6 +739,8 @@ namespace { bool processNonLocalLoad(LoadInst* L, SmallVector<Instruction*, 4>& toErase); bool processMemCpy(MemCpyInst* M, SmallVector<Instruction*, 4>& toErase); + bool performReturnSlotOptzn(MemCpyInst* cpy, CallInst* C, + SmallVector<Instruction*, 4>& toErase); Value *GetValueForBlock(BasicBlock *BB, LoadInst* orig, DenseMap<BasicBlock*, Value*> &Phis, bool top_level = false); @@ -1048,6 +1051,62 @@ bool GVN::processLoad(LoadInst* L, return deletedLoad; } +/// performReturnSlotOptzn - takes a memcpy and a call that it depends on, +/// and checks for the possibility of a return slot optimization by having +/// the call write its result directly into the callees return parameter +/// rather than using memcpy +bool GVN::performReturnSlotOptzn(MemCpyInst* cpy, CallInst* C, + SmallVector<Instruction*, 4>& toErase) { + // Check that we're copying to an argument... + Value* cpyDest = cpy->getDest(); + if (!isa<Argument>(cpyDest)) + return false; + + // And that the argument is the return slot + Argument* sretArg = cast<Argument>(cpyDest); + if (!sretArg->hasStructRetAttr()) + return false; + + // Make sure the return slot is otherwise dead + std::set<User*> useList(sretArg->use_begin(), sretArg->use_end()); + while (!useList.empty()) { + User* UI = *useList.begin(); + + if (isa<GetElementPtrInst>(UI) || isa<BitCastInst>(UI)) { + useList.insert(UI->use_begin(), UI->use_end()); + useList.erase(UI); + } else if (UI == cpy) + useList.erase(UI); + else + return false; + } + + // Make sure the call cannot modify the return slot in some unpredicted way + AliasAnalysis& AA = getAnalysis<AliasAnalysis>(); + if (AA.getModRefInfo(C, cpy->getRawDest(), ~0UL) != AliasAnalysis::NoModRef) + return false; + + // If all checks passed, then we can perform the transformation + CallSite CS = CallSite::get(C); + for (unsigned i = 0; i < CS.arg_size(); ++i) { + if (CS.paramHasAttr(i+1, ParamAttr::StructRet)) { + if (CS.getArgument(i)->getType() != cpyDest->getType()) + return false; + + CS.setArgument(i, cpyDest); + break; + } + } + + MemoryDependenceAnalysis& MD = getAnalysis<MemoryDependenceAnalysis>(); + MD.dropInstruction(C); + + // Remove the memcpy + toErase.push_back(cpy); + + return true; +} + /// processMemCpy - perform simplication of memcpy's. If we have memcpy A which /// copies X to Y, and memcpy B which copies Y to Z, then we can rewrite B to be /// a memcpy from X to Z (or potentially a memmove, depending on circumstances). @@ -1059,9 +1118,14 @@ bool GVN::processMemCpy(MemCpyInst* M, // First, we have to check that the dependency is another memcpy Instruction* dep = MD.getDependency(M); if (dep == MemoryDependenceAnalysis::None || - dep == MemoryDependenceAnalysis::NonLocal || - !isa<MemCpyInst>(dep)) + dep == MemoryDependenceAnalysis::NonLocal) return false; + else if (!isa<MemCpyInst>(dep)) { + if (CallInst* C = dyn_cast<CallInst>(dep)) + return performReturnSlotOptzn(M, C, toErase); + else + return false; + } // We can only transforms memcpy's where the dest of one is the source of the // other |