xref: /llvm-project/llvm/test/Transforms/IndVarSimplify/ult-sub-to-eq.ll (revision 864bb84a427de367528d15270790dd152871daf2)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -S -passes=indvars < %s | FileCheck %s
3
4; Provide legal integer types.
5target datalayout = "n8:16:32:64"
6
7
8define void @test1(ptr nocapture %autoc, ptr nocapture %data, float %d, i32 %data_len, i32 %sample) nounwind {
9; CHECK-LABEL: @test1(
10; CHECK-NEXT:  entry:
11; CHECK-NEXT:    [[SUB:%.*]] = sub i32 [[DATA_LEN:%.*]], [[SAMPLE:%.*]]
12; CHECK-NEXT:    [[CMP4:%.*]] = icmp eq i32 [[DATA_LEN]], [[SAMPLE]]
13; CHECK-NEXT:    br i1 [[CMP4]], label [[FOR_END:%.*]], label [[FOR_BODY_PREHEADER:%.*]]
14; CHECK:       for.body.preheader:
15; CHECK-NEXT:    [[WIDE_TRIP_COUNT:%.*]] = zext i32 [[SUB]] to i64
16; CHECK-NEXT:    br label [[FOR_BODY:%.*]]
17; CHECK:       for.body:
18; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ [[INDVARS_IV_NEXT:%.*]], [[FOR_BODY]] ], [ 0, [[FOR_BODY_PREHEADER]] ]
19; CHECK-NEXT:    [[TMP0:%.*]] = trunc i64 [[INDVARS_IV]] to i32
20; CHECK-NEXT:    [[ADD:%.*]] = add i32 [[TMP0]], [[SAMPLE]]
21; CHECK-NEXT:    [[IDXPROM:%.*]] = zext i32 [[ADD]] to i64
22; CHECK-NEXT:    [[ARRAYIDX:%.*]] = getelementptr inbounds float, ptr [[DATA:%.*]], i64 [[IDXPROM]]
23; CHECK-NEXT:    [[TMP1:%.*]] = load float, ptr [[ARRAYIDX]], align 4
24; CHECK-NEXT:    [[MUL:%.*]] = fmul float [[TMP1]], [[D:%.*]]
25; CHECK-NEXT:    [[ARRAYIDX2:%.*]] = getelementptr inbounds float, ptr [[AUTOC:%.*]], i64 [[INDVARS_IV]]
26; CHECK-NEXT:    [[TMP2:%.*]] = load float, ptr [[ARRAYIDX2]], align 4
27; CHECK-NEXT:    [[ADD3:%.*]] = fadd float [[TMP2]], [[MUL]]
28; CHECK-NEXT:    store float [[ADD3]], ptr [[ARRAYIDX2]], align 4
29; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add nuw nsw i64 [[INDVARS_IV]], 1
30; CHECK-NEXT:    [[EXITCOND:%.*]] = icmp ne i64 [[INDVARS_IV_NEXT]], [[WIDE_TRIP_COUNT]]
31; CHECK-NEXT:    br i1 [[EXITCOND]], label [[FOR_BODY]], label [[FOR_END_LOOPEXIT:%.*]]
32; CHECK:       for.end.loopexit:
33; CHECK-NEXT:    br label [[FOR_END]]
34; CHECK:       for.end:
35; CHECK-NEXT:    ret void
36;
37entry:
38  %sub = sub i32 %data_len, %sample
39  %cmp4 = icmp eq i32 %data_len, %sample
40  br i1 %cmp4, label %for.end, label %for.body
41
42for.body:                                         ; preds = %entry, %for.body
43  %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
44  %0 = trunc i64 %indvars.iv to i32
45  %add = add i32 %0, %sample
46  %idxprom = zext i32 %add to i64
47  %arrayidx = getelementptr inbounds float, ptr %data, i64 %idxprom
48  %1 = load float, ptr %arrayidx, align 4
49  %mul = fmul float %1, %d
50  %arrayidx2 = getelementptr inbounds float, ptr %autoc, i64 %indvars.iv
51  %2 = load float, ptr %arrayidx2, align 4
52  %add3 = fadd float %2, %mul
53  store float %add3, ptr %arrayidx2, align 4
54  %indvars.iv.next = add i64 %indvars.iv, 1
55  %3 = trunc i64 %indvars.iv.next to i32
56  %cmp = icmp ult i32 %3, %sub
57  br i1 %cmp, label %for.body, label %for.end
58
59for.end:                                          ; preds = %for.body, %entry
60  ret void
61
62
63; check that we turn the IV test into an eq.
64}
65
66