From 8641ab97a49b7e1d5b3920ecf99029f2a50762be Mon Sep 17 00:00:00 2001
From: Michael Zolotukhin <mzolotukhin@apple.com>
Date: Tue, 25 Jul 2017 10:35:16 +0000
Subject: [PATCH] [tests] Cleanup vect.omp.persistence.ll test.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@308964 91177308-0d34-0410-b5e6-96231b3b80d8
---
 .../LoopRotate/vect.omp.persistence.ll        | 34 ++++++++
 .../LoopVectorize/vect.omp.persistence.ll     | 83 ++++---------------
 2 files changed, 50 insertions(+), 67 deletions(-)
 create mode 100644 test/Transforms/LoopRotate/vect.omp.persistence.ll

diff --git a/test/Transforms/LoopRotate/vect.omp.persistence.ll b/test/Transforms/LoopRotate/vect.omp.persistence.ll
new file mode 100644
index 00000000000..6a1865499d3
--- /dev/null
+++ b/test/Transforms/LoopRotate/vect.omp.persistence.ll
@@ -0,0 +1,34 @@
+; RUN: opt < %s -loop-rotate -S | FileCheck %s
+
+target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
+
+; Ensure that "llvm.loop.vectorize.enable" metadata was not lost after loop-rotate.
+; In past LoopRotate was clearing that metadata.
+;
+; See http://reviews.llvm.org/D3348 for details.
+
+; CHECK-LABEL: @foo
+; CHECK: loop_cond:
+; CHECK-NOT: loop-inc
+; CHECK: br i1 %cmp, label %return, label %loop_cond, !llvm.loop !0
+; CHECK: !0 = distinct !{!0, !1}
+; CHECK: !1 = !{!"llvm.loop.vectorize.enable", i1 true}
+define i32 @foo(i32 %a) {
+entry:
+  br label %loop_cond
+
+loop_cond:
+  %indx = phi i32 [ 1, %entry ], [ %inc, %loop_inc ]
+  %cmp = icmp ne i32 %indx, %a
+  br i1 %cmp, label %return, label %loop_inc
+
+loop_inc:
+  %inc = add i32 %indx, 1
+  br label %loop_cond, !llvm.loop !0
+
+return:
+  ret i32 0
+}
+
+!0 = !{!0, !1}
+!1 = !{!"llvm.loop.vectorize.enable", i1 true}
diff --git a/test/Transforms/LoopVectorize/vect.omp.persistence.ll b/test/Transforms/LoopVectorize/vect.omp.persistence.ll
index 888deccb338..995fc1f2c3d 100644
--- a/test/Transforms/LoopVectorize/vect.omp.persistence.ll
+++ b/test/Transforms/LoopVectorize/vect.omp.persistence.ll
@@ -1,87 +1,36 @@
-; RUN: opt < %s -O2 -force-vector-interleave=2 -force-vector-width=4 -debug-only=loop-vectorize -stats -S 2>&1 | FileCheck %s
+; RUN: opt < %s -O2 -force-vector-interleave=2 -force-vector-width=4 -debug-only=loop-vectorize -S 2>&1 | FileCheck %s
 ; REQUIRES: asserts
 
-; Loop from "rotated"
+; CHECK: LV: Checking a loop in "foo"
 ; CHECK: LV: Loop hints: force=enabled
-; Loop from "nonrotated"
-; CHECK: LV: Loop hints: force=enabled
-; No more loops in the module
-; CHECK-NOT: LV: Loop hints: force=
-; In total only 1 loop should be rotated.
-; CHECK: 1 loop-rotate
 
 target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
 
-; See http://reviews.llvm.org/D3348 for details.
-
-;
-; Test #1
-;
-; Ensure that "llvm.loop.vectorize.enable" metadata was not lost prior to LoopVectorize pass.
-; In past LoopRotate was clearing that metadata.
-;
-; The source C code is:
-; void rotated(float *a, int size)
-; {
-;   int t = 0;
-;   #pragma omp simd
-;   for (int i = 0; i < size; ++i) {
-;     a[i] = a[i-5] * a[i+2];
-;     ++t;
-;   }
-;}
-
-define void @rotated(float* nocapture %a, i64 %size) {
-entry:
-  %cmp1 = icmp sgt i64 %size, 0
-  br i1 %cmp1, label %for.header, label %for.end
-
-for.header:
-  %indvars.iv = phi i64 [ %indvars.iv.next, %for.body ], [ 0, %entry ]
-  %cmp2 = icmp sgt i64 %indvars.iv, %size
-  br i1 %cmp2, label %for.end, label %for.body
-
-for.body:
-
-  %0 = add nsw i64 %indvars.iv, -5
-  %arrayidx = getelementptr inbounds float, float* %a, i64 %0
-  %1 = load float, float* %arrayidx, align 4, !llvm.mem.parallel_loop_access !1
-  %2 = add nsw i64 %indvars.iv, 2
-  %arrayidx2 = getelementptr inbounds float, float* %a, i64 %2
-  %3 = load float, float* %arrayidx2, align 4, !llvm.mem.parallel_loop_access !1
-  %mul = fmul float %1, %3
-  %arrayidx4 = getelementptr inbounds float, float* %a, i64 %indvars.iv
-  store float %mul, float* %arrayidx4, align 4, !llvm.mem.parallel_loop_access !1
-
-  %indvars.iv.next = add nuw nsw i64 %indvars.iv, 1
-  br label %for.header, !llvm.loop !1
-
-for.end:
-  ret void
-}
-
-!1 = !{!1, !2}
-!2 = !{!"llvm.loop.vectorize.enable", i1 true}
-
-;
-; Test #2
-;
 ; Ensure that "llvm.loop.vectorize.enable" metadata was not lost even
-; if loop was not rotated (see http://reviews.llvm.org/D3348#comment-4).
+; if loop was not rotated.
 ;
-define i32 @nonrotated(i32 %a) {
+; See http://reviews.llvm.org/D3348 for details.
+;
+; CHECK-LABEL: @foo
+; CHECK: !llvm.loop !0
+; CHECK: !0 = distinct !{!0, !1}
+; CHECK: !1 = !{!"llvm.loop.vectorize.enable", i1 true}
+define i32 @foo(i32 %a) {
 entry:
   br label %loop_cond
+
 loop_cond:
   %indx = phi i32 [ 1, %entry ], [ %inc, %loop_inc ]
   %cmp = icmp ne i32 %indx, %a
   br i1 %cmp, label %return, label %loop_inc
+
 loop_inc:
   %inc = add i32 %indx, 1
-  br label %loop_cond, !llvm.loop !3
+  br label %loop_cond, !llvm.loop !0
+
 return:
   ret i32 0
 }
 
-!3 = !{!3, !4}
-!4 = !{!"llvm.loop.vectorize.enable", i1 true}
+!0 = !{!0, !1}
+!1 = !{!"llvm.loop.vectorize.enable", i1 true}
-- 
2.40.0