[llvm] 86e6523 - [SLP] add tests for poison-safe logical reductions; NFC
Sanjay Patel via llvm-commits
llvm-commits at lists.llvm.org
Fri Jul 9 12:32:21 PDT 2021
Author: Sanjay Patel
Date: 2021-07-09T15:32:12-04:00
New Revision: 86e65234404fa329cd65b5522aae8e82f4fa152b
URL: https://github.com/llvm/llvm-project/commit/86e65234404fa329cd65b5522aae8e82f4fa152b
DIFF: https://github.com/llvm/llvm-project/commit/86e65234404fa329cd65b5522aae8e82f4fa152b.diff
LOG: [SLP] add tests for poison-safe logical reductions; NFC
Added:
llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll
Modified:
Removed:
################################################################################
diff --git a/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll b/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll
new file mode 100644
index 000000000000..c911fdc301dc
--- /dev/null
+++ b/llvm/test/Transforms/SLPVectorizer/X86/reduction-logical.ll
@@ -0,0 +1,213 @@
+; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
+; RUN: opt < %s -slp-vectorizer -mtriple=x86_64-- -S | FileCheck %s
+
+define i1 @logical_and_icmp(<4 x i32> %x) {
+; CHECK-LABEL: @logical_and_icmp(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x i32> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x i32> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x i32> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = icmp slt i32 [[X0]], 0
+; CHECK-NEXT: [[C1:%.*]] = icmp slt i32 [[X1]], 0
+; CHECK-NEXT: [[C2:%.*]] = icmp slt i32 [[X2]], 0
+; CHECK-NEXT: [[C3:%.*]] = icmp slt i32 [[X3]], 0
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 [[C2]], i1 false
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 [[C3]], i1 false
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x i32> %x, i32 0
+ %x1 = extractelement <4 x i32> %x, i32 1
+ %x2 = extractelement <4 x i32> %x, i32 2
+ %x3 = extractelement <4 x i32> %x, i32 3
+ %c0 = icmp slt i32 %x0, 0
+ %c1 = icmp slt i32 %x1, 0
+ %c2 = icmp slt i32 %x2, 0
+ %c3 = icmp slt i32 %x3, 0
+ %s1 = select i1 %c0, i1 %c1, i1 false
+ %s2 = select i1 %s1, i1 %c2, i1 false
+ %s3 = select i1 %s2, i1 %c3, i1 false
+ ret i1 %s3
+}
+
+define i1 @logical_or_icmp(<4 x i32> %x, <4 x i32> %y) {
+; CHECK-LABEL: @logical_or_icmp(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x i32> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x i32> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x i32> [[X]], i32 3
+; CHECK-NEXT: [[Y0:%.*]] = extractelement <4 x i32> [[Y:%.*]], i32 0
+; CHECK-NEXT: [[Y1:%.*]] = extractelement <4 x i32> [[Y]], i32 1
+; CHECK-NEXT: [[Y2:%.*]] = extractelement <4 x i32> [[Y]], i32 2
+; CHECK-NEXT: [[Y3:%.*]] = extractelement <4 x i32> [[Y]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = icmp slt i32 [[X0]], [[Y0]]
+; CHECK-NEXT: [[C1:%.*]] = icmp slt i32 [[X1]], [[Y1]]
+; CHECK-NEXT: [[C2:%.*]] = icmp slt i32 [[X2]], [[Y2]]
+; CHECK-NEXT: [[C3:%.*]] = icmp slt i32 [[X3]], [[Y3]]
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 true, i1 [[C1]]
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 true, i1 [[C2]]
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 true, i1 [[C3]]
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x i32> %x, i32 0
+ %x1 = extractelement <4 x i32> %x, i32 1
+ %x2 = extractelement <4 x i32> %x, i32 2
+ %x3 = extractelement <4 x i32> %x, i32 3
+ %y0 = extractelement <4 x i32> %y, i32 0
+ %y1 = extractelement <4 x i32> %y, i32 1
+ %y2 = extractelement <4 x i32> %y, i32 2
+ %y3 = extractelement <4 x i32> %y, i32 3
+ %c0 = icmp slt i32 %x0, %y0
+ %c1 = icmp slt i32 %x1, %y1
+ %c2 = icmp slt i32 %x2, %y2
+ %c3 = icmp slt i32 %x3, %y3
+ %s1 = select i1 %c0, i1 true, i1 %c1
+ %s2 = select i1 %s1, i1 true, i1 %c2
+ %s3 = select i1 %s2, i1 true, i1 %c3
+ ret i1 %s3
+}
+
+define i1 @logical_and_fcmp(<4 x float> %x) {
+; CHECK-LABEL: @logical_and_fcmp(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x float> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x float> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x float> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x float> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = fcmp olt float [[X0]], 0.000000e+00
+; CHECK-NEXT: [[C1:%.*]] = fcmp olt float [[X1]], 0.000000e+00
+; CHECK-NEXT: [[C2:%.*]] = fcmp olt float [[X2]], 0.000000e+00
+; CHECK-NEXT: [[C3:%.*]] = fcmp olt float [[X3]], 0.000000e+00
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 [[C2]], i1 false
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 [[C3]], i1 false
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x float> %x, i32 0
+ %x1 = extractelement <4 x float> %x, i32 1
+ %x2 = extractelement <4 x float> %x, i32 2
+ %x3 = extractelement <4 x float> %x, i32 3
+ %c0 = fcmp olt float %x0, 0.0
+ %c1 = fcmp olt float %x1, 0.0
+ %c2 = fcmp olt float %x2, 0.0
+ %c3 = fcmp olt float %x3, 0.0
+ %s1 = select i1 %c0, i1 %c1, i1 false
+ %s2 = select i1 %s1, i1 %c2, i1 false
+ %s3 = select i1 %s2, i1 %c3, i1 false
+ ret i1 %s3
+}
+
+define i1 @logical_or_fcmp(<4 x float> %x) {
+; CHECK-LABEL: @logical_or_fcmp(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x float> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x float> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x float> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x float> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = fcmp olt float [[X0]], 0.000000e+00
+; CHECK-NEXT: [[C1:%.*]] = fcmp olt float [[X1]], 0.000000e+00
+; CHECK-NEXT: [[C2:%.*]] = fcmp olt float [[X2]], 0.000000e+00
+; CHECK-NEXT: [[C3:%.*]] = fcmp olt float [[X3]], 0.000000e+00
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 true, i1 [[C1]]
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 true, i1 [[C2]]
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 true, i1 [[C3]]
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x float> %x, i32 0
+ %x1 = extractelement <4 x float> %x, i32 1
+ %x2 = extractelement <4 x float> %x, i32 2
+ %x3 = extractelement <4 x float> %x, i32 3
+ %c0 = fcmp olt float %x0, 0.0
+ %c1 = fcmp olt float %x1, 0.0
+ %c2 = fcmp olt float %x2, 0.0
+ %c3 = fcmp olt float %x3, 0.0
+ %s1 = select i1 %c0, i1 true, i1 %c1
+ %s2 = select i1 %s1, i1 true, i1 %c2
+ %s3 = select i1 %s2, i1 true, i1 %c3
+ ret i1 %s3
+}
+
+define i1 @logical_and_icmp_
diff _preds(<4 x i32> %x) {
+; CHECK-LABEL: @logical_and_icmp_
diff _preds(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x i32> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x i32> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x i32> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = icmp ult i32 [[X0]], 0
+; CHECK-NEXT: [[C1:%.*]] = icmp slt i32 [[X1]], 0
+; CHECK-NEXT: [[C2:%.*]] = icmp sgt i32 [[X2]], 0
+; CHECK-NEXT: [[C3:%.*]] = icmp slt i32 [[X3]], 0
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 [[C2]], i1 false
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 [[C3]], i1 false
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x i32> %x, i32 0
+ %x1 = extractelement <4 x i32> %x, i32 1
+ %x2 = extractelement <4 x i32> %x, i32 2
+ %x3 = extractelement <4 x i32> %x, i32 3
+ %c0 = icmp ult i32 %x0, 0
+ %c1 = icmp slt i32 %x1, 0
+ %c2 = icmp sgt i32 %x2, 0
+ %c3 = icmp slt i32 %x3, 0
+ %s1 = select i1 %c0, i1 %c1, i1 false
+ %s2 = select i1 %s1, i1 %c2, i1 false
+ %s3 = select i1 %s2, i1 %c3, i1 false
+ ret i1 %s3
+}
+
+define i1 @logical_and_icmp_
diff _const(<4 x i32> %x) {
+; CHECK-LABEL: @logical_and_icmp_
diff _const(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x i32> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x i32> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x i32> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = icmp sgt i32 [[X0]], 0
+; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[X1]], 1
+; CHECK-NEXT: [[C2:%.*]] = icmp sgt i32 [[X2]], 2
+; CHECK-NEXT: [[C3:%.*]] = icmp sgt i32 [[X3]], 3
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 [[C2]], i1 false
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 [[C3]], i1 false
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x i32> %x, i32 0
+ %x1 = extractelement <4 x i32> %x, i32 1
+ %x2 = extractelement <4 x i32> %x, i32 2
+ %x3 = extractelement <4 x i32> %x, i32 3
+ %c0 = icmp sgt i32 %x0, 0
+ %c1 = icmp sgt i32 %x1, 1
+ %c2 = icmp sgt i32 %x2, 2
+ %c3 = icmp sgt i32 %x3, 3
+ %s1 = select i1 %c0, i1 %c1, i1 false
+ %s2 = select i1 %s1, i1 %c2, i1 false
+ %s3 = select i1 %s2, i1 %c3, i1 false
+ ret i1 %s3
+}
+
+define i1 @mixed_logical_icmp(<4 x i32> %x) {
+; CHECK-LABEL: @mixed_logical_icmp(
+; CHECK-NEXT: [[X0:%.*]] = extractelement <4 x i32> [[X:%.*]], i32 0
+; CHECK-NEXT: [[X1:%.*]] = extractelement <4 x i32> [[X]], i32 1
+; CHECK-NEXT: [[X2:%.*]] = extractelement <4 x i32> [[X]], i32 2
+; CHECK-NEXT: [[X3:%.*]] = extractelement <4 x i32> [[X]], i32 3
+; CHECK-NEXT: [[C0:%.*]] = icmp sgt i32 [[X0]], 0
+; CHECK-NEXT: [[C1:%.*]] = icmp sgt i32 [[X1]], 0
+; CHECK-NEXT: [[C2:%.*]] = icmp sgt i32 [[X2]], 0
+; CHECK-NEXT: [[C3:%.*]] = icmp sgt i32 [[X3]], 0
+; CHECK-NEXT: [[S1:%.*]] = select i1 [[C0]], i1 [[C1]], i1 false
+; CHECK-NEXT: [[S2:%.*]] = select i1 [[S1]], i1 true, i1 [[C2]]
+; CHECK-NEXT: [[S3:%.*]] = select i1 [[S2]], i1 [[C3]], i1 false
+; CHECK-NEXT: ret i1 [[S3]]
+;
+ %x0 = extractelement <4 x i32> %x, i32 0
+ %x1 = extractelement <4 x i32> %x, i32 1
+ %x2 = extractelement <4 x i32> %x, i32 2
+ %x3 = extractelement <4 x i32> %x, i32 3
+ %c0 = icmp sgt i32 %x0, 0
+ %c1 = icmp sgt i32 %x1, 0
+ %c2 = icmp sgt i32 %x2, 0
+ %c3 = icmp sgt i32 %x3, 0
+ %s1 = select i1 %c0, i1 %c1, i1 false
+ %s2 = select i1 %s1, i1 true, i1 %c2
+ %s3 = select i1 %s2, i1 %c3, i1 false
+ ret i1 %s3
+}
More information about the llvm-commits
mailing list