Skip to content

Commit c39cc9f

Browse files
fhahnfhahn
committed
[LV] Add tests for maximumnum/minimumnum reductions.
Add reduction tests with maximumnum/minimumnum intrinsics.
1 parent ee2aba7 commit c39cc9f

File tree

1 file changed

+146
-0
lines changed

1 file changed

+146
-0
lines changed

‎llvm/test/Transforms/LoopVectorize/minimumnum-maximumnum-reductions.ll

+146
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,146 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2+
; RUN: opt -passes=loop-vectorize -force-vector-width=2 -force-vector-interleave=2 -S %s | FileCheck %s
3+
; RUN: opt -passes=loop-vectorize -force-vector-width=2 -force-vector-interleave=2 -prefer-inloop-reductions -S %s | FileCheck %s
4+
5+
define float @maximumnum_intrinsic(ptr readonly %x) {
6+
; CHECK-LABEL: define float @maximumnum_intrinsic(
7+
; CHECK-SAME: ptr readonly [[X:%.*]]) {
8+
; CHECK-NEXT: [[ENTRY:.*]]:
9+
; CHECK-NEXT: br label %[[LOOP:.*]]
10+
; CHECK: [[LOOP]]:
11+
; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[INC:%.*]], %[[LOOP]] ]
12+
; CHECK-NEXT: [[RED:%.*]] = phi float [ 0.000000e+00, %[[ENTRY]] ], [ [[RED_NEXT:%.*]], %[[LOOP]] ]
13+
; CHECK-NEXT: [[GEP:%.*]] = getelementptr inbounds float, ptr [[X]], i32 [[IV]]
14+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP]], align 4
15+
; CHECK-NEXT: [[RED_NEXT]] = tail call float @llvm.maximumnum.f32(float [[RED]], float [[L]])
16+
; CHECK-NEXT: [[INC]] = add nuw nsw i32 [[IV]], 1
17+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i32 [[INC]], 1024
18+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
19+
; CHECK: [[EXIT]]:
20+
; CHECK-NEXT: [[RED_NEXT_LCSSA:%.*]] = phi float [ [[RED_NEXT]], %[[LOOP]] ]
21+
; CHECK-NEXT: ret float [[RED_NEXT_LCSSA]]
22+
;
23+
entry:
24+
br label %loop
25+
26+
loop:
27+
%iv = phi i32 [ 0, %entry ], [ %inc, %loop ]
28+
%red = phi float [ 0.000000e+00, %entry ], [ %red.next, %loop ]
29+
%gep = getelementptr inbounds float, ptr %x, i32 %iv
30+
%l = load float, ptr %gep, align 4
31+
%red.next = tail call float @llvm.maximumnum.f32(float %red, float %l)
32+
%inc = add nuw nsw i32 %iv, 1
33+
%ec = icmp eq i32 %inc, 1024
34+
br i1 %ec, label %exit, label %loop
35+
36+
exit:
37+
ret float %red.next
38+
}
39+
40+
define float @maximumnum_intrinsic_fast(ptr readonly %x) {
41+
; CHECK-LABEL: define float @maximumnum_intrinsic_fast(
42+
; CHECK-SAME: ptr readonly [[X:%.*]]) {
43+
; CHECK-NEXT: [[ENTRY:.*]]:
44+
; CHECK-NEXT: br label %[[LOOP:.*]]
45+
; CHECK: [[LOOP]]:
46+
; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[INC:%.*]], %[[LOOP]] ]
47+
; CHECK-NEXT: [[RED:%.*]] = phi float [ 0.000000e+00, %[[ENTRY]] ], [ [[RED_NEXT:%.*]], %[[LOOP]] ]
48+
; CHECK-NEXT: [[GEP:%.*]] = getelementptr inbounds float, ptr [[X]], i32 [[IV]]
49+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP]], align 4
50+
; CHECK-NEXT: [[RED_NEXT]] = tail call fast float @llvm.maximumnum.f32(float [[RED]], float [[L]])
51+
; CHECK-NEXT: [[INC]] = add nuw nsw i32 [[IV]], 1
52+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i32 [[INC]], 1024
53+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
54+
; CHECK: [[EXIT]]:
55+
; CHECK-NEXT: [[RED_NEXT_LCSSA:%.*]] = phi float [ [[RED_NEXT]], %[[LOOP]] ]
56+
; CHECK-NEXT: ret float [[RED_NEXT_LCSSA]]
57+
;
58+
entry:
59+
br label %loop
60+
61+
loop:
62+
%iv = phi i32 [ 0, %entry ], [ %inc, %loop ]
63+
%red = phi float [ 0.000000e+00, %entry ], [ %red.next, %loop ]
64+
%gep = getelementptr inbounds float, ptr %x, i32 %iv
65+
%l = load float, ptr %gep, align 4
66+
%red.next = tail call fast float @llvm.maximumnum.f32(float %red, float %l)
67+
%inc = add nuw nsw i32 %iv, 1
68+
%ec = icmp eq i32 %inc, 1024
69+
br i1 %ec, label %exit, label %loop
70+
71+
exit:
72+
ret float %red.next
73+
}
74+
75+
define float @minimumnum_intrinsic(ptr readonly %x) {
76+
; CHECK-LABEL: define float @minimumnum_intrinsic(
77+
; CHECK-SAME: ptr readonly [[X:%.*]]) {
78+
; CHECK-NEXT: [[ENTRY:.*]]:
79+
; CHECK-NEXT: br label %[[LOOP:.*]]
80+
; CHECK: [[LOOP]]:
81+
; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[INC:%.*]], %[[LOOP]] ]
82+
; CHECK-NEXT: [[RED:%.*]] = phi float [ 0.000000e+00, %[[ENTRY]] ], [ [[RED_NEXT:%.*]], %[[LOOP]] ]
83+
; CHECK-NEXT: [[GEP:%.*]] = getelementptr inbounds float, ptr [[X]], i32 [[IV]]
84+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP]], align 4
85+
; CHECK-NEXT: [[RED_NEXT]] = tail call float @llvm.minimumnum.f32(float [[RED]], float [[L]])
86+
; CHECK-NEXT: [[INC]] = add nuw nsw i32 [[IV]], 1
87+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i32 [[INC]], 1024
88+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
89+
; CHECK: [[EXIT]]:
90+
; CHECK-NEXT: [[RED_NEXT_LCSSA:%.*]] = phi float [ [[RED_NEXT]], %[[LOOP]] ]
91+
; CHECK-NEXT: ret float [[RED_NEXT_LCSSA]]
92+
;
93+
entry:
94+
br label %loop
95+
96+
loop:
97+
%iv = phi i32 [ 0, %entry ], [ %inc, %loop ]
98+
%red = phi float [ 0.000000e+00, %entry ], [ %red.next, %loop ]
99+
%gep = getelementptr inbounds float, ptr %x, i32 %iv
100+
%l = load float, ptr %gep, align 4
101+
%red.next = tail call float @llvm.minimumnum.f32(float %red, float %l)
102+
%inc = add nuw nsw i32 %iv, 1
103+
%ec = icmp eq i32 %inc, 1024
104+
br i1 %ec, label %exit, label %loop
105+
106+
exit:
107+
ret float %red.next
108+
}
109+
110+
define float @minimumnum_intrinsic_fast(ptr readonly %x) {
111+
; CHECK-LABEL: define float @minimumnum_intrinsic_fast(
112+
; CHECK-SAME: ptr readonly [[X:%.*]]) {
113+
; CHECK-NEXT: [[ENTRY:.*]]:
114+
; CHECK-NEXT: br label %[[LOOP:.*]]
115+
; CHECK: [[LOOP]]:
116+
; CHECK-NEXT: [[IV:%.*]] = phi i32 [ 0, %[[ENTRY]] ], [ [[INC:%.*]], %[[LOOP]] ]
117+
; CHECK-NEXT: [[RED:%.*]] = phi float [ 0.000000e+00, %[[ENTRY]] ], [ [[RED_NEXT:%.*]], %[[LOOP]] ]
118+
; CHECK-NEXT: [[GEP:%.*]] = getelementptr inbounds float, ptr [[X]], i32 [[IV]]
119+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP]], align 4
120+
; CHECK-NEXT: [[RED_NEXT]] = tail call fast float @llvm.minimumnum.f32(float [[RED]], float [[L]])
121+
; CHECK-NEXT: [[INC]] = add nuw nsw i32 [[IV]], 1
122+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i32 [[INC]], 1024
123+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
124+
; CHECK: [[EXIT]]:
125+
; CHECK-NEXT: [[RED_NEXT_LCSSA:%.*]] = phi float [ [[RED_NEXT]], %[[LOOP]] ]
126+
; CHECK-NEXT: ret float [[RED_NEXT_LCSSA]]
127+
;
128+
entry:
129+
br label %loop
130+
131+
loop:
132+
%iv = phi i32 [ 0, %entry ], [ %inc, %loop ]
133+
%red = phi float [ 0.000000e+00, %entry ], [ %red.next, %loop ]
134+
%gep = getelementptr inbounds float, ptr %x, i32 %iv
135+
%l = load float, ptr %gep, align 4
136+
%red.next = tail call fast float @llvm.minimumnum.f32(float %red, float %l)
137+
%inc = add nuw nsw i32 %iv, 1
138+
%ec = icmp eq i32 %inc, 1024
139+
br i1 %ec, label %exit, label %loop
140+
141+
exit:
142+
ret float %red.next
143+
}
144+
145+
declare float @llvm.minimumnum.f32(float, float)
146+
declare float @llvm.maximumnum.f32(float, float)

0 commit comments

Comments
 (0)