Skip to content

Commit 81c0373

Browse files
committed
[LoopVectorize] Add test case for minloc reduction
This patch adds a test case extracted from Polyhedron benchmark. https://fortran.uk/fortran-compiler-comparisons/the-polyhedron-solutions-benchmark-suite/ The test is specifically interesting for vectorizing min/max reduction pattern.
1 parent 9e6fc8d commit 81c0373

File tree

1 file changed

+84
-0
lines changed

1 file changed

+84
-0
lines changed
Lines changed: 84 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,84 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2+
; RUN: opt -passes=loop-vectorize -S %s | FileCheck %s --check-prefix=CHECK-REV-MIN
3+
4+
; This test case is extracted from rnflow (fortran) benchmark in polyhedron benchmark suite.
5+
; The function minlst primarily takes two indices (i.e. range), scans backwards in the range
6+
; and returns the firstIV of the minimum value.
7+
8+
define fastcc i32 @minlst(i32 %.0.val, i32 %.0.val1, ptr %.0.val3) {
9+
; CHECK-REV-MIN-LABEL: define internal fastcc i32 @_QFcptrf2Pminlst(
10+
; CHECK-REV-MIN-SAME: i32 [[DOT0_VAL:%.*]], i32 [[DOT0_VAL1:%.*]], ptr [[DOT0_VAL3:%.*]]) unnamed_addr {
11+
; CHECK-REV-MIN-NEXT: [[TMP1:%.*]] = sext i32 [[DOT0_VAL]] to i64
12+
; CHECK-REV-MIN-NEXT: [[TMP2:%.*]] = sub i32 0, [[DOT0_VAL1]]
13+
; CHECK-REV-MIN-NEXT: [[TMP3:%.*]] = sext i32 [[TMP2]] to i64
14+
; CHECK-REV-MIN-NEXT: [[TMP4:%.*]] = add nsw i64 [[TMP1]], [[TMP3]]
15+
; CHECK-REV-MIN-NEXT: [[TMP5:%.*]] = sub nsw i64 0, [[TMP4]]
16+
; CHECK-REV-MIN-NEXT: [[INVARIANT_GEP:%.*]] = getelementptr i8, ptr [[DOT0_VAL3]], i64 -8
17+
; CHECK-REV-MIN-NEXT: [[INVARIANT_GEP5:%.*]] = getelementptr i8, ptr [[DOT0_VAL3]], i64 -4
18+
; CHECK-REV-MIN-NEXT: [[TMP6:%.*]] = icmp slt i64 [[TMP4]], 0
19+
; CHECK-REV-MIN-NEXT: br i1 [[TMP6]], label %[[DOTLR_PH_PREHEADER:.*]], [[DOT_CRIT_EDGE:label %.*]]
20+
; CHECK-REV-MIN: [[_LR_PH_PREHEADER:.*:]]
21+
; CHECK-REV-MIN-NEXT: [[TMP7:%.*]] = sext i32 [[DOT0_VAL1]] to i64
22+
; CHECK-REV-MIN-NEXT: br label %[[DOTLR_PH:.*]]
23+
; CHECK-REV-MIN: [[_LR_PH:.*:]]
24+
; CHECK-REV-MIN-NEXT: [[INDVARS_IV:%.*]] = phi i64 [ [[TMP7]], %[[DOTLR_PH_PREHEADER]] ], [ [[INDVARS_IV_NEXT:%.*]], %[[DOTLR_PH]] ]
25+
; CHECK-REV-MIN-NEXT: [[TMP8:%.*]] = phi i64 [ [[TMP14:%.*]], %[[DOTLR_PH]] ], [ [[TMP5]], %[[DOTLR_PH_PREHEADER]] ]
26+
; CHECK-REV-MIN-NEXT: [[DOT07:%.*]] = phi i32 [ [[DOT1:%.*]], %[[DOTLR_PH]] ], [ [[DOT0_VAL1]], %[[DOTLR_PH_PREHEADER]] ]
27+
; CHECK-REV-MIN-NEXT: [[INDVARS_IV_NEXT]] = add nsw i64 [[INDVARS_IV]], -1
28+
; CHECK-REV-MIN-NEXT: [[GEP:%.*]] = getelementptr float, ptr [[INVARIANT_GEP]], i64 [[INDVARS_IV]]
29+
; CHECK-REV-MIN-NEXT: [[TMP9:%.*]] = load float, ptr [[GEP]], align 4
30+
; CHECK-REV-MIN-NEXT: [[TMP10:%.*]] = sext i32 [[DOT07]] to i64
31+
; CHECK-REV-MIN-NEXT: [[GEP6:%.*]] = getelementptr float, ptr [[INVARIANT_GEP5]], i64 [[TMP10]]
32+
; CHECK-REV-MIN-NEXT: [[TMP11:%.*]] = load float, ptr [[GEP6]], align 4
33+
; CHECK-REV-MIN-NEXT: [[TMP12:%.*]] = fcmp contract olt float [[TMP9]], [[TMP11]]
34+
; CHECK-REV-MIN-NEXT: [[TMP13:%.*]] = trunc nsw i64 [[INDVARS_IV_NEXT]] to i32
35+
; CHECK-REV-MIN-NEXT: [[DOT1]] = select i1 [[TMP12]], i32 [[TMP13]], i32 [[DOT07]]
36+
; CHECK-REV-MIN-NEXT: [[TMP14]] = add nsw i64 [[TMP8]], -1
37+
; CHECK-REV-MIN-NEXT: [[TMP15:%.*]] = icmp sgt i64 [[TMP8]], 1
38+
; CHECK-REV-MIN-NEXT: br i1 [[TMP15]], label %[[DOTLR_PH]], label %[[DOT_CRIT_EDGE_LOOPEXIT:.*]]
39+
; CHECK-REV-MIN: [[__CRIT_EDGE_LOOPEXIT:.*:]]
40+
; CHECK-REV-MIN-NEXT: [[DOT1_LCSSA:%.*]] = phi i32 [ [[DOT1]], %[[DOTLR_PH]] ]
41+
; CHECK-REV-MIN-NEXT: br [[DOT_CRIT_EDGE]]
42+
; CHECK-REV-MIN: [[__CRIT_EDGE:.*:]]
43+
; CHECK-REV-MIN-NEXT: [[DOT0_LCSSA:%.*]] = phi i32 [ [[DOT0_VAL1]], [[TMP0:%.*]] ], [ [[DOT1_LCSSA]], %[[DOT_CRIT_EDGE_LOOPEXIT]] ]
44+
; CHECK-REV-MIN-NEXT: ret i32 [[DOT0_LCSSA]]
45+
;
46+
%1 = sext i32 %.0.val to i64
47+
%2 = sub i32 0, %.0.val1
48+
%3 = sext i32 %2 to i64
49+
%4 = add nsw i64 %1, %3
50+
%5 = sub nsw i64 0, %4
51+
%invariant.gep = getelementptr i8, ptr %.0.val3, i64 -8
52+
%invariant.gep5 = getelementptr i8, ptr %.0.val3, i64 -4
53+
%6 = icmp slt i64 %4, 0
54+
br i1 %6, label %.lr.ph.preheader, label %._crit_edge
55+
56+
.lr.ph.preheader: ; preds = %0
57+
%7 = sext i32 %.0.val1 to i64
58+
br label %.lr.ph
59+
60+
.lr.ph: ; preds = %.lr.ph.preheader, %.lr.ph
61+
%indvars.iv = phi i64 [ %7, %.lr.ph.preheader ], [ %indvars.iv.next, %.lr.ph ]
62+
%8 = phi i64 [ %14, %.lr.ph ], [ %5, %.lr.ph.preheader ]
63+
%.07 = phi i32 [ %.1, %.lr.ph ], [ %.0.val1, %.lr.ph.preheader ]
64+
%indvars.iv.next = add nsw i64 %indvars.iv, -1
65+
%gep = getelementptr float, ptr %invariant.gep, i64 %indvars.iv
66+
%9 = load float, ptr %gep, align 4
67+
%10 = sext i32 %.07 to i64
68+
%gep6 = getelementptr float, ptr %invariant.gep5, i64 %10
69+
%11 = load float, ptr %gep6, align 4
70+
%12 = fcmp contract olt float %9, %11
71+
%13 = trunc nsw i64 %indvars.iv.next to i32
72+
%.1 = select i1 %12, i32 %13, i32 %.07
73+
%14 = add nsw i64 %8, -1
74+
%15 = icmp sgt i64 %8, 1
75+
br i1 %15, label %.lr.ph, label %._crit_edge.loopexit
76+
77+
._crit_edge.loopexit: ; preds = %.lr.ph
78+
%.1.lcssa = phi i32 [ %.1, %.lr.ph ]
79+
br label %._crit_edge
80+
81+
._crit_edge: ; preds = %._crit_edge.loopexit, %0
82+
%.0.lcssa = phi i32 [ %.0.val1, %0 ], [ %.1.lcssa, %._crit_edge.loopexit ]
83+
ret i32 %.0.lcssa
84+
}

0 commit comments

Comments
 (0)