llvm/test/Transforms/LoopDistribute/symbolic-stride.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt -passes=loop-distribute -enable-loop-distribute -S < %s | \
   3 ; RUN:     FileCheck %s --check-prefix=DEFAULT
   4
   5 ; RUN: opt -passes=loop-distribute -enable-loop-distribute -S -enable-mem-access-versioning=0 < %s | \
   6 ; RUN:     FileCheck %s --check-prefix=NO-VERSION
   7
   8 ; If we don't speculate stride for 1 we can't distribute along the line
   9 ; because we could have a backward dependence:
  10 ;
  11 ;   for (i = 0; i < n; i++) {
  12 ;     A[i + 1] = A[i] * B[i];
  13 ;     =======================
  14 ;     C[i] = D[i] * A[stride * i];
  15 ;   }
  16
  17 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
  18 target triple = "x86_64-apple-macosx10.10.0"
  19
  20 define void @f(ptr noalias %a,
  21 ;
  22 ;
  23 ; DEFAULT-LABEL: @f(
  24 ; DEFAULT-NEXT:  entry:
  25 ; DEFAULT-NEXT:    br label [[FOR_BODY_LVER_CHECK:%.*]]
  26 ; DEFAULT:       for.body.lver.check:
  27 ; DEFAULT-NEXT:    [[IDENT_CHECK:%.*]] = icmp ne i64 [[STRIDE:%.*]], 1
  28 ; DEFAULT-NEXT:    br i1 [[IDENT_CHECK]], label [[FOR_BODY_PH_LVER_ORIG:%.*]], label [[FOR_BODY_PH_LDIST1:%.*]]
  29 ; DEFAULT:       for.body.ph.lver.orig:
  30 ; DEFAULT-NEXT:    br label [[FOR_BODY_LVER_ORIG:%.*]]
  31 ; DEFAULT:       for.body.lver.orig:
  32 ; DEFAULT-NEXT:    [[IND_LVER_ORIG:%.*]] = phi i64 [ 0, [[FOR_BODY_PH_LVER_ORIG]] ], [ [[ADD_LVER_ORIG:%.*]], [[FOR_BODY_LVER_ORIG]] ]
  33 ; DEFAULT-NEXT:    [[ARRAYIDXA_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[A:%.*]], i64 [[IND_LVER_ORIG]]
  34 ; DEFAULT-NEXT:    [[LOADA_LVER_ORIG:%.*]] = load i32, ptr [[ARRAYIDXA_LVER_ORIG]], align 4
  35 ; DEFAULT-NEXT:    [[ARRAYIDXB_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[B:%.*]], i64 [[IND_LVER_ORIG]]
  36 ; DEFAULT-NEXT:    [[LOADB_LVER_ORIG:%.*]] = load i32, ptr [[ARRAYIDXB_LVER_ORIG]], align 4
  37 ; DEFAULT-NEXT:    [[MULA_LVER_ORIG:%.*]] = mul i32 [[LOADB_LVER_ORIG]], [[LOADA_LVER_ORIG]]
  38 ; DEFAULT-NEXT:    [[ADD_LVER_ORIG]] = add nuw nsw i64 [[IND_LVER_ORIG]], 1
  39 ; DEFAULT-NEXT:    [[ARRAYIDXA_PLUS_4_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[ADD_LVER_ORIG]]
  40 ; DEFAULT-NEXT:    store i32 [[MULA_LVER_ORIG]], ptr [[ARRAYIDXA_PLUS_4_LVER_ORIG]], align 4
  41 ; DEFAULT-NEXT:    [[ARRAYIDXD_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[D:%.*]], i64 [[IND_LVER_ORIG]]
  42 ; DEFAULT-NEXT:    [[LOADD_LVER_ORIG:%.*]] = load i32, ptr [[ARRAYIDXD_LVER_ORIG]], align 4
  43 ; DEFAULT-NEXT:    [[MUL_LVER_ORIG:%.*]] = mul i64 [[IND_LVER_ORIG]], [[STRIDE]]
  44 ; DEFAULT-NEXT:    [[ARRAYIDXSTRIDEDA_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[MUL_LVER_ORIG]]
  45 ; DEFAULT-NEXT:    [[LOADSTRIDEDA_LVER_ORIG:%.*]] = load i32, ptr [[ARRAYIDXSTRIDEDA_LVER_ORIG]], align 4
  46 ; DEFAULT-NEXT:    [[MULC_LVER_ORIG:%.*]] = mul i32 [[LOADD_LVER_ORIG]], [[LOADSTRIDEDA_LVER_ORIG]]
  47 ; DEFAULT-NEXT:    [[ARRAYIDXC_LVER_ORIG:%.*]] = getelementptr inbounds i32, ptr [[C:%.*]], i64 [[IND_LVER_ORIG]]
  48 ; DEFAULT-NEXT:    store i32 [[MULC_LVER_ORIG]], ptr [[ARRAYIDXC_LVER_ORIG]], align 4
  49 ; DEFAULT-NEXT:    [[EXITCOND_LVER_ORIG:%.*]] = icmp eq i64 [[ADD_LVER_ORIG]], 20
  50 ; DEFAULT-NEXT:    br i1 [[EXITCOND_LVER_ORIG]], label [[FOR_END_LOOPEXIT:%.*]], label [[FOR_BODY_LVER_ORIG]]
  51 ; DEFAULT:       for.body.ph.ldist1:
  52 ; DEFAULT-NEXT:    br label [[FOR_BODY_LDIST1:%.*]]
  53 ; DEFAULT:       for.body.ldist1:
  54 ; DEFAULT-NEXT:    [[IND_LDIST1:%.*]] = phi i64 [ 0, [[FOR_BODY_PH_LDIST1]] ], [ [[ADD_LDIST1:%.*]], [[FOR_BODY_LDIST1]] ]
  55 ; DEFAULT-NEXT:    [[ARRAYIDXA_LDIST1:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[IND_LDIST1]]
  56 ; DEFAULT-NEXT:    [[LOADA_LDIST1:%.*]] = load i32, ptr [[ARRAYIDXA_LDIST1]], align 4
  57 ; DEFAULT-NEXT:    [[ARRAYIDXB_LDIST1:%.*]] = getelementptr inbounds i32, ptr [[B]], i64 [[IND_LDIST1]]
  58 ; DEFAULT-NEXT:    [[LOADB_LDIST1:%.*]] = load i32, ptr [[ARRAYIDXB_LDIST1]], align 4
  59 ; DEFAULT-NEXT:    [[MULA_LDIST1:%.*]] = mul i32 [[LOADB_LDIST1]], [[LOADA_LDIST1]]
  60 ; DEFAULT-NEXT:    [[ADD_LDIST1]] = add nuw nsw i64 [[IND_LDIST1]], 1
  61 ; DEFAULT-NEXT:    [[ARRAYIDXA_PLUS_4_LDIST1:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[ADD_LDIST1]]
  62 ; DEFAULT-NEXT:    store i32 [[MULA_LDIST1]], ptr [[ARRAYIDXA_PLUS_4_LDIST1]], align 4
  63 ; DEFAULT-NEXT:    [[EXITCOND_LDIST1:%.*]] = icmp eq i64 [[ADD_LDIST1]], 20
  64 ; DEFAULT-NEXT:    br i1 [[EXITCOND_LDIST1]], label [[FOR_BODY_PH:%.*]], label [[FOR_BODY_LDIST1]]
  65 ; DEFAULT:       for.body.ph:
  66 ; DEFAULT-NEXT:    br label [[FOR_BODY:%.*]]
  67 ; DEFAULT:       for.body:
  68 ; DEFAULT-NEXT:    [[IND:%.*]] = phi i64 [ 0, [[FOR_BODY_PH]] ], [ [[ADD:%.*]], [[FOR_BODY]] ]
  69 ; DEFAULT-NEXT:    [[ADD]] = add nuw nsw i64 [[IND]], 1
  70 ; DEFAULT-NEXT:    [[ARRAYIDXD:%.*]] = getelementptr inbounds i32, ptr [[D]], i64 [[IND]]
  71 ; DEFAULT-NEXT:    [[LOADD:%.*]] = load i32, ptr [[ARRAYIDXD]], align 4
  72 ; DEFAULT-NEXT:    [[MUL:%.*]] = mul i64 [[IND]], [[STRIDE]]
  73 ; DEFAULT-NEXT:    [[ARRAYIDXSTRIDEDA:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[MUL]]
  74 ; DEFAULT-NEXT:    [[LOADSTRIDEDA:%.*]] = load i32, ptr [[ARRAYIDXSTRIDEDA]], align 4
  75 ; DEFAULT-NEXT:    [[MULC:%.*]] = mul i32 [[LOADD]], [[LOADSTRIDEDA]]
  76 ; DEFAULT-NEXT:    [[ARRAYIDXC:%.*]] = getelementptr inbounds i32, ptr [[C]], i64 [[IND]]
  77 ; DEFAULT-NEXT:    store i32 [[MULC]], ptr [[ARRAYIDXC]], align 4
  78 ; DEFAULT-NEXT:    [[EXITCOND:%.*]] = icmp eq i64 [[ADD]], 20
  79 ; DEFAULT-NEXT:    br i1 [[EXITCOND]], label [[FOR_END_LOOPEXIT1:%.*]], label [[FOR_BODY]]
  80 ; DEFAULT:       for.end.loopexit:
  81 ; DEFAULT-NEXT:    br label [[FOR_END:%.*]]
  82 ; DEFAULT:       for.end.loopexit1:
  83 ; DEFAULT-NEXT:    br label [[FOR_END]]
  84 ; DEFAULT:       for.end:
  85 ; DEFAULT-NEXT:    ret void
  86 ;
  87 ; NO-VERSION-LABEL: @f(
  88 ; NO-VERSION-NEXT:  entry:
  89 ; NO-VERSION-NEXT:    br label [[FOR_BODY:%.*]]
  90 ; NO-VERSION:       for.body:
  91 ; NO-VERSION-NEXT:    [[IND:%.*]] = phi i64 [ 0, [[ENTRY:%.*]] ], [ [[ADD:%.*]], [[FOR_BODY]] ]
  92 ; NO-VERSION-NEXT:    [[ARRAYIDXA:%.*]] = getelementptr inbounds i32, ptr [[A:%.*]], i64 [[IND]]
  93 ; NO-VERSION-NEXT:    [[LOADA:%.*]] = load i32, ptr [[ARRAYIDXA]], align 4
  94 ; NO-VERSION-NEXT:    [[ARRAYIDXB:%.*]] = getelementptr inbounds i32, ptr [[B:%.*]], i64 [[IND]]
  95 ; NO-VERSION-NEXT:    [[LOADB:%.*]] = load i32, ptr [[ARRAYIDXB]], align 4
  96 ; NO-VERSION-NEXT:    [[MULA:%.*]] = mul i32 [[LOADB]], [[LOADA]]
  97 ; NO-VERSION-NEXT:    [[ADD]] = add nuw nsw i64 [[IND]], 1
  98 ; NO-VERSION-NEXT:    [[ARRAYIDXA_PLUS_4:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[ADD]]
  99 ; NO-VERSION-NEXT:    store i32 [[MULA]], ptr [[ARRAYIDXA_PLUS_4]], align 4
 100 ; NO-VERSION-NEXT:    [[ARRAYIDXD:%.*]] = getelementptr inbounds i32, ptr [[D:%.*]], i64 [[IND]]
 101 ; NO-VERSION-NEXT:    [[LOADD:%.*]] = load i32, ptr [[ARRAYIDXD]], align 4
 102 ; NO-VERSION-NEXT:    [[MUL:%.*]] = mul i64 [[IND]], [[STRIDE:%.*]]
 103 ; NO-VERSION-NEXT:    [[ARRAYIDXSTRIDEDA:%.*]] = getelementptr inbounds i32, ptr [[A]], i64 [[MUL]]
 104 ; NO-VERSION-NEXT:    [[LOADSTRIDEDA:%.*]] = load i32, ptr [[ARRAYIDXSTRIDEDA]], align 4
 105 ; NO-VERSION-NEXT:    [[MULC:%.*]] = mul i32 [[LOADD]], [[LOADSTRIDEDA]]
 106 ; NO-VERSION-NEXT:    [[ARRAYIDXC:%.*]] = getelementptr inbounds i32, ptr [[C:%.*]], i64 [[IND]]
 107 ; NO-VERSION-NEXT:    store i32 [[MULC]], ptr [[ARRAYIDXC]], align 4
 108 ; NO-VERSION-NEXT:    [[EXITCOND:%.*]] = icmp eq i64 [[ADD]], 20
 109 ; NO-VERSION-NEXT:    br i1 [[EXITCOND]], label [[FOR_END:%.*]], label [[FOR_BODY]]
 110 ; NO-VERSION:       for.end:
 111 ; NO-VERSION-NEXT:    ret void
 112 ;
 113   i32* noalias %b,
 114   ptr noalias %c,
 115   ptr noalias %d,
 116   i64 %stride) {
 117 entry:
 118   br label %for.body
 119
 120
 121
 122 for.body:                                         ; preds = %for.body, %entry
 123   %ind = phi i64 [ 0, %entry ], [ %add, %for.body ]
 124
 125   %arrayidxA = getelementptr inbounds i32, ptr %a, i64 %ind
 126   %loadA = load i32, ptr %arrayidxA, align 4
 127
 128   %arrayidxB = getelementptr inbounds i32, ptr %b, i64 %ind
 129   %loadB = load i32, ptr %arrayidxB, align 4
 130
 131   %mulA = mul i32 %loadB, %loadA
 132
 133   %add = add nuw nsw i64 %ind, 1
 134   %arrayidxA_plus_4 = getelementptr inbounds i32, ptr %a, i64 %add
 135   store i32 %mulA, ptr %arrayidxA_plus_4, align 4
 136
 137   %arrayidxD = getelementptr inbounds i32, ptr %d, i64 %ind
 138   %loadD = load i32, ptr %arrayidxD, align 4
 139
 140   %mul = mul i64 %ind, %stride
 141   %arrayidxStridedA = getelementptr inbounds i32, ptr %a, i64 %mul
 142   %loadStridedA = load i32, ptr %arrayidxStridedA, align 4
 143
 144   %mulC = mul i32 %loadD, %loadStridedA
 145
 146   %arrayidxC = getelementptr inbounds i32, ptr %c, i64 %ind
 147   store i32 %mulC, ptr %arrayidxC, align 4
 148
 149   %exitcond = icmp eq i64 %add, 20
 150   br i1 %exitcond, label %for.end, label %for.body
 151
 152 for.end:                                          ; preds = %for.body
 153   ret void
 154 }