xref: /llvm-project/llvm/test/Transforms/Coroutines/coro-async-no-cse-swift-async-context-addr.ll (revision 2d69827c5c754f0eca98e497ecf0e52ed54b4fd3) 
1; RUN: opt < %s -passes='default<O2>' -S | FileCheck --check-prefixes=CHECK %s
2target datalayout = "p:64:64:64"
3
4%swift.async_func_pointer = type <{ i32, i32 }>
5%swift.context = type { ptr, ptr }
6
7@repoTU = global %swift.async_func_pointer <{ i32 trunc (i64 sub (i64 ptrtoint (ptr @repo to i64), i64 ptrtoint (ptr @repoTU to i64)) to i32), i32 16 }>, align 8
8
9declare swifttailcc void @callee.0(ptr swiftasync, ptr, i64, i64)
10
11define internal swifttailcc void @callee(ptr %0, i64 %1, i64 %2, ptr %3) {
12entry:
13  musttail call swifttailcc void @callee.0(ptr swiftasync %3, ptr %0, i64 %1, i64 %2)
14  ret void
15}
16
17define swifttailcc void @repo(ptr swiftasync %0) {
18entry:
19  %1 = alloca ptr, align 8
20  %2 = call token @llvm.coro.id.async(i32 16, i32 16, i32 0, ptr @repoTU)
21  %3 = call ptr @llvm.coro.begin(token %2, ptr null)
22  store ptr %0, ptr %1, align 8
23
24  ; This context.addr is the address in the frame of the first partial function after splitting.
25  %4 = call ptr @llvm.swift.async.context.addr()
26	store ptr null, ptr %4, align 8
27
28  %5 = call ptr @llvm.coro.async.resume()
29  %6 = call { ptr } (i32, ptr, ptr, ...) @llvm.coro.suspend.async.sl_p0i8s(i32 0,
30                                                                           ptr %5,
31                                                                           ptr @__swift_async_resume_get_context,
32                                                                           ptr @callee,
33                                                                           ptr %5, i64 0, i64 0, ptr %0)
34  %7 = load ptr, ptr %1, align 8
35  %8 = getelementptr inbounds <{ ptr, ptr }>, ptr %7, i32 0, i32 1
36  %9 = load ptr, ptr %8, align 8
37  %10 = load ptr, ptr %1, align 8
38
39  ; This context.addr is the address in the frame of the second partial function after splitting.
40  ; It is not valid to CSE it with the previous call.
41  %11 = call ptr @llvm.swift.async.context.addr()
42	store ptr %9, ptr %11, align 8
43
44  %12 = call i1 (ptr, i1, ...) @llvm.coro.end.async(ptr %3, i1 false, ptr @repo.0, ptr %9, ptr %10)
45  unreachable
46}
47
48; Make sure we don't CSE the llvm.swift.async.context.addr calls
49; CHECK: define swifttailcc void @repo
50; CHECK: call ptr @llvm.swift.async.context.addr()
51
52; CHECK: define {{.*}}swifttailcc void @repoTY0_
53; CHECK: call ptr @llvm.swift.async.context.addr()
54
55define internal swifttailcc void @repo.0(ptr %0, ptr %1) #1 {
56entry:
57  musttail call swifttailcc void %0(ptr swiftasync %1)
58  ret void
59}
60
61define linkonce_odr hidden ptr @__swift_async_resume_get_context(ptr %0) #1 {
62entry:
63  ret ptr %0
64}
65
66declare { ptr } @llvm.coro.suspend.async.sl_p0i8s(i32, ptr, ptr, ...) #1
67declare token @llvm.coro.id.async(i32, i32, i32, ptr) #1
68declare ptr @llvm.coro.begin(token, ptr writeonly) #1
69declare i1 @llvm.coro.end.async(ptr, i1, ...) #1
70declare ptr @llvm.coro.async.resume() #1
71declare ptr @llvm.swift.async.context.addr() #1
72
73attributes #1 = { nounwind }
74