test/Transforms/LoopDataPrefetch/AArch64/kryo-large-stride.ll

   1 ; RUN: opt -mcpu=kryo -mtriple=aarch64-gnu-linux -loop-data-prefetch -max-prefetch-iters-ahead=1000 -S < %s | FileCheck %s --check-prefix=LARGE_PREFETCH --check-prefix=ALL
   2 ; RUN: opt -mcpu=kryo -mtriple=aarch64-gnu-linux -loop-data-prefetch -S < %s | FileCheck %s --check-prefix=NO_LARGE_PREFETCH --check-prefix=ALL
   3 ; RUN: opt -mcpu=kryo -mtriple=aarch64-gnu-linux -passes=loop-data-prefetch -max-prefetch-iters-ahead=1000 -S < %s | FileCheck %s --check-prefix=LARGE_PREFETCH --check-prefix=ALL
   4 ; RUN: opt -mcpu=kryo -mtriple=aarch64-gnu-linux -passes=loop-data-prefetch -S < %s | FileCheck %s --check-prefix=NO_LARGE_PREFETCH --check-prefix=ALL
   5
   6 target 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-n32:64-S128"
   7
   8 ; ALL-LABEL: @small_stride(
   9 define void @small_stride(double* nocapture %a, double* nocapture readonly %b) {
  10 entry:
  11   br label %for.body
  12
  13 ; ALL: for.body:
  14 for.body:                                         ; preds = %for.body, %entry
  15   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  16   %arrayidx = getelementptr inbounds double, double* %b, i64 %indvars.iv
  17 ; ALL-NOT: call void @llvm.prefetch
  18   %0 = load double, double* %arrayidx, align 8
  19   %add = fadd double %0, 1.000000e+00
  20   %arrayidx2 = getelementptr inbounds double, double* %a, i64 %indvars.iv
  21   store double %add, double* %arrayidx2, align 8
  22   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  23   %exitcond = icmp eq i64 %indvars.iv.next, 1600
  24   br i1 %exitcond, label %for.end, label %for.body
  25
  26 ; ALL: for.end:
  27 for.end:                                          ; preds = %for.body
  28   ret void
  29 }
  30
  31 ; ALL-LABEL: @large_stride(
  32 define void @large_stride(double* nocapture %a, double* nocapture readonly %b) {
  33 entry:
  34   br label %for.body
  35
  36 ; ALL: for.body:
  37 for.body:                                         ; preds = %for.body, %entry
  38   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  39   %arrayidx = getelementptr inbounds double, double* %b, i64 %indvars.iv
  40 ; LARGE_PREFETCH: call void @llvm.prefetch
  41 ; NO_LARGE_PREFETCH-NOT: call void @llvm.prefetch
  42   %0 = load double, double* %arrayidx, align 8
  43   %add = fadd double %0, 1.000000e+00
  44   %arrayidx2 = getelementptr inbounds double, double* %a, i64 %indvars.iv
  45   store double %add, double* %arrayidx2, align 8
  46   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 150
  47   %exitcond = icmp eq i64 %indvars.iv.next, 160000
  48   br i1 %exitcond, label %for.end, label %for.body
  49
  50 ; ALL: for.end:
  51 for.end:                                          ; preds = %for.body
  52   ret void
  53 }