DeadStoreElimination can treat byval parameters as if there were alloca's for the purpose of removing end-of-function stores.

llvm-svn: 46351

2 files changed, 16 insertions, 5 deletions

diff --git a/llvm/lib/Transforms/Scalar/DeadStoreElimination.cpp b/llvm/lib/Transforms/Scalar/DeadStoreElimination.cpp
index 6aa1c63f929..f4b432f88e5 100644
--- a/llvm/lib/Transforms/Scalar/DeadStoreElimination.cpp
+++ b/llvm/lib/Transforms/Scalar/DeadStoreElimination.cpp
@@ -261,9 +261,6 @@ bool DSE::handleEndBlock(BasicBlock& BB,
   for (BasicBlock::iterator BBI = BB.end(); BBI != BB.begin(); ){
     --BBI;
     
-    if (deadPointers.empty())
-      break;
-    
     // If we find a store whose pointer is dead...
     if (StoreInst* S = dyn_cast<StoreInst>(BBI)) {
       if (!S->isVolatile()) {
@@ -271,8 +268,12 @@ bool DSE::handleEndBlock(BasicBlock& BB,
         // See through pointer-to-pointer bitcasts
         TranslatePointerBitCasts(pointerOperand);
       
-        if (isa<AllocaInst>(pointerOperand) && 
-            deadPointers.count(cast<AllocaInst>(pointerOperand))) {
+        // Alloca'd pointers or byval arguments (which are functionally like
+        // alloca's) are valid candidates for removal.
+        if ( (isa<AllocaInst>(pointerOperand) && 
+              deadPointers.count(cast<AllocaInst>(pointerOperand))) ||
+             (isa<Argument>(pointerOperand) &&
+              cast<Argument>(pointerOperand)->hasByValAttr())) {
           // Remove it!
           MD.removeInstruction(S);
         
diff --git a/llvm/test/Transforms/DeadStoreElimination/byval.ll b/llvm/test/Transforms/DeadStoreElimination/byval.ll
new file mode 100644
index 00000000000..08f69a40c57
--- /dev/null
+++ b/llvm/test/Transforms/DeadStoreElimination/byval.ll
@@ -0,0 +1,10 @@
+; RUN: llvm-as < %s | opt -dse | llvm-dis | not grep store
+
+%struct.x = type { i32, i32, i32, i32 }
+
+define i32 @foo(%struct.x* byval  %a) nounwind  {
+entry:
+	%tmp2 = getelementptr %struct.x* %a, i32 0, i32 0
+	store i32 1, i32* %tmp2, align 4
+	ret i32 1
+}