From 157382600535c95bdbf5d6e8c4b9e5ca71749018 Mon Sep 17 00:00:00 2001
From: Nia Waldvogel <niaow1234@gmail.com>
Date: Fri, 17 Sep 2021 12:23:01 -0400
Subject: [PATCH] transform (coroutines): move any misplaced entry-block
 allocas to the start of the entry block before lowering

---
 transform/coroutines.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/transform/coroutines.go b/transform/coroutines.go
index 9a382c3aed..e1f109b1b6 100644
--- a/transform/coroutines.go
+++ b/transform/coroutines.go
@@ -763,6 +763,27 @@ func (c *coroutineLoweringPass) lowerCallReturn(caller *asyncFunc, call llvm.Val
 // lowerFuncCoro transforms an async function into a coroutine by lowering async operations to `llvm.coro` intrinsics.
 // See https://llvm.org/docs/Coroutines.html for more information on these intrinsics.
 func (c *coroutineLoweringPass) lowerFuncCoro(fn *asyncFunc) {
+	// Ensure that any alloca instructions in the entry block are at the start.
+	// Otherwise, block splitting would result in unintended behavior.
+	{
+		// Skip alloca instructions at the start of the block.
+		inst := fn.fn.FirstBasicBlock().FirstInstruction()
+		for !inst.IsAAllocaInst().IsNil() {
+			inst = llvm.NextInstruction(inst)
+		}
+
+		// Find any other alloca instructions and move them after the other allocas.
+		c.builder.SetInsertPointBefore(inst)
+		for !inst.IsNil() {
+			next := llvm.NextInstruction(inst)
+			if !inst.IsAAllocaInst().IsNil() {
+				inst.RemoveFromParentAsInstruction()
+				c.builder.Insert(inst)
+			}
+			inst = next
+		}
+	}
+
 	returnType := fn.fn.Type().ElementType().ReturnType()
 
 	// Prepare coroutine state.