xref: /llvm-project/llvm/test/Transforms/LoopVectorize/non-const-n.ll (revision 3e992d81afc3925a8685eb15f794dd4a6ba3e97e)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 3
2; RUN: opt < %s -passes=loop-vectorize,dce,instcombine -force-vector-interleave=1 -force-vector-width=4 -S | FileCheck %s
3
4target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
5
6@b = common global [2048 x i32] zeroinitializer, align 16
7@c = common global [2048 x i32] zeroinitializer, align 16
8@a = common global [2048 x i32] zeroinitializer, align 16
9
10define void @example1(i32 %n) nounwind uwtable ssp {
11; CHECK-LABEL: define void @example1(
12; CHECK-SAME: i32 [[N:%.*]]) #[[ATTR0:[0-9]+]] {
13; CHECK-NEXT:  entry:
14; CHECK-NEXT:    [[N4:%.*]] = shl i32 [[N]], 2
15; CHECK-NEXT:    [[TMP0:%.*]] = add i32 [[N4]], -4
16; CHECK-NEXT:    [[TMP1:%.*]] = zext i32 [[TMP0]] to i64
17; CHECK-NEXT:    br i1 false, label [[SCALAR_PH:%.*]], label [[VECTOR_PH:%.*]]
18; CHECK:       vector.ph:
19; CHECK-NEXT:    br label [[VECTOR_BODY:%.*]]
20; CHECK:       vector.body:
21; CHECK-NEXT:    [[INDEX:%.*]] = phi i64 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ]
22; CHECK-NEXT:    [[TMP2:%.*]] = getelementptr inbounds [2048 x i32], ptr @b, i64 0, i64 [[INDEX]]
23; CHECK-NEXT:    [[WIDE_LOAD:%.*]] = load <4 x i32>, ptr [[TMP2]], align 4
24; CHECK-NEXT:    [[TMP3:%.*]] = getelementptr inbounds [2048 x i32], ptr @c, i64 0, i64 [[INDEX]]
25; CHECK-NEXT:    [[WIDE_LOAD1:%.*]] = load <4 x i32>, ptr [[TMP3]], align 4
26; CHECK-NEXT:    [[TMP4:%.*]] = add nsw <4 x i32> [[WIDE_LOAD1]], [[WIDE_LOAD]]
27; CHECK-NEXT:    [[TMP5:%.*]] = getelementptr inbounds [2048 x i32], ptr @a, i64 0, i64 [[INDEX]]
28; CHECK-NEXT:    store <4 x i32> [[TMP4]], ptr [[TMP5]], align 4
29; CHECK-NEXT:    [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 4
30; CHECK-NEXT:    [[TMP6:%.*]] = icmp eq i64 [[INDEX]], [[TMP1]]
31; CHECK-NEXT:    br i1 [[TMP6]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]
32; CHECK:       middle.block:
33; CHECK-NEXT:    br i1 true, label [[EXIT:%.*]], label [[SCALAR_PH]]
34; CHECK:       scalar.ph:
35; CHECK-NEXT:    br label [[LOOP:%.*]]
36; CHECK:       loop:
37; CHECK-NEXT:    br i1 poison, label [[EXIT]], label [[LOOP]], !llvm.loop [[LOOP3:![0-9]+]]
38; CHECK:       exit:
39; CHECK-NEXT:    ret void
40;
41entry:
42  %n4 = shl i32 %n, 2
43  br label %loop
44
45loop:
46  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %loop ]
47  %gep1 = getelementptr inbounds [2048 x i32], ptr @b, i64 0, i64 %indvars.iv
48  %load1 = load i32, ptr %gep1, align 4
49  %gep2 = getelementptr inbounds [2048 x i32], ptr @c, i64 0, i64 %indvars.iv
50  %load2 = load i32, ptr %gep2, align 4
51  %add = add nsw i32 %load2, %load1
52  %gep3 = getelementptr inbounds [2048 x i32], ptr @a, i64 0, i64 %indvars.iv
53  store i32 %add, ptr %gep3, align 4
54  %indvars.iv.next = add i64 %indvars.iv, 1
55  %lftr.wideiv = trunc i64 %indvars.iv.next to i32
56  %exitcond = icmp eq i32 %lftr.wideiv, %n4
57  br i1 %exitcond, label %exit, label %loop
58
59exit:
60  ret void
61}
62
63