test/Transforms/LoopUnroll/unroll-pragmas-disabled.ll

   1 ; RUN: opt < %s -loop-unroll -S | FileCheck %s
   2 ;
   3 ; Verify that the unrolling pass removes existing unroll count metadata
   4 ; and adds a disable unrolling node after unrolling is complete.
   5
   6 target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
   7 target triple = "x86_64-unknown-linux-gnu"
   8
   9 ; #pragma clang loop  vectorize(enable) unroll_count(4) vectorize_width(8)
  10 ;
  11 ; Unroll count metadata should be replaced with unroll(disable).  Vectorize
  12 ; metadata should be untouched.
  13 ;
  14 ; CHECK-LABEL: @loop1(
  15 ; CHECK: br i1 {{.*}}, label {{.*}}, label {{.*}}, !llvm.loop ![[LOOP_1:.*]]
  16 define void @loop1(i32* nocapture %a) {
  17 entry:
  18   br label %for.body
  19
  20 for.body:                                         ; preds = %for.body, %entry
  21   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  22   %arrayidx = getelementptr inbounds i32* %a, i64 %indvars.iv
  23   %0 = load i32* %arrayidx, align 4
  24   %inc = add nsw i32 %0, 1
  25   store i32 %inc, i32* %arrayidx, align 4
  26   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  27   %exitcond = icmp eq i64 %indvars.iv.next, 64
  28   br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !1
  29
  30 for.end:                                          ; preds = %for.body
  31   ret void
  32 }
  33 !1 = metadata !{metadata !1, metadata !2, metadata !3, metadata !4}
  34 !2 = metadata !{metadata !"llvm.loop.vectorize.enable", i1 true}
  35 !3 = metadata !{metadata !"llvm.loop.unroll.count", i32 4}
  36 !4 = metadata !{metadata !"llvm.loop.vectorize.width", i32 8}
  37
  38 ; #pragma clang loop unroll(full)
  39 ;
  40 ; An unroll disable metadata node is only added for the unroll count case.
  41 ; In this case, the loop has a full unroll metadata but can't be fully unrolled
  42 ; because the trip count is dynamic.  The full unroll metadata should remain
  43 ; after unrolling.
  44 ;
  45 ; CHECK-LABEL: @loop2(
  46 ; CHECK: br i1 {{.*}}, label {{.*}}, label {{.*}}, !llvm.loop ![[LOOP_2:.*]]
  47 define void @loop2(i32* nocapture %a, i32 %b) {
  48 entry:
  49   %cmp3 = icmp sgt i32 %b, 0
  50   br i1 %cmp3, label %for.body, label %for.end, !llvm.loop !5
  51
  52 for.body:                                         ; preds = %entry, %for.body
  53   %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
  54   %arrayidx = getelementptr inbounds i32* %a, i64 %indvars.iv
  55   %0 = load i32* %arrayidx, align 4
  56   %inc = add nsw i32 %0, 1
  57   store i32 %inc, i32* %arrayidx, align 4
  58   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  59   %lftr.wideiv = trunc i64 %indvars.iv.next to i32
  60   %exitcond = icmp eq i32 %lftr.wideiv, %b
  61   br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !5
  62
  63 for.end:                                          ; preds = %for.body, %entry
  64   ret void
  65 }
  66 !5 = metadata !{metadata !5, metadata !6}
  67 !6 = metadata !{metadata !"llvm.loop.unroll.full"}
  68
  69 ; #pragma clang loop unroll(disable)
  70 ;
  71 ; Unroll metadata should not change.
  72 ;
  73 ; CHECK-LABEL: @loop3(
  74 ; CHECK: br i1 {{.*}}, label {{.*}}, label {{.*}}, !llvm.loop ![[LOOP_3:.*]]
  75 define void @loop3(i32* nocapture %a) {
  76 entry:
  77   br label %for.body
  78
  79 for.body:                                         ; preds = %for.body, %entry
  80   %indvars.iv = phi i64 [ 0, %entry ], [ %indvars.iv.next, %for.body ]
  81   %arrayidx = getelementptr inbounds i32* %a, i64 %indvars.iv
  82   %0 = load i32* %arrayidx, align 4
  83   %inc = add nsw i32 %0, 1
  84   store i32 %inc, i32* %arrayidx, align 4
  85   %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
  86   %exitcond = icmp eq i64 %indvars.iv.next, 64
  87   br i1 %exitcond, label %for.end, label %for.body, !llvm.loop !7
  88
  89 for.end:                                          ; preds = %for.body
  90   ret void
  91 }
  92 !7 = metadata !{metadata !7, metadata !8}
  93 !8 = metadata !{metadata !"llvm.loop.unroll.disable"}
  94
  95 ; CHECK: ![[LOOP_1]] = metadata !{metadata ![[LOOP_1]], metadata ![[VEC_ENABLE:.*]], metadata ![[WIDTH_8:.*]], metadata ![[UNROLL_DISABLE:.*]]}
  96 ; CHECK: ![[VEC_ENABLE]] = metadata !{metadata !"llvm.loop.vectorize.enable", i1 true}
  97 ; CHECK: ![[WIDTH_8]] = metadata !{metadata !"llvm.loop.vectorize.width", i32 8}
  98 ; CHECK: ![[UNROLL_DISABLE]] = metadata !{metadata !"llvm.loop.unroll.disable"}
  99 ; CHECK: ![[LOOP_2]] = metadata !{metadata ![[LOOP_2]], metadata ![[UNROLL_FULL:.*]]}
 100 ; CHECK: ![[UNROLL_FULL]] = metadata !{metadata !"llvm.loop.unroll.full"}
 101 ; CHECK: ![[LOOP_3]] = metadata !{metadata ![[LOOP_3]], metadata ![[UNROLL_DISABLE:.*]]}