llvm/test/Transforms/Coroutines/coro-async-no-cse-swift-async-context-addr.ll

   1 ; RUN: opt < %s -passes='default<O2>' -S | FileCheck --check-prefixes=CHECK %s
   2 target 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
   9 declare swifttailcc void @callee.0(ptr swiftasync, ptr, i64, i64)
  10
  11 define internal swifttailcc void @callee(ptr %0, i64 %1, i64 %2, ptr %3) {
  12 entry:
  13   musttail call swifttailcc void @callee.0(ptr swiftasync %3, ptr %0, i64 %1, i64 %2)
  14   ret void
  15 }
  16
  17 define swifttailcc void @repo(ptr swiftasync %0) {
  18 entry:
  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
  55 define internal swifttailcc void @repo.0(ptr %0, ptr %1) #1 {
  56 entry:
  57   musttail call swifttailcc void %0(ptr swiftasync %1)
  58   ret void
  59 }
  60
  61 define linkonce_odr hidden ptr @__swift_async_resume_get_context(ptr %0) #1 {
  62 entry:
  63   ret ptr %0
  64 }
  65
  66 declare { ptr } @llvm.coro.suspend.async.sl_p0i8s(i32, ptr, ptr, ...) #1
  67 declare token @llvm.coro.id.async(i32, i32, i32, ptr) #1
  68 declare ptr @llvm.coro.begin(token, ptr writeonly) #1
  69 declare i1 @llvm.coro.end.async(ptr, i1, ...) #1
  70 declare ptr @llvm.coro.async.resume() #1
  71 declare ptr @llvm.swift.async.context.addr() #1
  72
  73 attributes #1 = { nounwind }