xref: /llvm-project/llvm/test/Transforms/LoopDataPrefetch/LoongArch/basic.ll (revision ca4886bf96f0b6dcc151c03bd8c7df414f3f659b)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
2; RUN: opt --mtriple=loongarch64 -mattr=+d --passes=loop-data-prefetch -S < %s | FileCheck %s
3
4define void @foo(ptr %a, ptr %b) {
5; CHECK-LABEL: define void @foo(
6; CHECK-SAME: ptr [[A:%.*]], ptr [[B:%.*]]) #[[ATTR0:[0-9]+]] {
7; CHECK-NEXT:  [[ENTRY:.*]]:
8; CHECK-NEXT:    br label %[[FOR_BODY:.*]]
9; CHECK:       [[FOR_BODY]]:
10; CHECK-NEXT:    [[INDVARS_IV:%.*]] = phi i64 [ 0, %[[ENTRY]] ], [ [[INDVARS_IV_NEXT:%.*]], %[[FOR_BODY]] ]
11; CHECK-NEXT:    [[TMP0:%.*]] = shl nuw nsw i64 [[INDVARS_IV]], 3
12; CHECK-NEXT:    [[TMP1:%.*]] = add i64 [[TMP0]], 200
13; CHECK-NEXT:    [[SCEVGEP1:%.*]] = getelementptr i8, ptr [[A]], i64 [[TMP1]]
14; CHECK-NEXT:    [[TMP2:%.*]] = shl nuw nsw i64 [[INDVARS_IV]], 3
15; CHECK-NEXT:    [[TMP3:%.*]] = add i64 [[TMP2]], 200
16; CHECK-NEXT:    [[SCEVGEP:%.*]] = getelementptr i8, ptr [[B]], i64 [[TMP3]]
17; CHECK-NEXT:    [[ARRAYIDX:%.*]] = getelementptr inbounds double, ptr [[B]], i64 [[INDVARS_IV]]
18; CHECK-NEXT:    call void @llvm.prefetch.p0(ptr [[SCEVGEP]], i32 0, i32 3, i32 1)
19; CHECK-NEXT:    [[TMP4:%.*]] = load double, ptr [[ARRAYIDX]], align 8
20; CHECK-NEXT:    [[ADD:%.*]] = fadd double [[TMP4]], 1.000000e+00
21; CHECK-NEXT:    [[ARRAYIDX2:%.*]] = getelementptr inbounds double, ptr [[A]], i64 [[INDVARS_IV]]
22; CHECK-NEXT:    call void @llvm.prefetch.p0(ptr [[SCEVGEP1]], i32 1, i32 3, i32 1)
23; CHECK-NEXT:    store double [[ADD]], ptr [[ARRAYIDX2]], align 8
24; CHECK-NEXT:    [[INDVARS_IV_NEXT]] = add nuw nsw i64 [[INDVARS_IV]], 1
25; CHECK-NEXT:    [[EXITCOND:%.*]] = icmp eq i64 [[INDVARS_IV_NEXT]], 1600
26; CHECK-NEXT:    br i1 [[EXITCOND]], label %[[FOR_END:.*]], label %[[FOR_BODY]]
27; CHECK:       [[FOR_END]]:
28; CHECK-NEXT:    ret void
29;
30entry:
31  br label %for.body
32
33for.body:                                         ; preds = %for.body, %entry
34  %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
35  %arrayidx = getelementptr inbounds double, ptr %b, i64 %indvars.iv
36  %0 = load double, ptr %arrayidx, align 8
37  %add = fadd double %0, 1.000000e+00
38  %arrayidx2 = getelementptr inbounds double, ptr %a, i64 %indvars.iv
39  store double %add, ptr %arrayidx2, align 8
40  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
41  %exitcond = icmp eq i64 %indvars.iv.next, 1600
42  br i1 %exitcond, label %for.end, label %for.body
43
44for.end:                                          ; preds = %for.body
45  ret void
46}
47