96 lines · plain
1; RUN: opt < %s -passes='default<O2>' -S | FileCheck --check-prefixes=CHECK %s2 3target datalayout = "p:64:64:64"4 5%async.task = type { i64 }6%async.actor = type { i64 }7%async.fp = type <{ i32, i32 }>8 9%async.ctxt = type { ptr, ptr }10 11; The async callee.12@my_other_async_function_fp = external global <{ i32, i32 }>13declare void @my_other_async_function(ptr %async.ctxt)14 15; Function that implements the dispatch to the callee function.16define swiftcc void @my_async_function.my_other_async_function_fp.apply(ptr %fnPtr, ptr %async.ctxt, ptr %task, ptr %actor) {17 tail call swiftcc void %fnPtr(ptr %async.ctxt, ptr %task, ptr %actor)18 ret void19}20 21declare void @some_user(i64)22declare void @some_may_write(ptr)23 24define ptr @resume_context_projection(ptr %ctxt) {25entry:26 %resume_ctxt = load ptr, ptr %ctxt, align 827 ret ptr %resume_ctxt28}29 30 31@unreachable_fp = constant <{ i32, i32 }>32 <{ i32 trunc ( ; Relative pointer to async function33 i64 sub (34 i64 ptrtoint (ptr @unreachable to i64),35 i64 ptrtoint (ptr getelementptr inbounds (<{ i32, i32 }>, ptr @unreachable_fp, i32 0, i32 1) to i64)36 )37 to i32),38 i32 128 ; Initial async context size without space for frame39}>40 41define swiftcc void @unreachable(ptr %async.ctxt, ptr %task, ptr %actor) {42entry:43 %tmp = alloca { i64, i64 }, align 844 %proj.1 = getelementptr inbounds { i64, i64 }, ptr %tmp, i64 0, i32 045 %proj.2 = getelementptr inbounds { i64, i64 }, ptr %tmp, i64 0, i32 146 47 %id = call token @llvm.coro.id.async(i32 128, i32 16, i32 0,48 ptr @unreachable_fp)49 %hdl = call ptr @llvm.coro.begin(token %id, ptr null)50 store i64 0, ptr %proj.1, align 851 store i64 1, ptr %proj.2, align 852 call void @some_may_write(ptr %proj.1)53 54 ; Begin lowering: apply %my_other_async_function(%args...)55 ; setup callee context56 %arg1 = bitcast ptr @my_other_async_function_fp to ptr57 %callee_context = call ptr @llvm.coro.async.context.alloc(ptr %task, ptr %arg1)58 ; store the return continuation59 %callee_context.return_to_caller.addr = getelementptr inbounds %async.ctxt, ptr %callee_context, i32 0, i32 160 %resume.func_ptr = call ptr @llvm.coro.async.resume()61 store ptr %resume.func_ptr, ptr %callee_context.return_to_caller.addr62 63 ; store caller context into callee context64 store ptr %async.ctxt, ptr %callee_context65 %res = call {ptr, ptr, ptr} (i32, ptr, ptr, ...) @llvm.coro.suspend.async(i32 0,66 ptr %resume.func_ptr,67 ptr @resume_context_projection,68 ptr @my_async_function.my_other_async_function_fp.apply,69 ptr @asyncSuspend, ptr %callee_context, ptr %task, ptr %actor)70 71 call void @llvm.coro.async.context.dealloc(ptr %callee_context)72 %continuation_task_arg = extractvalue {ptr, ptr, ptr} %res, 173 %val = load i64, ptr %proj.174 call void @some_user(i64 %val)75 %val.2 = load i64, ptr %proj.276 call void @some_user(i64 %val.2)77 unreachable78}79 80; CHECK: define swiftcc void @unreachable81; CHECK-NOT: @llvm.coro.suspend.async82; CHECK: return83 84; CHECK: define internal swiftcc void @unreachable.resume.085; CHECK: unreachable86 87declare ptr @llvm.coro.prepare.async(ptr)88declare token @llvm.coro.id.async(i32, i32, i32, ptr)89declare ptr @llvm.coro.begin(token, ptr)90declare {ptr, ptr, ptr} @llvm.coro.suspend.async(i32, ptr, ptr, ...)91declare ptr @llvm.coro.async.context.alloc(ptr, ptr)92declare void @llvm.coro.async.context.dealloc(ptr)93declare swiftcc void @asyncReturn(ptr, ptr, ptr)94declare swiftcc void @asyncSuspend(ptr, ptr, ptr)95declare ptr @llvm.coro.async.resume()96