[llvm] 9b41821 - [InstCombine] add tests for sext+lshr+trunc; NFC
Sanjay Patel via llvm-commits
llvm-commits at lists.llvm.org
Mon Jun 8 11:43:38 PDT 2020
Author: Sanjay Patel
Date: 2020-06-08T14:41:50-04:00
New Revision: 9b41821c1b25986c5997377777f1843feb5294a2
URL: https://github.com/llvm/llvm-project/commit/9b41821c1b25986c5997377777f1843feb5294a2
DIFF: https://github.com/llvm/llvm-project/commit/9b41821c1b25986c5997377777f1843feb5294a2.diff
LOG: [InstCombine] add tests for sext+lshr+trunc; NFC
Shows missing transforms with extra uses and vectors.
Added:
Modified:
llvm/test/Transforms/InstCombine/cast.ll
Removed:
################################################################################
diff --git a/llvm/test/Transforms/InstCombine/cast.ll b/llvm/test/Transforms/InstCombine/cast.ll
index ac6c814cd45a..89a294c142df 100644
--- a/llvm/test/Transforms/InstCombine/cast.ll
+++ b/llvm/test/Transforms/InstCombine/cast.ll
@@ -1543,7 +1543,7 @@ define i8 @trunc_lshr_sext_uses2(i8 %A) {
ret i8 %D
}
-define <2 x i8 >@trunc_lshr_sext_uses3(<2 x i8> %A) {
+define <2 x i8> @trunc_lshr_sext_uses3(<2 x i8> %A) {
; ALL-LABEL: @trunc_lshr_sext_uses3(
; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
; ALL-NEXT: call void @use_v2i32(<2 x i32> [[B]])
@@ -1617,6 +1617,243 @@ define i8 @trunc_lshr_overshift_sext_uses3(i8 %A) {
ret i8 %D
}
+define i8 @trunc_lshr_sext_wide_input(i16 %A) {
+; ALL-LABEL: @trunc_lshr_sext_wide_input(
+; ALL-NEXT: [[C:%.*]] = ashr i16 [[A:%.*]], 9
+; ALL-NEXT: [[D:%.*]] = trunc i16 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i16 %A to i32
+ %C = lshr i32 %B, 9
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_sext_wide_input_uses1(<2 x i16> %A) {
+; ALL-LABEL: @trunc_lshr_sext_wide_input_uses1(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i16> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 9, i32 9>
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; ALL-NEXT: ret <2 x i8> [[D]]
+;
+ %B = sext <2 x i16> %A to <2 x i32>
+ call void @use_v2i32(<2 x i32> %B)
+ %C = lshr <2 x i32> %B, <i32 9, i32 9>
+ %D = trunc <2 x i32> %C to <2 x i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_sext_wide_input_uses2(i16 %A) {
+; ALL-LABEL: @trunc_lshr_sext_wide_input_uses2(
+; ALL-NEXT: [[B:%.*]] = sext i16 [[A:%.*]] to i32
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 9
+; ALL-NEXT: call void @use_i32(i32 [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i16 %A to i32
+ %C = lshr i32 %B, 9
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_sext_wide_input_uses3(<2 x i16> %A) {
+; ALL-LABEL: @trunc_lshr_sext_wide_input_uses3(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i16> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 9, i32 9>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; ALL-NEXT: ret <2 x i8> [[D]]
+;
+ %B = sext <2 x i16 >%A to <2 x i32>
+ call void @use_v2i32(<2 x i32> %B)
+ %C = lshr <2 x i32> %B, <i32 9, i32 9>
+ call void @use_v2i32(<2 x i32> %C)
+ %D = trunc <2 x i32 >%C to <2 x i8>
+ ret <2 x i8> %D
+}
+
+define <2 x i8> @trunc_lshr_overshift_wide_input_sext(<2 x i16> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_wide_input_sext(
+; ALL-NEXT: [[TMP1:%.*]] = ashr <2 x i16> [[A:%.*]], <i16 15, i16 15>
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i16> [[TMP1]] to <2 x i8>
+; ALL-NEXT: ret <2 x i8> [[D]]
+;
+ %B = sext <2 x i16> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 16, i32 16>
+ %D = trunc <2 x i32> %C to <2 x i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_overshift_sext_wide_input_uses1(i16 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_wide_input_uses1(
+; ALL-NEXT: [[B:%.*]] = sext i16 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 16
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i16 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 16
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_overshift_sext_wide_input_uses2(<2 x i16> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_wide_input_uses2(
+; ALL-NEXT: [[TMP1:%.*]] = ashr <2 x i16> [[A:%.*]], <i16 15, i16 15>
+; ALL-NEXT: [[C:%.*]] = zext <2 x i16> [[TMP1]] to <2 x i32>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i16> [[TMP1]] to <2 x i8>
+; ALL-NEXT: ret <2 x i8> [[D]]
+;
+ %B = sext <2 x i16> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 16, i32 16>
+ call void @use_v2i32(<2 x i32> %C)
+ %D = trunc <2 x i32> %C to <2 x i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_overshift_sext_wide_input_uses3(i16 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_wide_input_uses3(
+; ALL-NEXT: [[B:%.*]] = sext i16 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 16
+; ALL-NEXT: call void @use_i32(i32 [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i16 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 16
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define i16 @trunc_lshr_sext_narrow_input(i8 %A) {
+; ALL-LABEL: @trunc_lshr_sext_narrow_input(
+; ALL-NEXT: [[C:%.*]] = ashr i8 [[A:%.*]], 6
+; ALL-NEXT: [[D:%.*]] = sext i8 [[C]] to i16
+; ALL-NEXT: ret i16 [[D]]
+;
+ %B = sext i8 %A to i32
+ %C = lshr i32 %B, 6
+ %D = trunc i32 %C to i16
+ ret i16 %D
+}
+
+define <2 x i16> @trunc_lshr_sext_narrow_input_uses1(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_sext_narrow_input_uses1(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 6, i32 6>
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i16>
+; ALL-NEXT: ret <2 x i16> [[D]]
+;
+ %B = sext <2 x i8> %A to <2 x i32>
+ call void @use_v2i32(<2 x i32> %B)
+ %C = lshr <2 x i32> %B, <i32 6, i32 6>
+ %D = trunc <2 x i32> %C to <2 x i16>
+ ret <2 x i16> %D
+}
+
+define i16 @trunc_lshr_sext_narrow_input_uses2(i8 %A) {
+; ALL-LABEL: @trunc_lshr_sext_narrow_input_uses2(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 6
+; ALL-NEXT: call void @use_i32(i32 [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i16
+; ALL-NEXT: ret i16 [[D]]
+;
+ %B = sext i8 %A to i32
+ %C = lshr i32 %B, 6
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i16
+ ret i16 %D
+}
+
+define <2 x i16> @trunc_lshr_sext_narrow_input_uses3(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_sext_narrow_input_uses3(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 6, i32 6>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i16>
+; ALL-NEXT: ret <2 x i16> [[D]]
+;
+ %B = sext <2 x i8 >%A to <2 x i32>
+ call void @use_v2i32(<2 x i32> %B)
+ %C = lshr <2 x i32> %B, <i32 6, i32 6>
+ call void @use_v2i32(<2 x i32> %C)
+ %D = trunc <2 x i32 >%C to <2 x i16>
+ ret <2 x i16> %D
+}
+
+define <2 x i16> @trunc_lshr_overshift_narrow_input_sext(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_narrow_input_sext(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 8, i32 8>
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i16>
+; ALL-NEXT: ret <2 x i16> [[D]]
+;
+ %B = sext <2 x i8> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 8, i32 8>
+ %D = trunc <2 x i32> %C to <2 x i16>
+ ret <2 x i16> %D
+}
+
+define i16 @trunc_lshr_overshift_sext_narrow_input_uses1(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_narrow_input_uses1(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 8
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i16
+; ALL-NEXT: ret i16 [[D]]
+;
+ %B = sext i8 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 8
+ %D = trunc i32 %C to i16
+ ret i16 %D
+}
+
+define <2 x i16> @trunc_lshr_overshift_sext_narrow_input_uses2(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_narrow_input_uses2(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 8, i32 8>
+; ALL-NEXT: call void @use_v2i32(<2 x i32> [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i16>
+; ALL-NEXT: ret <2 x i16> [[D]]
+;
+ %B = sext <2 x i8> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 8, i32 8>
+ call void @use_v2i32(<2 x i32> %C)
+ %D = trunc <2 x i32> %C to <2 x i16>
+ ret <2 x i16> %D
+}
+
+define i16 @trunc_lshr_overshift_sext_narrow_input_uses3(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_narrow_input_uses3(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 8
+; ALL-NEXT: call void @use_i32(i32 [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i16
+; ALL-NEXT: ret i16 [[D]]
+;
+ %B = sext i8 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 8
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i16
+ ret i16 %D
+}
+
define <2 x i8> @trunc_lshr_overshift2_sext(<2 x i8> %A) {
; ALL-LABEL: @trunc_lshr_overshift2_sext(
; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
More information about the llvm-commits
mailing list