blob: aac3cc970962a184531b68b76c6379ec9a4fe3a3 [file] [log] [blame]
; RUN: opt < %s -passes='default<O2>' -S | FileCheck --check-prefixes=CHECK %s
target datalayout = "p:64:64:64"
%swift.async_func_pointer = type <{ i32, i32 }>
%swift.context = type { ptr, ptr }
@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
declare swifttailcc void @callee.0(ptr swiftasync, ptr, i64, i64)
define internal swifttailcc void @callee(ptr %0, i64 %1, i64 %2, ptr %3) {
entry:
musttail call swifttailcc void @callee.0(ptr swiftasync %3, ptr %0, i64 %1, i64 %2)
ret void
}
define swifttailcc void @repo(ptr swiftasync %0) {
entry:
%1 = alloca ptr, align 8
%2 = call token @llvm.coro.id.async(i32 16, i32 16, i32 0, ptr @repoTU)
%3 = call ptr @llvm.coro.begin(token %2, ptr null)
store ptr %0, ptr %1, align 8
; This context.addr is the address in the frame of the first partial function after splitting.
%4 = call ptr @llvm.swift.async.context.addr()
store ptr null, ptr %4, align 8
%5 = call ptr @llvm.coro.async.resume()
%6 = call { ptr } (i32, ptr, ptr, ...) @llvm.coro.suspend.async.sl_p0i8s(i32 0,
ptr %5,
ptr @__swift_async_resume_get_context,
ptr @callee,
ptr %5, i64 0, i64 0, ptr %0)
%7 = load ptr, ptr %1, align 8
%8 = getelementptr inbounds <{ ptr, ptr }>, <{ ptr, ptr }>* %7, i32 0, i32 1
%9 = load ptr, ptr %8, align 8
%10 = load ptr, ptr %1, align 8
; This context.addr is the address in the frame of the second partial function after splitting.
; It is not valid to CSE it with the previous call.
%11 = call ptr @llvm.swift.async.context.addr()
store ptr %9, ptr %11, align 8
%12 = call i1 (ptr, i1, ...) @llvm.coro.end.async(ptr %3, i1 false, ptr @repo.0, ptr %9, ptr %10)
unreachable
}
; Make sure we don't CSE the llvm.swift.async.context.addr calls
; CHECK: define swifttailcc void @repo
; CHECK: call ptr @llvm.swift.async.context.addr()
; CHECK: define {{.*}}swifttailcc void @repoTY0_
; CHECK: call ptr @llvm.swift.async.context.addr()
define internal swifttailcc void @repo.0(ptr %0, ptr %1) #1 {
entry:
musttail call swifttailcc void %0(ptr swiftasync %1)
ret void
}
define linkonce_odr hidden ptr @__swift_async_resume_get_context(ptr %0) #1 {
entry:
ret ptr %0
}
declare { ptr } @llvm.coro.suspend.async.sl_p0i8s(i32, ptr, ptr, ...) #1
declare token @llvm.coro.id.async(i32, i32, i32, ptr) #1
declare ptr @llvm.coro.begin(token, ptr writeonly) #1
declare i1 @llvm.coro.end.async(ptr, i1, ...) #1
declare ptr @llvm.coro.async.resume() #1
declare ptr @llvm.swift.async.context.addr() #1
attributes #1 = { nounwind }