commit 5efc61866d24b11e8aabba8d7d9443cb1f1e3215
author Gor Nishanov <GorNishanov@gmail.com> Tue May 23 05:04:01 2017 +0000
committer Gor Nishanov <GorNishanov@gmail.com> Tue May 23 05:04:01 2017 +0000
tree 2f9574d626f78b388092a2da91fced7d1851dfcb
parent c46203eaf557f738b25e1d261272d51ef953e561

[coroutines] Add emission of initial and final suspends

https://reviews.llvm.org/D31608

llvm-svn: 303603

clang/lib/CodeGen/CGCoroutine.cpp

diff --git a/clang/lib/CodeGen/CGCoroutine.cpp b/clang/lib/CodeGen/CGCoroutine.cpp
index 26291c9..bb98c24 100644
--- a/clang/lib/CodeGen/CGCoroutine.cpp
+++ b/clang/lib/CodeGen/CGCoroutine.cpp
@@ -388,7 +388,10 @@
 
     CurCoro.Data->FinalJD = getJumpDestInCurrentScope(FinalBB);
 
-    // FIXME: Emit initial suspend and more before the body.
+    // FIXME: Emit param moves.
+
+    CurCoro.Data->CurrentAwaitKind = AwaitKind::Init;
+    EmitStmt(S.getInitSuspendStmt());
 
     CurCoro.Data->CurrentAwaitKind = AwaitKind::Normal;
 
@@ -410,7 +413,8 @@
     const bool HasCoreturns = CurCoro.Data->CoreturnCount > 0;
     if (CanFallthrough || HasCoreturns) {
       EmitBlock(FinalBB);
-      // FIXME: Emit final suspend.
+      CurCoro.Data->CurrentAwaitKind = AwaitKind::Final;
+      EmitStmt(S.getFinalSuspendStmt());
     }
   }