[llvm] 2e5bba6 - [InstCombine] add tests for sext+lshr+trunc; NFC
Sanjay Patel via llvm-commits
llvm-commits at lists.llvm.org
Mon Jun 8 08:17:44 PDT 2020
Author: Sanjay Patel
Date: 2020-06-08T11:15:44-04:00
New Revision: 2e5bba67870cd77659e7921e35384b9f4edcc658
URL: https://github.com/llvm/llvm-project/commit/2e5bba67870cd77659e7921e35384b9f4edcc658
DIFF: https://github.com/llvm/llvm-project/commit/2e5bba67870cd77659e7921e35384b9f4edcc658.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 6ecd0e398654..6215e6694cd6 100644
--- a/llvm/test/Transforms/InstCombine/cast.ll
+++ b/llvm/test/Transforms/InstCombine/cast.ll
@@ -3,6 +3,9 @@
; RUN: opt < %s -instcombine -S -data-layout="E-p:64:64:64-p1:32:32:32-p2:64:64:64-p3:64:64:64-a0:0:8-f32:32:32-f64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-v64:64:64-v128:128:128-n8:16:32:64" | FileCheck %s --check-prefixes=ALL,BE
; RUN: opt < %s -instcombine -S -data-layout="e-p:64:64:64-p1:32:32:32-p2:64:64:64-p3:64:64:64-a0:0:8-f32:32:32-f64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-v64:64:64-v128:128:128-n8:16:32:64" | FileCheck %s --check-prefixes=ALL,LE
+declare void @use_i32(i32)
+declare void @use_v2i32(<2 x i32>)
+
@inbuf = external global [32832 x i8]
define i32 @test1(i32 %A) {
@@ -242,8 +245,8 @@ define i1 @test20(i1 %B) {
define i32 @test21(i32 %X) {
; ALL-LABEL: @test21(
-; ALL-NEXT: [[TMP1:%.*]] = and i32 [[X:%.*]], 255
-; ALL-NEXT: ret i32 [[TMP1]]
+; ALL-NEXT: [[SEXT:%.*]] = and i32 [[X:%.*]], 255
+; ALL-NEXT: ret i32 [[SEXT]]
;
%c1 = trunc i32 %X to i8
%c2 = sext i8 %c1 to i32
@@ -1500,6 +1503,183 @@ define i32 @test93(i32 %A) {
ret i32 %D
}
+define i8 @trunc_lshr_sext(i8 %A) {
+; ALL-LABEL: @trunc_lshr_sext(
+; ALL-NEXT: [[D:%.*]] = ashr i8 [[A:%.*]], 6
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i8 %A to i32
+ %C = lshr i32 %B, 6
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_sext_uses1(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_sext_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 i8>
+; ALL-NEXT: ret <2 x i8> [[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 i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_sext_uses2(i8 %A) {
+; ALL-LABEL: @trunc_lshr_sext_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 i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i8 %A to i32
+ %C = lshr i32 %B, 6
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+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]])
+; 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 i8>
+; ALL-NEXT: ret <2 x i8> [[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 i8>
+ ret <2 x i8> %D
+}
+
+define <2 x i8> @trunc_lshr_overshift_sext(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_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 i8>
+; ALL-NEXT: ret <2 x i8> [[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 i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_overshift_sext_uses1(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_uses1(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[D:%.*]] = ashr i8 [[A]], 7
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i8 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 8
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_overshift_sext_uses2(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_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 i8>
+; ALL-NEXT: ret <2 x i8> [[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 i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_overshift_sext_uses3(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift_sext_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 i8
+; ALL-NEXT: ret i8 [[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 i8
+ ret i8 %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>
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 25, i32 25>
+; ALL-NEXT: [[D:%.*]] = trunc <2 x i32> [[C]] to <2 x i8>
+; ALL-NEXT: ret <2 x i8> [[D]]
+;
+ %B = sext <2 x i8> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 25, i32 25>
+ %D = trunc <2 x i32> %C to <2 x i8>
+ ret <2 x i8> %D
+}
+
+define i8 @trunc_lshr_overshift2_sext_uses1(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift2_sext_uses1(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 25
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i8 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 25
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
+define <2 x i8> @trunc_lshr_overshift2_sext_uses2(<2 x i8> %A) {
+; ALL-LABEL: @trunc_lshr_overshift2_sext_uses2(
+; ALL-NEXT: [[B:%.*]] = sext <2 x i8> [[A:%.*]] to <2 x i32>
+; ALL-NEXT: [[C:%.*]] = lshr <2 x i32> [[B]], <i32 25, i32 25>
+; 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 i8> %A to <2 x i32>
+ %C = lshr <2 x i32> %B, <i32 25, i32 25>
+ 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_overshift2_sext_uses3(i8 %A) {
+; ALL-LABEL: @trunc_lshr_overshift2_sext_uses3(
+; ALL-NEXT: [[B:%.*]] = sext i8 [[A:%.*]] to i32
+; ALL-NEXT: call void @use_i32(i32 [[B]])
+; ALL-NEXT: [[C:%.*]] = lshr i32 [[B]], 25
+; ALL-NEXT: call void @use_i32(i32 [[C]])
+; ALL-NEXT: [[D:%.*]] = trunc i32 [[C]] to i8
+; ALL-NEXT: ret i8 [[D]]
+;
+ %B = sext i8 %A to i32
+ call void @use_i32(i32 %B)
+ %C = lshr i32 %B, 25
+ call void @use_i32(i32 %C)
+ %D = trunc i32 %C to i8
+ ret i8 %D
+}
+
; The following four tests sext + lshr + trunc patterns.
; PR33078
More information about the llvm-commits
mailing list