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