llvm/test/Transforms/LoopIdiom/memcpy-vectors.ll

   1 ; RUN: opt -loop-idiom -S <%s | FileCheck %s
   2
   3 define void @memcpy_fixed_vec(i64* noalias %a, i64* noalias %b) local_unnamed_addr #1 {
   4 ; CHECK-LABEL: @memcpy_fixed_vec(
   5 ; CHECK: entry:
   6 ; CHECK: memcpy
   7 ; CHECK: vector.body
   8 entry:
   9   br label %vector.body
  10
  11 vector.body:                                      ; preds = %vector.body, %entry
  12   %index = phi i64 [ 0, %entry ], [ %index.next, %vector.body ]
  13   %0 = getelementptr inbounds i64, i64* %a, i64 %index
  14   %1 = bitcast i64* %0 to <2 x i64>*
  15   %wide.load = load <2 x i64>, <2 x i64>* %1, align 8
  16   %2 = getelementptr inbounds i64, i64* %b, i64 %index
  17   %3 = bitcast i64* %2 to <2 x i64>*
  18   store <2 x i64> %wide.load, <2 x i64>* %3, align 8
  19   %index.next = add nuw nsw i64 %index, 2
  20   %4 = icmp eq i64 %index.next, 1024
  21   br i1 %4, label %for.cond.cleanup, label %vector.body
  22
  23 for.cond.cleanup:                                 ; preds = %vector.body
  24   ret void
  25 }
  26
  27 define void @memcpy_scalable_vec(i64* noalias %a, i64* noalias %b) local_unnamed_addr #1 {
  28 ; CHECK-LABEL: @memcpy_scalable_vec(
  29 ; CHECK: entry:
  30 ; CHECK-NOT: memcpy
  31 ; CHECK: vector.body
  32 entry:
  33   br label %vector.body
  34
  35 vector.body:                                      ; preds = %vector.body, %entry
  36   %index = phi i64 [ 0, %entry ], [ %index.next, %vector.body ]
  37   %0 = bitcast i64* %a to <vscale x 2 x i64>*
  38   %1 = getelementptr inbounds <vscale x 2 x i64>, <vscale x 2 x i64>* %0, i64 %index
  39   %wide.load = load <vscale x 2 x i64>, <vscale x 2 x i64>* %1, align 16
  40   %2 = bitcast i64* %b to <vscale x 2 x i64>*
  41   %3 = getelementptr inbounds <vscale x 2 x i64>, <vscale x 2 x i64>* %2, i64 %index
  42   store <vscale x 2 x i64> %wide.load, <vscale x 2 x i64>* %3, align 16
  43   %index.next = add nuw nsw i64 %index, 1
  44   %4 = icmp eq i64 %index.next, 1024
  45   br i1 %4, label %for.cond.cleanup, label %vector.body
  46
  47 for.cond.cleanup:                                 ; preds = %vector.body
  48   ret void
  49 }