Skip to content

Commit cfdd5ca

Browse files
committed
[LV] Add tests for fmin reductions without fast-math flags.
Some of those reductions can be vectorized with extra checks. Extra tests for llvm#148239 and follow-ups.
1 parent a9cfe84 commit cfdd5ca

File tree

2 files changed

+509
-0
lines changed

2 files changed

+509
-0
lines changed
Lines changed: 74 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals none --version 5
2+
; RUN: opt -p loop-vectorize -mtriple=arm64-apple-macosx -S %s | FileCheck %s
3+
4+
define float @fmax_ogt_with_select(ptr %src, i64 %n) {
5+
; CHECK-LABEL: define float @fmax_ogt_with_select(
6+
; CHECK-SAME: ptr [[SRC:%.*]], i64 [[N:%.*]]) {
7+
; CHECK-NEXT: [[ENTRY:.*]]:
8+
; CHECK-NEXT: br label %[[LOOP:.*]]
9+
; CHECK: [[LOOP]]:
10+
; CHECK-NEXT: [[IV:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP]] ]
11+
; CHECK-NEXT: [[MAX:%.*]] = phi float [ -1.000000e+07, %[[ENTRY]] ], [ [[MAX_NEXT:%.*]], %[[LOOP]] ]
12+
; CHECK-NEXT: [[GEP_SRC:%.*]] = getelementptr inbounds nuw float, ptr [[SRC]], i64 [[IV]]
13+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP_SRC]], align 4
14+
; CHECK-NEXT: [[CMP:%.*]] = fcmp olt float [[L]], [[MAX]]
15+
; CHECK-NEXT: [[MAX_NEXT]] = select i1 [[CMP]], float [[L]], float [[MAX]]
16+
; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i64 [[IV]], 1
17+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i64 [[IV_NEXT]], [[N]]
18+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
19+
; CHECK: [[EXIT]]:
20+
; CHECK-NEXT: [[MAX_NEXT_LCSSA:%.*]] = phi float [ [[MAX_NEXT]], %[[LOOP]] ]
21+
; CHECK-NEXT: ret float [[MAX_NEXT_LCSSA]]
22+
;
23+
entry:
24+
br label %loop
25+
26+
loop:
27+
%iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
28+
%min = phi float [ -1.000000e+07, %entry ], [ %min.next, %loop ]
29+
%gep.src = getelementptr inbounds nuw float, ptr %src, i64 %iv
30+
%l = load float, ptr %gep.src, align 4
31+
%cmp = fcmp olt float %l, %min
32+
%min.next = select i1 %cmp, float %l, float %min
33+
%iv.next = add nuw nsw i64 %iv, 1
34+
%ec = icmp eq i64 %iv.next, %n
35+
br i1 %ec, label %exit, label %loop
36+
37+
exit:
38+
ret float %min.next
39+
}
40+
41+
define float @fminnum(ptr %src, i64 %n) {
42+
; CHECK-LABEL: define float @fminnum(
43+
; CHECK-SAME: ptr [[SRC:%.*]], i64 [[N:%.*]]) {
44+
; CHECK-NEXT: [[ENTRY:.*]]:
45+
; CHECK-NEXT: br label %[[LOOP:.*]]
46+
; CHECK: [[LOOP]]:
47+
; CHECK-NEXT: [[IV:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[IV_NEXT:%.*]], %[[LOOP]] ]
48+
; CHECK-NEXT: [[MAX:%.*]] = phi float [ -1.000000e+07, %[[ENTRY]] ], [ [[MAX_NEXT:%.*]], %[[LOOP]] ]
49+
; CHECK-NEXT: [[GEP_SRC:%.*]] = getelementptr inbounds nuw float, ptr [[SRC]], i64 [[IV]]
50+
; CHECK-NEXT: [[L:%.*]] = load float, ptr [[GEP_SRC]], align 4
51+
; CHECK-NEXT: [[MAX_NEXT]] = call float @llvm.minnum.f32(float [[MAX]], float [[L]])
52+
; CHECK-NEXT: [[IV_NEXT]] = add nuw nsw i64 [[IV]], 1
53+
; CHECK-NEXT: [[EC:%.*]] = icmp eq i64 [[IV_NEXT]], [[N]]
54+
; CHECK-NEXT: br i1 [[EC]], label %[[EXIT:.*]], label %[[LOOP]]
55+
; CHECK: [[EXIT]]:
56+
; CHECK-NEXT: [[MAX_NEXT_LCSSA:%.*]] = phi float [ [[MAX_NEXT]], %[[LOOP]] ]
57+
; CHECK-NEXT: ret float [[MAX_NEXT_LCSSA]]
58+
;
59+
entry:
60+
br label %loop
61+
62+
loop:
63+
%iv = phi i64 [ 0, %entry ], [ %iv.next, %loop ]
64+
%min = phi float [ -1.000000e+07, %entry ], [ %min.next, %loop ]
65+
%gep.src = getelementptr inbounds nuw float, ptr %src, i64 %iv
66+
%l = load float, ptr %gep.src, align 4
67+
%min.next = call float @llvm.minnum.f32(float %min, float %l)
68+
%iv.next = add nuw nsw i64 %iv, 1
69+
%ec = icmp eq i64 %iv.next, %n
70+
br i1 %ec, label %exit, label %loop
71+
72+
exit:
73+
ret float %min.next
74+
}

0 commit comments

Comments
 (0)