xref: /llvm-project/llvm/test/Transforms/LoopVectorize/multiple-address-spaces.ll (revision 23a239267e8a1d20ed10d3545feaf2a2bb70b085)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
2; RUN: opt < %s -passes=loop-vectorize,dce,instcombine -force-vector-interleave=1 -force-vector-width=4 -S | FileCheck %s
3
4; From a simple program with two address spaces:
5; char Y[4*10000] __attribute__((address_space(1)));
6; char X[4*10000];
7; int main() {
8;    for (int i = 0; i < 4*10000; ++i)
9;        X[i] = Y[i] + 1;
10;    return 0;
11;}
12
13target 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"
14
15@Y = common addrspace(1) global [40000 x i8] zeroinitializer, align 16
16@X = common global [40000 x i8] zeroinitializer, align 16
17
18define i32 @main() #0 {
19; CHECK-LABEL: define i32 @main
20; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
21; CHECK-NEXT:  entry:
22; CHECK-NEXT:    br i1 false, label [[SCALAR_PH:%.*]], label [[VECTOR_PH:%.*]]
23; CHECK:       vector.ph:
24; CHECK-NEXT:    br label [[VECTOR_BODY:%.*]]
25; CHECK:       vector.body:
26; CHECK-NEXT:    [[INDEX:%.*]] = phi i64 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ]
27; CHECK-NEXT:    [[TMP0:%.*]] = getelementptr inbounds [40000 x i8], ptr addrspace(1) @Y, i64 0, i64 [[INDEX]]
28; CHECK-NEXT:    [[WIDE_LOAD:%.*]] = load <4 x i8>, ptr addrspace(1) [[TMP0]], align 1
29; CHECK-NEXT:    [[TMP1:%.*]] = add <4 x i8> [[WIDE_LOAD]], splat (i8 1)
30; CHECK-NEXT:    [[TMP2:%.*]] = getelementptr inbounds [40000 x i8], ptr @X, i64 0, i64 [[INDEX]]
31; CHECK-NEXT:    store <4 x i8> [[TMP1]], ptr [[TMP2]], align 1
32; CHECK-NEXT:    [[INDEX_NEXT]] = add nuw i64 [[INDEX]], 4
33; CHECK-NEXT:    [[TMP3:%.*]] = icmp eq i64 [[INDEX_NEXT]], 40000
34; CHECK-NEXT:    br i1 [[TMP3]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]
35; CHECK:       middle.block:
36; CHECK-NEXT:    br i1 true, label [[FOR_END:%.*]], label [[SCALAR_PH]]
37; CHECK:       scalar.ph:
38; CHECK-NEXT:    br label [[FOR_BODY:%.*]]
39; CHECK:       for.body:
40; CHECK-NEXT:    br i1 poison, label [[FOR_END]], label [[FOR_BODY]], !llvm.loop [[LOOP3:![0-9]+]]
41; CHECK:       for.end:
42; CHECK-NEXT:    ret i32 0
43;
44entry:
45  br label %for.body
46
47for.body:                                         ; preds = %for.body, %entry
48  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
49  %arrayidx = getelementptr inbounds [40000 x i8], ptr addrspace(1) @Y, i64 0, i64 %indvars.iv
50  %0 = load i8, ptr addrspace(1) %arrayidx, align 1
51  %add = add i8 %0, 1
52  %arrayidx3 = getelementptr inbounds [40000 x i8], ptr @X, i64 0, i64 %indvars.iv
53  store i8 %add, ptr %arrayidx3, align 1
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, 40000
57  br i1 %exitcond, label %for.end, label %for.body
58
59for.end:                                          ; preds = %for.body
60  ret i32 0
61}
62
63attributes #0 = { nounwind uwtable "less-precise-fpmad"="false" "frame-pointer"="none" "no-infs-fp-math"="false" "no-nans-fp-math"="false" "unsafe-fp-math"="false" "use-soft-float"="false" }
64