[PATCH] D33663: CGCleanup: Use correct insertion point for AllocaInst

[Gor Nishanov via Phabricator via cfe-commits]

GorNishanov updated this revision to Diff 100889.
GorNishanov added a comment.

Make sure that we use AllocaInsertPt for only static allocas in the entry block


https://reviews.llvm.org/D33663

Files:
  lib/CodeGen/CGCleanup.cpp
  test/CodeGenCoroutines/coro-await-domination.cpp


Index: test/CodeGenCoroutines/coro-await-domination.cpp
===================================================================
--- /dev/null
+++ test/CodeGenCoroutines/coro-await-domination.cpp
@@ -0,0 +1,34 @@
+// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -fcoroutines-ts -std=c++14 -emit-llvm %s -o - | FileCheck %s
+#include "Inputs/coroutine.h"
+
+using namespace std::experimental;
+
+struct coro {
+  struct promise_type {
+    coro get_return_object();
+    suspend_never initial_suspend();
+    suspend_never final_suspend();
+    void return_void();
+    static void unhandled_exception();
+  };
+};
+
+struct A {
+  ~A();
+  bool await_ready();
+  int await_resume();
+  template <typename F> void await_suspend(F);
+};
+
+// Verifies that domination is properly built during cleanup.
+// Without CGCleanup.cpp fix verifier was reporting:
+// Instruction does not dominate all uses!
+//  %tmp.exprcleanup = alloca i32*, align 8
+//  store i32* %x, i32** %tmp.exprcleanup, align 8
+
+
+// CHECK-LABEL: f(
+extern "C" coro f(int) {
+  int x = 42;
+  x = co_await A{};
+}
Index: lib/CodeGen/CGCleanup.cpp
===================================================================
--- lib/CodeGen/CGCleanup.cpp
+++ lib/CodeGen/CGCleanup.cpp
@@ -416,6 +416,12 @@
     EHStack.popNullFixups();
 }
 
+static bool isStaticAlloca(llvm::Value* V) {
+  if (auto *AI = dyn_cast<llvm::AllocaInst>(V))
+    return AI->isStaticAlloca();
+  return false;
+}
+
 /// Pops cleanup blocks until the given savepoint is reached.
 void CodeGenFunction::PopCleanupBlocks(
     EHScopeStack::stable_iterator Old,
@@ -455,6 +461,8 @@
     llvm::BasicBlock::iterator InsertBefore;
     if (auto *Invoke = dyn_cast<llvm::InvokeInst>(Inst))
       InsertBefore = Invoke->getNormalDest()->getFirstInsertionPt();
+    else if (isStaticAlloca(Inst))
+      InsertBefore = std::next(AllocaInsertPt->getIterator());
     else
       InsertBefore = std::next(Inst->getIterator());
     CGBuilderTy(CGM, &*InsertBefore).CreateStore(Inst, Tmp);


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D33663.100889.patch
Type: text/x-patch
Size: 2015 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/cfe-commits/attachments/20170531/8e993c5d/attachment.bin>