[InstCombine] add tests for sdiv with (neg)pow2 divisor; NFC

diff --git a/llvm/test/Transforms/InstCombine/sdiv-exact-by-negative-power-of-two.ll b/llvm/test/Transforms/InstCombine/sdiv-exact-by-negative-power-of-two.ll
index e53fd37..de74b31 100644 (file)
--- a/llvm/test/Transforms/InstCombine/sdiv-exact-by-negative-power-of-two.ll
+++ b/llvm/test/Transforms/InstCombine/sdiv-exact-by-negative-power-of-two.ll
@@ -61,3 +61,58 @@ define <2 x i8> @n4_vec_undef(<2 x i8> %x) {
   %div = sdiv exact <2 x i8> %x, <i8 -32, i8 undef>
   ret <2 x i8> %div
 }
+
+define i8 @prove_exact_with_high_mask(i8 %x, i8 %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -32
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], -4
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -32
+  %d = sdiv i8 %a, -4
+  ret i8 %d
+}
+
+define i8 @prove_exact_with_high_mask_limit(i8 %x, i8 %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_limit(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -32
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], -32
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -32
+  %d = sdiv i8 %a, -32
+  ret i8 %d
+}
+
+define i8 @not_prove_exact_with_high_mask(i8 %x, i8 %y) {
+; CHECK-LABEL: @not_prove_exact_with_high_mask(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -32
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], -64
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -32
+  %d = sdiv i8 %a, -64
+  ret i8 %d
+}
+
+define <2 x i8> @prove_exact_with_high_mask_splat_vec(<2 x i8> %x, <2 x i8> %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_splat_vec(
+; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], <i8 5, i8 5>
+; CHECK-NEXT:    [[D:%.*]] = sdiv <2 x i8> [[A]], <i8 -16, i8 -16>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %a = shl <2 x i8> %x, <i8 5, i8 5>
+  %d = sdiv <2 x i8> %a, <i8 -16, i8 -16>
+  ret <2 x i8> %d
+}
+
+define <2 x i8> @prove_exact_with_high_mask_vec(<2 x i8> %x, <2 x i8> %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_vec(
+; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], <i8 3, i8 2>
+; CHECK-NEXT:    [[D:%.*]] = sdiv <2 x i8> [[A]], <i8 -8, i8 -4>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %a = shl <2 x i8> %x, <i8 3, i8 2>
+  %d = sdiv <2 x i8> %a, <i8 -8, i8 -4>
+  ret <2 x i8> %d
+}

diff --git a/llvm/test/Transforms/InstCombine/sdiv-exact-by-power-of-two.ll b/llvm/test/Transforms/InstCombine/sdiv-exact-by-power-of-two.ll
index 997910d..0cdc52b 100644 (file)
--- a/llvm/test/Transforms/InstCombine/sdiv-exact-by-power-of-two.ll
+++ b/llvm/test/Transforms/InstCombine/sdiv-exact-by-power-of-two.ll
@@ -107,3 +107,58 @@ define i8 @shl1_nsw_not_exact(i8 %x, i8 %y) {
   %div = sdiv i8 %x, %shl
   ret i8 %div
 }
+
+define i8 @prove_exact_with_high_mask(i8 %x, i8 %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -8
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], 4
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -8
+  %d = sdiv i8 %a, 4
+  ret i8 %d
+}
+
+define i8 @prove_exact_with_high_mask_limit(i8 %x, i8 %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_limit(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -8
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], 8
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -8
+  %d = sdiv i8 %a, 8
+  ret i8 %d
+}
+
+define i8 @not_prove_exact_with_high_mask(i8 %x, i8 %y) {
+; CHECK-LABEL: @not_prove_exact_with_high_mask(
+; CHECK-NEXT:    [[A:%.*]] = and i8 [[X:%.*]], -8
+; CHECK-NEXT:    [[D:%.*]] = sdiv i8 [[A]], 16
+; CHECK-NEXT:    ret i8 [[D]]
+;
+  %a = and i8 %x, -8
+  %d = sdiv i8 %a, 16
+  ret i8 %d
+}
+
+define <2 x i8> @prove_exact_with_high_mask_splat_vec(<2 x i8> %x, <2 x i8> %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_splat_vec(
+; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], <i8 3, i8 3>
+; CHECK-NEXT:    [[D:%.*]] = sdiv <2 x i8> [[A]], <i8 8, i8 8>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %a = shl <2 x i8> %x, <i8 3, i8 3>
+  %d = sdiv <2 x i8> %a, <i8 8, i8 8>
+  ret <2 x i8> %d
+}
+
+define <2 x i8> @prove_exact_with_high_mask_vec(<2 x i8> %x, <2 x i8> %y) {
+; CHECK-LABEL: @prove_exact_with_high_mask_vec(
+; CHECK-NEXT:    [[A:%.*]] = shl <2 x i8> [[X:%.*]], <i8 3, i8 2>
+; CHECK-NEXT:    [[D:%.*]] = sdiv <2 x i8> [[A]], <i8 8, i8 4>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %a = shl <2 x i8> %x, <i8 3, i8 2>
+  %d = sdiv <2 x i8> %a, <i8 8, i8 4>
+  ret <2 x i8> %d
+}