deepak2427 updated this revision to Diff 153393.
deepak2427 added a comment.
Add tests and the patch.
https://reviews.llvm.org/D48721
Files:
lib/CodeGen/CGStmt.cpp
test/CodeGen/pragma-do-while-unroll.cpp
test/CodeGen/pragma-do-while.cpp
Index: test/CodeGen/pragma-do-while.cpp
===================================================================
--- /dev/null
+++ test/CodeGen/pragma-do-while.cpp
@@ -0,0 +1,31 @@
+// RUN: %clang_cc1 -O3 -fno-unroll-loops -S -emit-llvm %s -o - | FileCheck %s
+int test(int a[], int n) {
+ int i = 0;
+ int sum = 0;
+
+//CHECK-NOT: llvm.loop
+
+#pragma unroll 4
+ do
+ //CHECK: do.body:
+ //CHECK: llvm.loop
+ {
+ a[i] = a[i] + 1;
+ sum = sum + a[i];
+ i++;
+ } while (i < n);
+
+ i = 0;
+
+#pragma unroll 8
+ do
+ //CHECK: do.body{{[0-9]+}}:
+ //CHECK: llvm.loop
+ {
+ a[i] = a[i] + 1;
+ sum = sum + a[i];
+ i++;
+ } while (i < n);
+
+ return sum;
+}
Index: test/CodeGen/pragma-do-while-unroll.cpp
===================================================================
--- /dev/null
+++ test/CodeGen/pragma-do-while-unroll.cpp
@@ -0,0 +1,37 @@
+// RUN: %clang_cc1 -O3 -funroll-loops -S -emit-llvm %s -o - | FileCheck %s
+int test(int a[], int n) {
+ int i = 0;
+ int sum = 0;
+
+#pragma unroll 3
+ do
+ //CHECK: do.body:
+ //CHECK: store i32 %add
+ //CHECK: store i32 %add.1
+ //CHECK: store i32 %add.2
+ //CHECK-NOT: store i32 %add.3
+ {
+ a[i] = a[i] + 1;
+ sum = sum + a[i];
+ i++;
+ } while (i < n);
+
+ i = 0;
+
+#pragma unroll 5
+ do
+ //CHECK: do.body{{[0-9]+}}:
+ //CHECK: store i32 %add{{[0-9]+}}
+ //CHECK: store i32 %add{{[0-9]+}}.1
+ //CHECK: store i32 %add{{[0-9]+}}.2
+ //CHECK: store i32 %add{{[0-9]+}}.3
+ //CHECK: store i32 %add{{[0-9]+}}.4
+ //CHECK-NOT: store i32 %add{{[0-9]+}}.5
+ {
+ a[i] = a[i] + 1;
+ sum = sum + a[i];
+ i++;
+ } while (i < n);
+
+ return sum;
+}
Index: lib/CodeGen/CGStmt.cpp
===================================================================
--- lib/CodeGen/CGStmt.cpp
+++ lib/CodeGen/CGStmt.cpp
@@ -777,19 +777,19 @@
// Emit the body of the loop.
llvm::BasicBlock *LoopBody = createBasicBlock("do.body");
- const SourceRange &R = S.getSourceRange();
- LoopStack.push(LoopBody, CGM.getContext(), DoAttrs,
- SourceLocToDebugLoc(R.getBegin()),
- SourceLocToDebugLoc(R.getEnd()));
-
EmitBlockWithFallThrough(LoopBody, &S);
{
RunCleanupsScope BodyScope(*this);
EmitStmt(S.getBody());
}
EmitBlock(LoopCond.getBlock());
+ const SourceRange &R = S.getSourceRange();
+ LoopStack.push(LoopBody, CGM.getContext(), DoAttrs,
+ SourceLocToDebugLoc(R.getBegin()),
+ SourceLocToDebugLoc(R.getEnd()));
+
// C99 6.8.5.2: "The evaluation of the controlling expression takes place
// after each execution of the loop body."
_______________________________________________
cfe-commits mailing list
[email protected]
http://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
