test/Transforms/LoopVectorize/X86/invariant-load-gather.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -loop-vectorize -S -mattr=avx512f -instcombine < %s | FileCheck %s
   3
   4 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
   5 target triple = "x86_64-unknown-linux-gnu"
   6
   7 define i32 @inv_load_conditional(i32* %a, i64 %n, i32* %b, i32 %k) {
   8 ; CHECK-LABEL: @inv_load_conditional(
   9 ; CHECK-NEXT:  entry:
  10 ; CHECK-NEXT:    [[NTRUNC:%.*]] = trunc i64 [[N:%.*]] to i32
  11 ; CHECK-NEXT:    [[TMP0:%.*]] = icmp sgt i64 [[N]], 1
  12 ; CHECK-NEXT:    [[SMAX:%.*]] = select i1 [[TMP0]], i64 [[N]], i64 1
  13 ; CHECK-NEXT:    [[MIN_ITERS_CHECK:%.*]] = icmp ult i64 [[SMAX]], 16
  14 ; CHECK-NEXT:    br i1 [[MIN_ITERS_CHECK]], label [[SCALAR_PH:%.*]], label [[VECTOR_MEMCHECK:%.*]]
  15 ; CHECK:       vector.memcheck:
  16 ; CHECK-NEXT:    [[A4:%.*]] = bitcast i32* [[A:%.*]] to i8*
  17 ; CHECK-NEXT:    [[B1:%.*]] = bitcast i32* [[B:%.*]] to i8*
  18 ; CHECK-NEXT:    [[TMP1:%.*]] = icmp sgt i64 [[N]], 1
  19 ; CHECK-NEXT:    [[SMAX2:%.*]] = select i1 [[TMP1]], i64 [[N]], i64 1
  20 ; CHECK-NEXT:    [[SCEVGEP:%.*]] = getelementptr i32, i32* [[B]], i64 [[SMAX2]]
  21 ; CHECK-NEXT:    [[UGLYGEP:%.*]] = getelementptr i8, i8* [[A4]], i64 1
  22 ; CHECK-NEXT:    [[BOUND0:%.*]] = icmp ugt i8* [[UGLYGEP]], [[B1]]
  23 ; CHECK-NEXT:    [[BOUND1:%.*]] = icmp ugt i32* [[SCEVGEP]], [[A]]
  24 ; CHECK-NEXT:    [[FOUND_CONFLICT:%.*]] = and i1 [[BOUND0]], [[BOUND1]]
  25 ; CHECK-NEXT:    br i1 [[FOUND_CONFLICT]], label [[SCALAR_PH]], label [[VECTOR_PH:%.*]]
  26 ; CHECK:       vector.ph:
  27 ; CHECK-NEXT:    [[N_VEC:%.*]] = and i64 [[SMAX]], 9223372036854775792
  28 ; CHECK-NEXT:    [[BROADCAST_SPLATINSERT5:%.*]] = insertelement <16 x i32*> undef, i32* [[A]], i32 0
  29 ; CHECK-NEXT:    [[BROADCAST_SPLAT6:%.*]] = shufflevector <16 x i32*> [[BROADCAST_SPLATINSERT5]], <16 x i32*> undef, <16 x i32> zeroinitializer
  30 ; CHECK-NEXT:    [[BROADCAST_SPLATINSERT7:%.*]] = insertelement <16 x i32> undef, i32 [[NTRUNC]], i32 0
  31 ; CHECK-NEXT:    [[BROADCAST_SPLAT8:%.*]] = shufflevector <16 x i32> [[BROADCAST_SPLATINSERT7]], <16 x i32> undef, <16 x i32> zeroinitializer
  32 ; CHECK-NEXT:    br label [[VECTOR_BODY:%.*]]
  33 ; CHECK:       vector.body:
  34 ; CHECK-NEXT:    [[INDEX:%.*]] = phi i64 [ 0, [[VECTOR_PH]] ], [ [[INDEX_NEXT:%.*]], [[VECTOR_BODY]] ]
  35 ; CHECK-NEXT:    [[TMP2:%.*]] = getelementptr inbounds i32, i32* [[B]], i64 [[INDEX]]
  36 ; CHECK-NEXT:    [[TMP3:%.*]] = icmp ne <16 x i32*> [[BROADCAST_SPLAT6]], zeroinitializer
  37 ; CHECK-NEXT:    [[TMP4:%.*]] = bitcast i32* [[TMP2]] to <16 x i32>*
  38 ; CHECK-NEXT:    store <16 x i32> [[BROADCAST_SPLAT8]], <16 x i32>* [[TMP4]], align 4, !alias.scope !0, !noalias !3
  39 ; CHECK-NEXT:    [[INDEX_NEXT]] = add i64 [[INDEX]], 16
  40 ; CHECK-NEXT:    [[TMP5:%.*]] = icmp eq i64 [[INDEX_NEXT]], [[N_VEC]]
  41 ; CHECK-NEXT:    br i1 [[TMP5]], label [[MIDDLE_BLOCK:%.*]], label [[VECTOR_BODY]], !llvm.loop !5
  42 ; CHECK:       middle.block:
  43 ; CHECK-NEXT:    [[WIDE_MASKED_GATHER:%.*]] = call <16 x i32> @llvm.masked.gather.v16i32.v16p0i32(<16 x i32*> [[BROADCAST_SPLAT6]], i32 4, <16 x i1> [[TMP3]], <16 x i32> undef), !alias.scope !3
  44 ; CHECK-NEXT:    [[PREDPHI:%.*]] = select <16 x i1> [[TMP3]], <16 x i32> [[WIDE_MASKED_GATHER]], <16 x i32> <i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 undef, i32 1>
  45 ; CHECK-NEXT:    [[CMP_N:%.*]] = icmp eq i64 [[SMAX]], [[N_VEC]]
  46 ; CHECK-NEXT:    [[TMP6:%.*]] = extractelement <16 x i32> [[PREDPHI]], i32 15
  47 ; CHECK-NEXT:    br i1 [[CMP_N]], label [[FOR_END:%.*]], label [[SCALAR_PH]]
  48 ; CHECK:       scalar.ph:
  49 ; CHECK-NEXT:    [[BC_RESUME_VAL:%.*]] = phi i64 [ [[N_VEC]], [[MIDDLE_BLOCK]] ], [ 0, [[ENTRY:%.*]] ], [ 0, [[VECTOR_MEMCHECK]] ]
  50 ; CHECK-NEXT:    br label [[FOR_BODY:%.*]]
  51 ; CHECK:       for.body:
  52 ; CHECK-NEXT:    [[I:%.*]] = phi i64 [ [[I_NEXT:%.*]], [[LATCH:%.*]] ], [ [[BC_RESUME_VAL]], [[SCALAR_PH]] ]
  53 ; CHECK-NEXT:    [[TMP1:%.*]] = getelementptr inbounds i32, i32* [[B]], i64 [[I]]
  54 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i32* [[A]], null
  55 ; CHECK-NEXT:    store i32 [[NTRUNC]], i32* [[TMP1]], align 4
  56 ; CHECK-NEXT:    br i1 [[CMP]], label [[LATCH]], label [[COND_LOAD:%.*]]
  57 ; CHECK:       cond_load:
  58 ; CHECK-NEXT:    [[ALOAD:%.*]] = load i32, i32* [[A]], align 4
  59 ; CHECK-NEXT:    br label [[LATCH]]
  60 ; CHECK:       latch:
  61 ; CHECK-NEXT:    [[A_LCSSA:%.*]] = phi i32 [ [[ALOAD]], [[COND_LOAD]] ], [ 1, [[FOR_BODY]] ]
  62 ; CHECK-NEXT:    [[I_NEXT]] = add nuw nsw i64 [[I]], 1
  63 ; CHECK-NEXT:    [[COND:%.*]] = icmp slt i64 [[I_NEXT]], [[N]]
  64 ; CHECK-NEXT:    br i1 [[COND]], label [[FOR_BODY]], label [[FOR_END]], !llvm.loop !7
  65 ; CHECK:       for.end:
  66 ; CHECK-NEXT:    [[A_LCSSA_LCSSA:%.*]] = phi i32 [ [[A_LCSSA]], [[LATCH]] ], [ [[TMP6]], [[MIDDLE_BLOCK]] ]
  67 ; CHECK-NEXT:    ret i32 [[A_LCSSA_LCSSA]]
  68 ;
  69 entry:
  70   %ntrunc = trunc i64 %n to i32
  71   br label %for.body
  72
  73 for.body:                                         ; preds = %for.body, %entry
  74   %i = phi i64 [ %i.next, %latch ], [ 0, %entry ]
  75   %tmp1 = getelementptr inbounds i32, i32* %b, i64 %i
  76   %tmp2 = load i32, i32* %tmp1, align 8
  77   %cmp = icmp ne i32* %a, null
  78   store i32 %ntrunc, i32* %tmp1
  79   br i1 %cmp, label %cond_load, label %latch
  80
  81 cond_load:
  82   %aload = load i32, i32* %a, align 4
  83   br label %latch
  84
  85 latch:
  86   %a.lcssa = phi i32 [ %aload, %cond_load ], [ 1, %for.body ]
  87   %i.next = add nuw nsw i64 %i, 1
  88   %cond = icmp slt i64 %i.next, %n
  89   br i1 %cond, label %for.body, label %for.end
  90
  91 for.end:                                          ; preds = %for.body
  92   ret i32 %a.lcssa
  93 }