1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
; RUN: opt < %s -O0 -S | FileCheck --check-prefixes=CHECK %s
target datalayout = "p:64:64:64"
%async.task = type { i64 }
%async.actor = type { i64 }
%async.fp = type <{ i32, i32 }>
%async.ctxt = type { ptr, ptr }
; The async callee.
@my_other_async_function_fp = external global <{ i32, i32 }>
declare void @my_other_async_function(ptr %async.ctxt)
@my_async_function_fp = constant <{ i32, i32 }>
<{ i32 trunc ( ; Relative pointer to async function
i64 sub (
i64 ptrtoint (ptr @my_async_function to i64),
i64 ptrtoint (ptr getelementptr inbounds (<{ i32, i32 }>, ptr @my_async_function_fp, i32 0, i32 1) to i64)
)
to i32),
i32 128 ; Initial async context size without space for frame
}>
define swiftcc void @my_other_async_function_fp.apply(ptr %fnPtr, ptr %async.ctxt) {
tail call swiftcc void %fnPtr(ptr %async.ctxt)
ret void
}
declare void @escape(ptr)
declare void @store_resume(ptr)
declare i1 @exitLoop()
define ptr @resume_context_projection(ptr %ctxt) {
entry:
%resume_ctxt = load ptr, ptr %ctxt, align 8
ret ptr %resume_ctxt
}
define swiftcc void @my_async_function(ptr swiftasync %async.ctxt) {
entry:
%escaped_addr = alloca i64
%id = call token @llvm.coro.id.async(i32 128, i32 16, i32 0,
ptr @my_async_function_fp)
%hdl = call ptr @llvm.coro.begin(token %id, ptr null)
call void @llvm.lifetime.start.p0(i64 4, ptr %escaped_addr)
br label %callblock
callblock:
%callee_context = call ptr @context_alloc()
%resume.func_ptr = call ptr @llvm.coro.async.resume()
call void @store_resume(ptr %resume.func_ptr)
%res = call {ptr, ptr, ptr} (i32, ptr, ptr, ...) @llvm.coro.suspend.async(i32 0,
ptr %resume.func_ptr,
ptr @resume_context_projection,
ptr @my_other_async_function_fp.apply,
ptr @asyncSuspend, ptr %callee_context)
call void @escape(ptr %escaped_addr)
%exitCond = call i1 @exitLoop()
;; We used to move the lifetime.start intrinsic here =>
;; This exposes two bugs:
; 1.) The code should use the basic block start not end as insertion point
; More problematically:
; 2.) The code marks the stack object as not alive for part of the loop.
br i1 %exitCond, label %loop_exit, label %loop
%res2 = call {ptr, ptr, ptr} (i32, ptr, ptr, ...) @llvm.coro.suspend.async(i32 0,
ptr %resume.func_ptr,
ptr @resume_context_projection,
ptr @my_other_async_function_fp.apply,
ptr @asyncSuspend, ptr %callee_context)
%exitCond2 = call i1 @exitLoop()
br i1 %exitCond2, label %loop_exit, label %loop
loop:
br label %callblock
loop_exit:
call void @llvm.lifetime.end.p0(i64 4, ptr %escaped_addr)
call i1 (ptr, i1, ...) @llvm.coro.end.async(ptr %hdl, i1 false)
unreachable
}
; CHECK: define {{.*}} void @my_async_function.resume.0(
; CHECK-NOT: call void @llvm.lifetime.start.p0(i64 4, ptr %3)
; CHECK: br i1 %exitCond, label %loop_exit, label %loop
; CHECK: lifetime.end
; CHECK: }
declare { ptr, ptr, ptr, ptr } @llvm.coro.suspend.async.sl_p0i8p0i8p0i8p0i8s(i32, ptr, ptr, ...)
declare ptr @llvm.coro.prepare.async(ptr)
declare token @llvm.coro.id.async(i32, i32, i32, ptr)
declare ptr @llvm.coro.begin(token, ptr)
declare i1 @llvm.coro.end.async(ptr, i1, ...)
declare i1 @llvm.coro.end(ptr, i1, token)
declare {ptr, ptr, ptr} @llvm.coro.suspend.async(i32, ptr, ptr, ...)
declare ptr @context_alloc()
declare void @llvm.coro.async.context.dealloc(ptr)
declare swiftcc void @asyncSuspend(ptr)
declare ptr @llvm.coro.async.resume()
declare void @llvm.coro.async.size.replace(ptr, ptr)
declare void @llvm.lifetime.start.p0(i64 immarg, ptr nocapture) #0
declare void @llvm.lifetime.end.p0(i64 immarg, ptr nocapture) #0
attributes #0 = { argmemonly nofree nosync nounwind willreturn }
|
