[PATCH] [SLSR] garbage-collect unused instructions

[Jingyue Wu]

REPOSITORY
  rL LLVM

http://reviews.llvm.org/D9101

Files:
  llvm/trunk/lib/Transforms/Scalar/StraightLineStrengthReduce.cpp
  llvm/trunk/test/Transforms/StraightLineStrengthReduce/X86/no-slsr.ll
  llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-add.ll
  llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-gep.ll
  llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-mul.ll

Index: llvm/trunk/lib/Transforms/Scalar/StraightLineStrengthReduce.cpp
===================================================================
--- llvm/trunk/lib/Transforms/Scalar/StraightLineStrengthReduce.cpp
+++ llvm/trunk/lib/Transforms/Scalar/StraightLineStrengthReduce.cpp
@@ -68,6 +68,7 @@
 #include "llvm/IR/PatternMatch.h"
 #include "llvm/Support/raw_ostream.h"
 #include "llvm/Transforms/Scalar.h"
+#include "llvm/Transforms/Utils/Local.h"
 
 using namespace llvm;
 using namespace PatternMatch;
@@ -599,9 +600,14 @@
   switch (C.CandidateKind) {
   case Candidate::Add:
   case Candidate::Mul:
+    // C = Basis + Bump
     if (BinaryOperator::isNeg(Bump)) {
+      // If Bump is a neg instruction, emit C = Basis - (-Bump).
       Reduced =
           Builder.CreateSub(Basis.Ins, BinaryOperator::getNegArgument(Bump));
+      // We only use the negative argument of Bump, and Bump itself may be
+      // trivially dead.
+      RecursivelyDeleteTriviallyDeadInstructions(Bump);
     } else {
       Reduced = Builder.CreateAdd(Basis.Ins, Bump);
     }
@@ -637,7 +643,6 @@
   };
   Reduced->takeName(C.Ins);
   C.Ins->replaceAllUsesWith(Reduced);
-  C.Ins->dropAllReferences();
   // Unlink C.Ins so that we can skip other candidates also corresponding to
   // C.Ins. The actual deletion is postponed to the end of runOnFunction.
   C.Ins->removeFromParent();
@@ -670,8 +675,13 @@
   }
 
   // Delete all unlink instructions.
-  for (auto I : UnlinkedInstructions) {
-    delete I;
+  for (auto *UnlinkedInst : UnlinkedInstructions) {
+    for (unsigned I = 0, E = UnlinkedInst->getNumOperands(); I != E; ++I) {
+      Value *Op = UnlinkedInst->getOperand(I);
+      UnlinkedInst->setOperand(I, nullptr);
+      RecursivelyDeleteTriviallyDeadInstructions(Op);
+    }
+    delete UnlinkedInst;
   }
   bool Ret = !UnlinkedInstructions.empty();
   UnlinkedInstructions.clear();
Index: llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-mul.ll
===================================================================
--- llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-mul.ll
+++ llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-mul.ll
@@ -1,4 +1,4 @@
-; RUN: opt < %s -slsr -gvn -dce -S | FileCheck %s
+; RUN: opt < %s -slsr -gvn -S | FileCheck %s
 
 target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64"
 
Index: llvm/trunk/test/Transforms/StraightLineStrengthReduce/X86/no-slsr.ll
===================================================================
--- llvm/trunk/test/Transforms/StraightLineStrengthReduce/X86/no-slsr.ll
+++ llvm/trunk/test/Transforms/StraightLineStrengthReduce/X86/no-slsr.ll
@@ -1,4 +1,4 @@
-; RUN: opt < %s -slsr -gvn -dce -S | FileCheck %s
+; RUN: opt < %s -slsr -gvn -S | FileCheck %s
 
 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
 target triple = "x86_64-unknown-linux-gnu"
Index: llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-add.ll
===================================================================
--- llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-add.ll
+++ llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-add.ll
@@ -1,4 +1,4 @@
-; RUN: opt < %s -slsr -gvn -dce -S | FileCheck %s
+; RUN: opt < %s -slsr -gvn -S | FileCheck %s
 
 target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64"
 
Index: llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-gep.ll
===================================================================
--- llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-gep.ll
+++ llvm/trunk/test/Transforms/StraightLineStrengthReduce/slsr-gep.ll
@@ -1,4 +1,4 @@
-; RUN: opt < %s -slsr -gvn -dce -S | FileCheck %s
+; RUN: opt < %s -slsr -gvn -S | FileCheck %s
 
 target datalayout = "e-i64:64-v16:16-v32:32-n16:32:64"

EMAIL PREFERENCES
  http://reviews.llvm.org/settings/panel/emailpreferences/
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D9101.24162.patch
Type: text/x-patch
Size: 3757 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20150421/18fcbe55/attachment.bin>