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
|
; Testing elide performed its job for calls to coroutines marked safe.
; RUN: opt < %s -S -passes='cgscc(coro-annotation-elide)' | FileCheck %s
%struct.Task = type { ptr }
declare void @print(i32) nounwind
; resume part of the coroutine
define fastcc void @callee.resume(ptr dereferenceable(1)) {
tail call void @print(i32 0)
ret void
}
; destroy part of the coroutine
define fastcc void @callee.destroy(ptr) {
tail call void @print(i32 1)
ret void
}
; cleanup part of the coroutine
define fastcc void @callee.cleanup(ptr) {
tail call void @print(i32 2)
ret void
}
@callee.resumers = internal constant [3 x ptr] [
ptr @callee.resume, ptr @callee.destroy, ptr @callee.cleanup]
declare void @alloc(i1) nounwind
; CHECK-LABEL: define ptr @callee
define ptr @callee(i8 %arg) {
entry:
%task = alloca %struct.Task, align 8
%id = call token @llvm.coro.id(i32 0, ptr null,
ptr @callee,
ptr @callee.resumers)
%alloc = call i1 @llvm.coro.alloc(token %id)
%hdl = call ptr @llvm.coro.begin(token %id, ptr null)
store ptr %hdl, ptr %task
ret ptr %task
}
; CHECK-LABEL: define ptr @callee.noalloc
define ptr @callee.noalloc(i8 %arg, ptr dereferenceable(32) align(8) %frame) {
entry:
%task = alloca %struct.Task, align 8
%id = call token @llvm.coro.id(i32 0, ptr null,
ptr @callee,
ptr @callee.resumers)
%hdl = call ptr @llvm.coro.begin(token %id, ptr null)
store ptr %hdl, ptr %task
ret ptr %task
}
; CHECK-LABEL: define ptr @caller()
; Function Attrs: presplitcoroutine
define ptr @caller() #0 {
entry:
%task = call ptr @callee(i8 0) #1
ret ptr %task
; CHECK: %[[TASK:.+]] = alloca %struct.Task, align 8
; CHECK-NEXT: %[[FRAME:.+]] = alloca [32 x i8], align 8
; CHECK-NEXT: call void @llvm.lifetime.start.p0(i64 8, ptr %[[TASK]])
; CHECK-NEXT: %[[ID:.+]] = call token @llvm.coro.id(i32 0, ptr null, ptr @callee, ptr @callee.resumers)
; CHECK-NEXT: %[[HDL:.+]] = call ptr @llvm.coro.begin(token %[[ID]], ptr null)
; CHECK-NEXT: store ptr %[[HDL]], ptr %[[TASK]], align 8
; CHECK-NEXT: call void @llvm.lifetime.end.p0(i64 8, ptr %[[TASK]])
; CHECK-NEXT: ret ptr %[[TASK]]
}
declare token @llvm.coro.id(i32, ptr, ptr, ptr)
declare ptr @llvm.coro.begin(token, ptr)
declare ptr @llvm.coro.frame()
declare ptr @llvm.coro.subfn.addr(ptr, i8)
declare i1 @llvm.coro.alloc(token)
attributes #0 = { presplitcoroutine }
attributes #1 = { coro_elide_safe }
|
