From 0cf5d40d4c77dd2e21429a099edb4319bf240cc9 Mon Sep 17 00:00:00 2001
From: Sanjay Patel <spatel@rotateright.com>
Date: Fri, 8 Jul 2022 08:35:36 -0400
Subject: [PATCH] [InstCombine] add tests for masked sub; NFC

---
 llvm/test/Transforms/InstCombine/sub-xor.ll | 35 ++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/llvm/test/Transforms/InstCombine/sub-xor.ll b/llvm/test/Transforms/InstCombine/sub-xor.ll
index 8159d3f..388fa92 100644
--- a/llvm/test/Transforms/InstCombine/sub-xor.ll
+++ b/llvm/test/Transforms/InstCombine/sub-xor.ll
@@ -25,11 +25,44 @@ define <2 x i32> @test1vec(<2 x i32> %x) {
   ret <2 x i32> %sub
 }
 
+define i8 @masked_sub_i8(i8 %x) {
+; CHECK-LABEL: @masked_sub_i8(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], 10
+; CHECK-NEXT:    [[M:%.*]] = sub nuw nsw i8 11, [[A]]
+; CHECK-NEXT:    ret i8 [[M]]
+;
+  %a = and i8 %x, 10 ; 0b00001010
+  %m = sub i8 11, %a ; 0b00001011
+  ret i8 %m
+}
+
+define <2 x i5> @masked_sub_v2i5(<2 x i5> %x) {
+; CHECK-LABEL: @masked_sub_v2i5(
+; CHECK-NEXT:    [[A:%.*]] = and <2 x i5> [[X:%.*]], <i5 -8, i5 -8>
+; CHECK-NEXT:    [[M:%.*]] = sub nuw nsw <2 x i5> <i5 -6, i5 -6>, [[A]]
+; CHECK-NEXT:    ret <2 x i5> [[M]]
+;
+  %a = and <2 x i5> %x, <i5 24, i5 24> ; 0b11000
+  %m = sub <2 x i5> <i5 26, i5 26>, %a ; 0b11010
+  ret <2 x i5> %m
+}
+
+define i8 @not_masked_sub_i8(i8 %x) {
+; CHECK-LABEL: @not_masked_sub_i8(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], 7
+; CHECK-NEXT:    [[M:%.*]] = sub nuw nsw i8 11, [[A]]
+; CHECK-NEXT:    ret i8 [[M]]
+;
+  %a = and i8 %x, 7  ; 0b00000111
+  %m = sub i8 11, %a ; 0b00001011
+  ret i8 %m
+}
+
 declare i32 @llvm.ctlz.i32(i32, i1) nounwind readnone
 
 define i32 @test2(i32 %x) nounwind {
 ; CHECK-LABEL: @test2(
-; CHECK-NEXT:    [[COUNT:%.*]] = tail call i32 @llvm.ctlz.i32(i32 [[X:%.*]], i1 true) [[ATTR2:#.*]], [[RNG0:!range !.*]]
+; CHECK-NEXT:    [[COUNT:%.*]] = tail call i32 @llvm.ctlz.i32(i32 [[X:%.*]], i1 true) #[[ATTR2:[0-9]+]], !range [[RNG0:![0-9]+]]
 ; CHECK-NEXT:    [[SUB:%.*]] = xor i32 [[COUNT]], 31
 ; CHECK-NEXT:    ret i32 [[SUB]]
 ;
-- 
2.7.4