[llvm] f126643 - [AArch64] add tests for masked subtract; NFC

[Sanjay Patel via llvm-commits]

Author: Sanjay Patel
Date: 2022-06-17T14:56:32-04:00
New Revision: f126643862611c02ec94e86da9b6f7921bc7a490

URL: https://github.com/llvm/llvm-project/commit/f126643862611c02ec94e86da9b6f7921bc7a490
DIFF: https://github.com/llvm/llvm-project/commit/f126643862611c02ec94e86da9b6f7921bc7a490.diff

LOG: [AArch64] add tests for masked subtract; NFC

Added: 
    

Modified: 
    llvm/test/CodeGen/AArch64/sub1.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/CodeGen/AArch64/sub1.ll b/llvm/test/CodeGen/AArch64/sub1.ll
index bb6a116ea12ff..688742bc907d0 100644
--- a/llvm/test/CodeGen/AArch64/sub1.ll
+++ b/llvm/test/CodeGen/AArch64/sub1.ll
@@ -14,3 +14,53 @@ define i64 @sub1_disguised_constant(i64 %x) {
   ret i64 %r
 }
 
+define i8 @masked_sub_i8(i8 %x) {
+; CHECK-LABEL: masked_sub_i8:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    mov w8, #5
+; CHECK-NEXT:    mov w9, #7
+; CHECK-NEXT:    and w8, w0, w8
+; CHECK-NEXT:    sub w0, w9, w8
+; CHECK-NEXT:    ret
+  %a = and i8 %x, 5
+  %m = sub i8 7, %a
+  ret i8 %m
+}
+
+define i8 @not_masked_sub_i8(i8 %x) {
+; CHECK-LABEL: not_masked_sub_i8:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    mov w8, #7
+; CHECK-NEXT:    and w9, w0, #0x8
+; CHECK-NEXT:    sub w0, w8, w9
+; CHECK-NEXT:    ret
+  %a = and i8 %x, 8
+  %m = sub i8 7, %a
+  ret i8 %m
+}
+
+define i32 @masked_sub_i32(i32 %x) {
+; CHECK-LABEL: masked_sub_i32:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    mov w8, #9
+; CHECK-NEXT:    mov w9, #31
+; CHECK-NEXT:    and w8, w0, w8
+; CHECK-NEXT:    sub w0, w9, w8
+; CHECK-NEXT:    ret
+  %a = and i32 %x, 9
+  %m = sub i32 31, %a
+  ret i32 %m
+}
+
+define <4 x i32> @masked_sub_v4i32(<4 x i32> %x) {
+; CHECK-LABEL: masked_sub_v4i32:
+; CHECK:       // %bb.0:
+; CHECK-NEXT:    movi v1.4s, #42
+; CHECK-NEXT:    movi v2.4s, #1, msl #8
+; CHECK-NEXT:    and v0.16b, v0.16b, v1.16b
+; CHECK-NEXT:    sub v0.4s, v2.4s, v0.4s
+; CHECK-NEXT:    ret
+  %a = and <4 x i32> %x, <i32 42, i32 42, i32 42, i32 42>
+  %m = sub <4 x i32> <i32 511, i32 511, i32 511, i32 511>, %a
+  ret <4 x i32>  %m
+}