Followup to 132316; accept arbitrary constants, add with a constant,

sub with a non-constant.  Fix comments, enlarge test case.
rdar://problem/6501862


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@132348 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 39 insertions, 6 deletions

diff --git a/test/Transforms/InstCombine/2011-05-28-swapmulsub.ll b/test/Transforms/InstCombine/2011-05-28-swapmulsub.ll
index e4417ef96a..b096d1f952 100644
--- a/test/Transforms/InstCombine/2011-05-28-swapmulsub.ll
+++ b/test/Transforms/InstCombine/2011-05-28-swapmulsub.ll
@@ -2,18 +2,14 @@
 ; RUN: opt -S -instcombine < %s | FileCheck %s
 target triple = "x86_64-apple-macosx10.6.6"
 
-define zeroext i16 @foo(i32 %on_off, i16* %puls) nounwind uwtable ssp {
+define zeroext i16 @foo1(i32 %on_off) nounwind uwtable ssp {
 entry:
   %on_off.addr = alloca i32, align 4
-  %puls.addr = alloca i16*, align 8
   %a = alloca i32, align 4
   store i32 %on_off, i32* %on_off.addr, align 4
-  store i16* %puls, i16** %puls.addr, align 8
   %tmp = load i32* %on_off.addr, align 4
-; CHECK-NOT: sub
-; CHECK-NOT: mul
-; (1 - %tmp) * (-2) -> (%tmp - 1) * 2
   %sub = sub i32 1, %tmp
+; CHECK-NOT: mul i32
   %mul = mul i32 %sub, -2
 ; CHECK: shl
 ; CHECK-NEXT: add
@@ -22,3 +18,40 @@ entry:
   %conv = trunc i32 %tmp1 to i16
   ret i16 %conv
 }
+
+define zeroext i16 @foo2(i32 %on_off, i32 %q) nounwind uwtable ssp {
+entry:
+  %on_off.addr = alloca i32, align 4
+  %q.addr = alloca i32, align 4
+  %a = alloca i32, align 4
+  store i32 %on_off, i32* %on_off.addr, align 4
+  store i32 %q, i32* %q.addr, align 4
+  %tmp = load i32* %q.addr, align 4
+  %tmp1 = load i32* %on_off.addr, align 4
+  %sub = sub i32 %tmp, %tmp1
+; CHECK-NOT: mul i32
+  %mul = mul i32 %sub, -4
+; CHECK: sub i32
+; CHECK-NEXT: shl
+  store i32 %mul, i32* %a, align 4
+  %tmp2 = load i32* %a, align 4
+  %conv = trunc i32 %tmp2 to i16
+  ret i16 %conv
+}
+
+define zeroext i16 @foo3(i32 %on_off) nounwind uwtable ssp {
+entry:
+  %on_off.addr = alloca i32, align 4
+  %a = alloca i32, align 4
+  store i32 %on_off, i32* %on_off.addr, align 4
+  %tmp = load i32* %on_off.addr, align 4
+  %sub = sub i32 7, %tmp
+; CHECK-NOT: mul i32
+  %mul = mul i32 %sub, -4
+; CHECK: shl
+; CHECK-NEXT: add
+  store i32 %mul, i32* %a, align 4
+  %tmp1 = load i32* %a, align 4
+  %conv = trunc i32 %tmp1 to i16
+  ret i16 %conv
+}