llvm/test/Transforms/LoopVectorize/widen-gep-all-indices-invariant.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -passes=loop-vectorize -force-vector-width=4 -force-vector-interleave=1 -S %s | FileCheck %s
   3
   4 define void @pr63340(ptr %A, ptr %B) {
   5 ; CHECK-LABEL: define void @pr63340
   6 ; CHECK-SAME: (ptr [[A:%.*]], ptr [[B:%.*]]) {
   7 ; CHECK-NEXT:  entry:
   8 ; CHECK-NEXT:    br i1 false, label [[SCALAR_PH:%.*]], label [[VECTOR_PH:%.*]]
   9 ; CHECK:       vector.ph:
  10 ; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr i8, ptr [[A]], i64 1
  11 ; CHECK-NEXT:    [[DOTSPLATINSERT:%.*]] = insertelement <4 x ptr> poison, ptr [[TMP1]], i64 0
  12 ; CHECK-NEXT:    [[DOTSPLAT:%.*]] = shufflevector <4 x ptr> [[DOTSPLATINSERT]], <4 x ptr> poison, <4 x i32> zeroinitializer
  13 ; CHECK-NEXT:    br label [[VECTOR_BODY:%.*]]
  14 ; CHECK:       vector.body:
  15 ; CHECK-NEXT:    [[INDEX:%.*]] = phi i32 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ]
  16 ; CHECK-NEXT:    [[OFFSET_IDX:%.*]] = trunc i32 [[INDEX]] to i8
  17 ; CHECK-NEXT:    [[TMP0:%.*]] = add i8 [[OFFSET_IDX]], 0
  18 ; CHECK-NEXT:    [[TMP2:%.*]] = getelementptr inbounds ptr, ptr [[B]], i8 [[TMP0]]
  19 ; CHECK-NEXT:    [[TMP3:%.*]] = getelementptr inbounds ptr, ptr [[TMP2]], i32 0
  20 ; CHECK-NEXT:    store <4 x ptr> [[DOTSPLAT]], ptr [[TMP3]], align 8
  21 ; CHECK-NEXT:    [[INDEX_NEXT]] = add nuw i32 [[INDEX]], 4
  22 ; CHECK-NEXT:    [[TMP4:%.*]] = icmp eq i32 [[INDEX_NEXT]], 128
  23 ; CHECK-NEXT:    br i1 [[TMP4]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop [[LOOP0:![0-9]+]]
  24 ; CHECK:       middle.block:
  25 ; CHECK-NEXT:    br i1 true, label [[EXIT:%.*]], label [[SCALAR_PH]]
  26 ; CHECK:       scalar.ph:
  27 ; CHECK-NEXT:    [[BC_RESUME_VAL:%.*]] = phi i8 [ -128, [[MIDDLE_BLOCK]] ], [ 0, [[ENTRY:%.*]] ]
  28 ; CHECK-NEXT:    br label [[LOOP_HEADER:%.*]]
  29 ; CHECK:       loop.header:
  30 ; CHECK-NEXT:    [[IV:%.*]] = phi i8 [ [[BC_RESUME_VAL]], [[SCALAR_PH]] ], [ [[IV_NEXT:%.*]], [[LOOP_LATCH:%.*]] ]
  31 ; CHECK-NEXT:    br label [[LOOP_LATCH]]
  32 ; CHECK:       loop.latch:
  33 ; CHECK-NEXT:    [[F_0_I:%.*]] = phi ptr [ [[A]], [[LOOP_HEADER]] ]
  34 ; CHECK-NEXT:    [[GEP:%.*]] = getelementptr i8, ptr [[F_0_I]], i64 1
  35 ; CHECK-NEXT:    [[GEP_B:%.*]] = getelementptr inbounds ptr, ptr [[B]], i8 [[IV]]
  36 ; CHECK-NEXT:    store ptr [[GEP]], ptr [[GEP_B]], align 8
  37 ; CHECK-NEXT:    [[IV_NEXT]] = add i8 [[IV]], 1
  38 ; CHECK-NEXT:    [[EC:%.*]] = icmp eq i8 [[IV_NEXT]], -128
  39 ; CHECK-NEXT:    br i1 [[EC]], label [[EXIT]], label [[LOOP_HEADER]], !llvm.loop [[LOOP3:![0-9]+]]
  40 ; CHECK:       exit:
  41 ; CHECK-NEXT:    ret void
  42 ;
  43 entry:
  44   br label %loop.header
  45
  46 loop.header:
  47   %iv = phi i8 [ 0, %entry ], [ %iv.next, %loop.latch ]
  48   br label %loop.latch
  49
  50 loop.latch:
  51   %f.0.i = phi ptr [ %A, %loop.header ]
  52   %gep = getelementptr i8, ptr %f.0.i, i64 1
  53   %gep.B = getelementptr inbounds ptr, ptr %B, i8 %iv
  54   store ptr %gep, ptr %gep.B
  55   %iv.next = add i8 %iv, 1
  56   %ec = icmp eq i8 %iv.next, 128
  57   br i1 %ec, label %exit, label %loop.header
  58
  59 exit:
  60   ret void
  61 }