[AArch64] Precommit test for D137069; NFC

1 files changed, 89 insertions, 7 deletions

diff --git a/llvm/test/CodeGen/AArch64/addsub.ll b/llvm/test/CodeGen/AArch64/addsub.ll
index 33302f0..ba7edaf 100644
--- a/llvm/test/CodeGen/AArch64/addsub.ll
+++ b/llvm/test/CodeGen/AArch64/addsub.ll
@@ -694,7 +694,7 @@ if.end:                                           ; preds = %if.then, %lor.lhs.f
   ret i32 undef
 }
 
-; ((X >> C) - Y) + Z --> (Z - Y) + (X >> C)
+; ((X << C) - Y) + Z --> (Z - Y) + (X << C)
 define i32 @commute_subop0(i32 %x, i32 %y, i32 %z) {
 ; CHECK-LABEL: commute_subop0:
 ; CHECK:       // %bb.0:
@@ -707,7 +707,7 @@ define i32 @commute_subop0(i32 %x, i32 %y, i32 %z) {
   ret i32 %add
 }
 
-; ((X << C) - Y) + Z --> (Z - Y) + (X << C)
+; ((X >> C) - Y) + Z --> (Z - Y) + (X >> C)
 define i32 @commute_subop0_lshr(i32 %x, i32 %y, i32 %z) {
 ; CHECK-LABEL: commute_subop0_lshr:
 ; CHECK:       // %bb.0:
@@ -721,7 +721,7 @@ define i32 @commute_subop0_lshr(i32 %x, i32 %y, i32 %z) {
   ret i32 %add
 }
 
-; ((X << C) - Y) + Z --> (Z - Y) + (X << C)
+; ((X >> C) - Y) + Z --> (Z - Y) + (X >> C)
 define i32 @commute_subop0_ashr(i32 %x, i32 %y, i32 %z) {
 ; CHECK-LABEL: commute_subop0_ashr:
 ; CHECK:       // %bb.0:
@@ -735,7 +735,89 @@ define i32 @commute_subop0_ashr(i32 %x, i32 %y, i32 %z) {
   ret i32 %add
 }
 
-; Z + ((X >> C) - Y) --> (Z - Y) + (X >> C)
+; ((sext X) - Y) + Z --> (Z - Y) + (sext X)
+define i64 @commute_subop0_sext(i32 %x, i64 %y, i64 %z) {
+; CHECK-LABEL: commute_subop0_sext:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    // kill: def $w0 killed $w0 def $x0
+; CHECK-NEXT:    sxtw x8, w0
+; CHECK-NEXT:    sub x8, x8, x1
+; CHECK-NEXT:    add x0, x8, x2
+; CHECK-NEXT:    ret
+  %sext = sext i32 %x to i64
+  %sub = sub i64 %sext, %y
+  %add = add i64 %sub, %z
+  ret i64 %add
+}
+
+; ((sext_inreg X) - Y) + Z --> (Z - Y) + (sext_inreg X)
+define i64 @commute_subop0_sext_inreg(i64 %x, i64 %y, i64 %z) {
+; CHECK-LABEL: commute_subop0_sext_inreg:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    sxth x8, w0
+; CHECK-NEXT:    sub x8, x8, x1
+; CHECK-NEXT:    add x0, x8, x2
+; CHECK-NEXT:    ret
+  %shl = shl i64 %x, 48
+  %ashr = ashr i64 %shl, 48
+  %sub = sub i64 %ashr, %y
+  %add = add i64 %sub, %z
+  ret i64 %add
+}
+
+; ((zext X) - Y) + Z --> (Z - Y) + (zext X)
+define i32 @commute_subop0_zext(i16 %x, i32 %y, i32 %z) {
+; CHECK-LABEL: commute_subop0_zext:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    and w8, w0, #0xffff
+; CHECK-NEXT:    sub w8, w8, w1
+; CHECK-NEXT:    add w0, w8, w2
+; CHECK-NEXT:    ret
+  %zext = zext i16 %x to i32
+  %sub = sub i32 %zext, %y
+  %add = add i32 %sub, %z
+  ret i32 %add
+}
+
+
+; ((anyext X) - Y) + Z --> (Z - Y) + (anyext X)
+define i8 @commute_subop0_anyext(i16 %a, i16 %b, i32 %c) {
+; CHECK-LABEL: commute_subop0_anyext:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    mov w8, #222
+; CHECK-NEXT:    mov w9, #111
+; CHECK-NEXT:    mul w8, w1, w8
+; CHECK-NEXT:    neg w8, w8
+; CHECK-NEXT:    madd w8, w0, w9, w8
+; CHECK-NEXT:    add w8, w8, w2
+; CHECK-NEXT:    add w0, w8, w8, lsl #5
+; CHECK-NEXT:    ret
+  %aa = mul i16 %a, 111
+  %bb = mul i16 %b, 222
+  %a_32 = zext i16 %aa to i32
+  %b_32 = zext i16 %bb to i32
+  %sub = sub i32 %a_32, %b_32
+  %add = add i32 %sub, %c
+  %trunc = trunc i32 %add to i8
+  %r = mul i8 %trunc, 33
+  ret i8 %r
+}
+
+; ((X and C) - Y) + Z --> (Z - Y) + (X and C)
+define i32 @commute_subop0_and(i32 %x, i32 %y, i32 %z) {
+; CHECK-LABEL: commute_subop0_and:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    and w8, w0, #0xff
+; CHECK-NEXT:    sub w8, w8, w1
+; CHECK-NEXT:    add w0, w8, w2
+; CHECK-NEXT:    ret
+  %and = and i32 %x, 255
+  %sub = sub i32 %and, %y
+  %add = add i32 %sub, %z
+  ret i32 %add
+}
+
+; Z + ((X << C) - Y) --> (Z - Y) + (X << C)
 define i32 @commute_subop0_cadd(i32 %x, i32 %y, i32 %z) {
 ; CHECK-LABEL: commute_subop0_cadd:
 ; CHECK:       // %bb.0:
@@ -748,7 +830,7 @@ define i32 @commute_subop0_cadd(i32 %x, i32 %y, i32 %z) {
   ret i32 %add
 }
 
-; Y + ((X >> C) - X) --> (Y - X) + (X >> C)
+; Y + ((X << C) - X) --> (Y - X) + (X << C)
 define i32 @commute_subop0_mul(i32 %x, i32 %y) {
 ; CHECK-LABEL: commute_subop0_mul:
 ; CHECK:       // %bb.0:
@@ -760,7 +842,7 @@ define i32 @commute_subop0_mul(i32 %x, i32 %y) {
   ret i32 %add
 }
 
-; negative case for ((X >> C) - Y) + Z --> (Z - Y) + (X >> C)
+; negative case for ((X << C) - Y) + Z --> (Z - Y) + (X << C)
 ; Y can't be constant to avoid dead loop
 define i32 @commute_subop0_zconst(i32 %x, i32 %y) {
 ; CHECK-LABEL: commute_subop0_zconst:
@@ -775,7 +857,7 @@ define i32 @commute_subop0_zconst(i32 %x, i32 %y) {
   ret i32 %add
 }
 
-; negative case for ((X >> C) - Y) + Z --> (Z - Y) + (X >> C)
+; negative case for ((X << C) - Y) + Z --> (Z - Y) + (X << C)
 ; Y can't be shift C also to avoid dead loop
 define i32 @commute_subop0_zshiftc_oneuse(i32 %x, i32 %y, i32 %z) {
 ; CHECK-LABEL: commute_subop0_zshiftc_oneuse: