test/Transforms/LoopDistribute/followup.ll

   1 ; RUN: opt -basicaa -loop-distribute -S < %s | FileCheck %s
   2 ;
   3 ; Check that followup loop-attributes are applied to the loops after
   4 ; loop distribution.
   5 ;
   6 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
   7
   8 define void @f(i32* %a, i32* %b, i32* %c, i32* %d, i32* %e) {
   9 entry:
  10   br label %for.body
  11
  12 for.body:
  13   %ind = phi i64 [ 0, %entry ], [ %add, %for.body ]
  14
  15   %arrayidxA = getelementptr inbounds i32, i32* %a, i64 %ind
  16   %loadA = load i32, i32* %arrayidxA, align 4
  17
  18   %arrayidxB = getelementptr inbounds i32, i32* %b, i64 %ind
  19   %loadB = load i32, i32* %arrayidxB, align 4
  20
  21   %mulA = mul i32 %loadB, %loadA
  22
  23   %add = add nuw nsw i64 %ind, 1
  24   %arrayidxA_plus_4 = getelementptr inbounds i32, i32* %a, i64 %add
  25   store i32 %mulA, i32* %arrayidxA_plus_4, align 4
  26
  27   %arrayidxD = getelementptr inbounds i32, i32* %d, i64 %ind
  28   %loadD = load i32, i32* %arrayidxD, align 4
  29
  30   %arrayidxE = getelementptr inbounds i32, i32* %e, i64 %ind
  31   %loadE = load i32, i32* %arrayidxE, align 4
  32
  33   %mulC = mul i32 %loadD, %loadE
  34
  35   %arrayidxC = getelementptr inbounds i32, i32* %c, i64 %ind
  36   store i32 %mulC, i32* %arrayidxC, align 4
  37
  38   %exitcond = icmp eq i64 %add, 20
  39   br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !0
  40
  41 for.end:
  42   ret void
  43 }
  44
  45 !0 = distinct !{!0, !1, !2, !3, !4, !5}
  46 !1 = !{!"llvm.loop.distribute.enable", i1 true}
  47 !2 = !{!"llvm.loop.distribute.followup_all", !{!"FollowupAll"}}
  48 !3 = !{!"llvm.loop.distribute.followup_coincident", !{!"FollowupCoincident", i1 false}}
  49 !4 = !{!"llvm.loop.distribute.followup_sequential", !{!"FollowupSequential", i32 8}}
  50 !5 = !{!"llvm.loop.distribute.followup_fallback", !{!"FollowupFallback"}}
  51
  52
  53 ; CHECK-LABEL: for.body.lver.orig:
  54 ; CHECK: br i1 %exitcond.lver.orig, label %for.end, label %for.body.lver.orig, !llvm.loop ![[LOOP_ORIG:[0-9]+]]
  55 ; CHECK-LABEL: for.body.ldist1:
  56 ; CHECK: br i1 %exitcond.ldist1, label %for.body.ph, label %for.body.ldist1, !llvm.loop ![[LOOP_SEQUENTIAL:[0-9]+]]
  57 ; CHECK-LABEL: for.body:
  58 ; CHECK: br i1 %exitcond, label %for.end, label %for.body, !llvm.loop ![[LOOP_COINCIDENT:[0-9]+]]
  59
  60 ; CHECK: ![[LOOP_ORIG]] = distinct !{![[LOOP_ORIG]], ![[FOLLOWUP_ALL:[0-9]+]], ![[FOLLOUP_FALLBACK:[0-9]+]]}
  61 ; CHECK: ![[FOLLOWUP_ALL]] = !{!"FollowupAll"}
  62 ; CHECK: ![[FOLLOUP_FALLBACK]] = !{!"FollowupFallback"}
  63 ; CHECK: ![[LOOP_SEQUENTIAL]] = distinct !{![[LOOP_SEQUENTIAL]], ![[FOLLOWUP_ALL]], ![[FOLLOWUP_SEQUENTIAL:[0-9]+]]}
  64 ; CHECK: ![[FOLLOWUP_SEQUENTIAL]] = !{!"FollowupSequential", i32 8}
  65 ; CHECK: ![[LOOP_COINCIDENT]] = distinct !{![[LOOP_COINCIDENT]], ![[FOLLOWUP_ALL]], ![[FOLLOWUP_COINCIDENT:[0-9]+]]}
  66 ; CHECK: ![[FOLLOWUP_COINCIDENT]] = !{!"FollowupCoincident", i1 false}