[llvm] d047bb1 - [InstCombine] Add trunc(shr(trunc(x), c)) non-uniform vector tests

[Simon Pilgrim via llvm-commits]

Author: Simon Pilgrim
Date: 2020-09-28T18:53:38+01:00
New Revision: d047bb1cf69316b39dd67ecc5af669414be9152c

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

LOG: [InstCombine] Add trunc(shr(trunc(x),c)) non-uniform vector tests

Added: 
    

Modified: 
    llvm/test/Transforms/InstCombine/trunc-shift-trunc.ll

Removed: 
    


################################################################################
diff  --git a/llvm/test/Transforms/InstCombine/trunc-shift-trunc.ll b/llvm/test/Transforms/InstCombine/trunc-shift-trunc.ll
index 0c4005fc849e..0838e82f32c7 100644
--- a/llvm/test/Transforms/InstCombine/trunc-shift-trunc.ll
+++ b/llvm/test/Transforms/InstCombine/trunc-shift-trunc.ll
@@ -33,6 +33,19 @@ define <2 x i8> @trunc_lshr_trunc_uniform(<2 x i64> %a) {
   ret <2 x i8> %d
 }
 
+define <2 x i8> @trunc_lshr_trunc_nonuniform(<2 x i64> %a) {
+; CHECK-LABEL: @trunc_lshr_trunc_nonuniform(
+; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
+; CHECK-NEXT:    [[C:%.*]] = lshr <2 x i32> [[B]], <i32 8, i32 2>
+; CHECK-NEXT:    [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %b = trunc <2 x i64> %a to <2 x i32>
+  %c = lshr <2 x i32> %b, <i32 8, i32 2>
+  %d = trunc <2 x i32> %c to <2 x i8>
+  ret <2 x i8> %d
+}
+
 define <2 x i8> @trunc_lshr_trunc_uniform_undef(<2 x i64> %a) {
 ; CHECK-LABEL: @trunc_lshr_trunc_uniform_undef(
 ; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
@@ -59,6 +72,19 @@ define i8 @trunc_lshr_trunc_outofrange(i64 %a) {
   ret i8 %d
 }
 
+define <2 x i8> @trunc_lshr_trunc_nonuniform_outofrange(<2 x i64> %a) {
+; CHECK-LABEL: @trunc_lshr_trunc_nonuniform_outofrange(
+; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
+; CHECK-NEXT:    [[C:%.*]] = lshr <2 x i32> [[B]], <i32 8, i32 11>
+; CHECK-NEXT:    [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %b = trunc <2 x i64> %a to <2 x i32>
+  %c = lshr <2 x i32> %b, <i32 8, i32 11>
+  %d = trunc <2 x i32> %c to <2 x i8>
+  ret <2 x i8> %d
+}
+
 define i8 @trunc_ashr_trunc(i64 %a) {
 ; CHECK-LABEL: @trunc_ashr_trunc(
 ; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32
@@ -85,6 +111,19 @@ define <2 x i8> @trunc_ashr_trunc_uniform(<2 x i64> %a) {
   ret <2 x i8> %d
 }
 
+define <2 x i8> @trunc_ashr_trunc_nonuniform(<2 x i64> %a) {
+; CHECK-LABEL: @trunc_ashr_trunc_nonuniform(
+; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
+; CHECK-NEXT:    [[C:%.*]] = ashr <2 x i32> [[B]], <i32 0, i32 2>
+; CHECK-NEXT:    [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %b = trunc <2 x i64> %a to <2 x i32>
+  %c = ashr <2 x i32> %b, <i32 0, i32 2>
+  %d = trunc <2 x i32> %c to <2 x i8>
+  ret <2 x i8> %d
+}
+
 define <2 x i8> @trunc_ashr_trunc_uniform_undef(<2 x i64> %a) {
 ; CHECK-LABEL: @trunc_ashr_trunc_uniform_undef(
 ; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
@@ -111,6 +150,19 @@ define i8 @trunc_ashr_trunc_outofrange(i64 %a) {
   ret i8 %d
 }
 
+define <2 x i8> @trunc_ashr_trunc_nonuniform_outofrange(<2 x i64> %a) {
+; CHECK-LABEL: @trunc_ashr_trunc_nonuniform_outofrange(
+; CHECK-NEXT:    [[B:%.*]] = trunc <2 x i64> [[A:%.*]] to <2 x i32>
+; CHECK-NEXT:    [[C:%.*]] = ashr <2 x i32> [[B]], <i32 8, i32 11>
+; CHECK-NEXT:    [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; CHECK-NEXT:    ret <2 x i8> [[D]]
+;
+  %b = trunc <2 x i64> %a to <2 x i32>
+  %c = ashr <2 x i32> %b, <i32 8, i32 11>
+  %d = trunc <2 x i32> %c to <2 x i8>
+  ret <2 x i8> %d
+}
+
 define i8 @trunc_ashr_trunc_multiuse(i64 %a) {
 ; CHECK-LABEL: @trunc_ashr_trunc_multiuse(
 ; CHECK-NEXT:    [[B:%.*]] = trunc i64 [[A:%.*]] to i32