llvm/test/Transforms/LoopDataPrefetch/RISCV/basic.ll

   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
   5 define void @foo(ptr nocapture %a, ptr nocapture readonly %b) {
   6 ; CHECK-LABEL: define void @foo(
   7 ; CHECK-SAME: ptr nocapture [[A:%.*]], ptr nocapture readonly [[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 ;
  27 entry:
  28   br label %for.body
  29
  30 for.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
  41 for.end:                                          ; preds = %for.body
  42   ret void
  43 }