averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity
Files
c141586838e304857cd573b639e3071574a14c4f
swift-mirror/test/IRGen/coroutine_accessors.swift
Nate Chandler c141586838 [CoroutineAccessors] Open code dealloc fn.
2025-03-27 19:23:09 -07:00

270 lines
11 KiB
Swift
Raw Blame History

// RUN: %target-swift-emit-irgen \
// RUN: %s \
// RUN: -enable-experimental-feature CoroutineAccessors \
// RUN: | %IRGenFileCheck %s
// REQUIRES: swift_feature_CoroutineAccessors
// CHECK-LABEL: @"$s19coroutine_accessors1SV3irmSivxTwc" = {{.*}}global{{.*}} %swift.coro_func_pointer <{
// : sub (
// CHECK-SAME: $s19coroutine_accessors1SV3irmSivx
// : $s19coroutine_accessors1SV3irmSivxTwc
// : ),
// CHECK-SAME: i32 0
// CHECK-SAME: }>
// CHECK-LABEL: _swift_coro_malloc_allocator = linkonce_odr hidden constant %swift.coro_allocator {
// CHECK-SAME: i32 258,
// CHECK-SAME: malloc,
// CHECK-SAME: free
// CHECK-SAME: }
// CHECK-LABEL: _swift_coro_async_allocator = linkonce_odr hidden constant %swift.coro_allocator {
// CHECK-SAME: i32 1,
// CHECK-SAME: swift_task_alloc,
// CHECK-SAME: swift_task_dealloc
// CHECK-SAME: }
// CHECK-LABEL: @_swift_coro_alloc(
// CHECK-SAME: ptr [[ALLOCATOR:%[^,]+]]
// CHECK-SAME: [[INT]] [[SIZE:%[^)]+]]
// CHECK-SAME: )
// CHECK-SAME: {
// CHECK: entry:
// CHECK: [[ALLOCATE_FN_PTR:%[^,]+]] = getelementptr inbounds %swift.coro_allocator
// CHECK-SAME: ptr [[ALLOCATOR]]
// CHECK-SAME: i32 0
// CHECK-SAME: i32 1
// CHECK: [[ALLOCATE_FN:%[^,]+]] = load ptr, ptr [[ALLOCATE_FN_PTR]]
// CHECK: [[ALLOCATION:%[^,]+]] = call swiftcc ptr [[ALLOCATE_FN]]([[INT]] [[SIZE]])
// CHECK: ret ptr [[ALLOCATION]]
// CHECK: }
// CHECK-LABEL: @_swift_coro_dealloc(
// CHECK-SAME: ptr [[ALLOCATOR:%[^,]+]]
// CHECK-SAME: ptr [[ADDRESS:%[^)]+]]
// CHECK-SAME: )
// CHECK-SAME: {
// CHECK: entry:
// CHECK: [[FLAGS_ADDR:%[^,]+]] = getelementptr inbounds %swift.coro_allocator
// CHECK-SAME: ptr [[ALLOCATOR]]
// CHECK-SAME: i32 0
// CHECK-SAME: i32 0
// CHECK: [[FLAGS:%[^,]+]] = load i32, ptr [[FLAGS_ADDR]], align 4
// CHECK: [[DEALLOC_DEFERRING_ALLOCATOR:%[^,]+]] = and i32 [[FLAGS]], 256
// CHECK: [[IS_DEALLOC_DEFERRING_ALLOCATOR:%[^,]+]] = icmp ne i32 [[DEALLOC_DEFERRING_ALLOCATOR]], 0
// CHECK: br i1 [[IS_DEALLOC_DEFERRING_ALLOCATOR]]
// CHECK-SAME: label %deferring_allocator
// CHECK-SAME: label %normal
// CHECK: deferring_allocator:
// CHECK: ret void
// CHECK: normal:
// CHECK: [[DEALLOCATE_FN_PTR:%[^,]+]] = getelementptr inbounds %swift.coro_allocator
// CHECK-SAME: ptr [[ALLOCATOR]]
// CHECK-SAME: i32 0
// CHECK-SAME: i32 2
// CHECK: [[DEALLOCATE_FN:%[^,]+]] = load ptr, ptr [[DEALLOCATE_FN_PTR]]
// CHECK: call swiftcc void [[DEALLOCATE_FN]](ptr [[ADDRESS]])
// CHECK: ret void
// CHECK: }
@frozen
public struct S {
public var o: any AnyObject
public var _i: Int = 0
public var irm: Int {
// CHECK-LABEL: define{{.*}} { ptr, {{i64|i32}} } @"$s19coroutine_accessors1SV3irmSivy"(
// CHECK-SAME: )
// CHECK-SAME: {
// CHECK: }
read {
yield _i
}
// CHECK-LABEL: define{{.*}} { ptr, ptr } @"$s19coroutine_accessors1SV3irmSivx"(
// CHECK-SAME: )
// CHECK-SAME: {
// CHECK: }
modify {
yield &_i
}
} // public var irm
// CHECK-LABEL: define{{.*}} void @increment_irm(
// CHECK-SAME: )
// CHECK-SAME: {
// : [[SIZE_32:%[^,]+]] = load i32
// : ptr getelementptr inbounds (
// : %swift.coro_func_pointer
// CHECK: $s19coroutine_accessors1SV3irmSivxTwc
// : i32 0
// : i32 1
// : )
// CHECK-64: [[SIZE_64:%[^,]+]] = zext i32 {{%[^,]+}} to i64
// CHECK-64: [[FRAME:%[^,]+]] = alloca i8, [[INT]] [[SIZE_64]]
// CHECK-32: [[FRAME:%[^,]+]] = alloca i8, [[INT]] {{%[^,]+}}
// CHECK: call void @llvm.lifetime.start.p0(i64 -1, ptr [[FRAME]])
// CHECK: [[RAMP:%[^,]+]] = call ptr @llvm.coro.prepare.retcon(ptr @"$s19coroutine_accessors1SV3irmSivx")
// CHECK: [[RETVAL:%[^,]+]] = call swiftcc { ptr, ptr } [[RAMP]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_malloc_allocator
// CHECK-SAME: )
// CHECK: [[CONTINUATION:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 0
// CHECK: [[YIELD:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 1
// CHECK: call swiftcc void @"$s19coroutine_accessors9incrementyySizF"(
// CHECK-SAME: [[YIELD]]
// CHECK-SAME: )
// CHECK: call swiftcc void [[CONTINUATION]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_malloc_allocator
// CHECK-SAME: )
// CHECK: call void @llvm.lifetime.end.p0(i64 -1, ptr [[FRAME]])
// CHECK: }
@_silgen_name("increment_irm")
public mutating func increment_irm() {
increment(&irm)
}
// CHECK-LABEL: define{{.*}} void @increment_irm_async(
// ptr swiftasync %0
// CHECK-SAME: )
// CHECK-SAME: {
// : [[SIZE_32:%[^,]+]] = load i32
// : ptr getelementptr inbounds (
// : %swift.coro_func_pointer
// CHECK: $s19coroutine_accessors1SV3irmSivxTwc
// : i32 0
// : i32 1
// : )
// CHECK-64: [[SIZE_64:%[^,]+]] = zext i32 {{%[^,]+}} to i64
// CHECK-64: [[SIZE_1:%[^,]+]] = add [[INT]] [[SIZE_64]], 15
// CHECK-32: [[SIZE_1:%[^,]+]] = add [[INT]] {{%[^,]+}}, 15
// CHECK: [[SIZE:%[^,]+]] = and [[INT]] [[SIZE_1]], -16
// CHECK: [[FRAME:%[^,]+]] = call swiftcc ptr @swift_task_alloc([[INT]] [[SIZE]])
// CHECK: call void @llvm.lifetime.start.p0(i64 -1, ptr [[FRAME]])
// CHECK: [[RAMP:%[^,]+]] = call ptr @llvm.coro.prepare.retcon(ptr @"$s19coroutine_accessors1SV3irmSivx")
// CHECK: [[RETVAL:%[^,]+]] = call swiftcc { ptr, ptr } [[RAMP]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_async_allocator
// CHECK-SAME: )
// CHECK: [[CONTINUATION:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 0
// CHECK: [[YIELD:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 1
// increment_async([[YIELD]])
// CHECK: call swiftcc void [[CONTINUATION]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_async_allocator
// CHECK-SAME: )
// CHECK: call void @llvm.lifetime.end.p0(i64 -1, ptr [[FRAME]])
// CHECK: call swiftcc void @swift_task_dealloc_through(ptr [[FRAME]])
// CHECK: }
@_silgen_name("increment_irm_async")
public mutating func increment_irm_async() async {
await increment_async(&irm)
}
public var force_yield_once_convention : () {
_read {
let nothing: () = ()
yield nothing
}
// CHECK-LABEL: define{{.*}} { ptr, ptr } @increment_irm_yield_once(
// ptr noalias dereferenceable(32) %0
// CHECK-SAME: )
// CHECK-SAME: {
// : [[SIZE_32:%[^,]+]] = load i32
// : ptr getelementptr inbounds (
// : %swift.coro_func_pointer
// : $s19coroutine_accessors1SV3irmSivxTwc
// : i32 0
// : i32 1
// : )
// CHECK-64: [[SIZE_64:%[^,]+]] = zext i32 {{%[^,]+}} to i64
// CHECK-64: [[ALLOCATION:%[^,]+]] = call token @llvm.coro.alloca.alloc.i64(i64 [[SIZE_64]], i32 16)
// CHECK-32: [[ALLOCATION:%[^,]+]] = call token @llvm.coro.alloca.alloc.i32(i32 {{%[^,]+}}, i32 16)
// CHECK: [[FRAME:%[^,]+]] = call ptr @llvm.coro.alloca.get(token [[ALLOCATION]])
// CHECK: call void @llvm.lifetime.start.p0(i64 -1, ptr [[FRAME]])
// CHECK: [[RAMP:%[^,]+]] = call ptr @llvm.coro.prepare.retcon(ptr @"$s19coroutine_accessors1SV3irmSivx")
// CHECK: [[RETVAL:%[^,]+]] = call swiftcc { ptr, ptr } [[RAMP]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_malloc_allocator
// CHECK-SAME: )
// CHECK: [[CONTINUATION:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 0
// CHECK: [[YIELD:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 1
// CHECK: call swiftcc void @"$s19coroutine_accessors9incrementyySizF"(
// CHECK-SAME: [[YIELD]]
// CHECK-SAME: )
// CHECK: call swiftcc void [[CONTINUATION]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: _swift_coro_malloc_allocator
// CHECK-SAME: )
// CHECK: call void @llvm.lifetime.end.p0(i64 -1, ptr [[FRAME]])
// CHECK: call void @llvm.coro.alloca.free(token [[ALLOCATION]])
// CHECK: }
@_silgen_name("increment_irm_yield_once")
_modify {
increment(&irm)
var nothing: () = ()
yield &nothing
}
}
public var force_yield_once_2_convention : () {
read {
let nothing: () = ()
yield nothing
}
// CHECK-LABEL: define{{.*}} { ptr, ptr } @increment_irm_yield_once_2(
// ptr noalias %0
// CHECK-SAME: ptr [[ALLOCATOR:%[^,]+]]
// ptr nocapture swiftself dereferenceable(16) %2
// CHECK-SAME: )
// CHECK-SAME: {
// : [[SIZE_32:%[^,]+]] = load i32
// : ptr getelementptr inbounds (
// : %swift.coro_func_pointer
// CHECK: $s19coroutine_accessors1SV3irmSivxTwc
// : i32 0
// : i32 1
// : )
// CHECK-64: [[SIZE_64:%[^,]+]] = zext i32 {{%[^,]+}} to i64
// CHECK-64: [[ALLOCATION:%[^,]+]] = call token @llvm.coro.alloca.alloc.i64(i64 [[SIZE_64]], i32 16)
// CHECK-32: [[ALLOCATION:%[^,]+]] = call token @llvm.coro.alloca.alloc.i32(i32 {{%[^,]+}}, i32 16)
// CHECK: [[FRAME:%[^,]+]] = call ptr @llvm.coro.alloca.get(token [[ALLOCATION]])
// CHECK: call void @llvm.lifetime.start.p0(i64 -1, ptr [[FRAME]])
// CHECK: [[RAMP:%[^,]+]] = call ptr @llvm.coro.prepare.retcon(ptr @"$s19coroutine_accessors1SV3irmSivx")
// CHECK: [[RETVAL:%[^,]+]] = call swiftcc { ptr, ptr } [[RAMP]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: [[ALLOCATOR]]
// CHECK-SAME: )
// CHECK: [[CONTINUATION:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 0
// CHECK: [[YIELD:%[^,]+]] = extractvalue { ptr, ptr } [[RETVAL]], 1
// CHECK: call swiftcc void @"$s19coroutine_accessors9incrementyySizF"(
// CHECK-SAME: [[YIELD]]
// CHECK-SAME: )
// CHECK: call swiftcc void [[CONTINUATION]](
// CHECK-SAME: [[FRAME]],
// CHECK-SAME: [[ALLOCATOR]]
// CHECK-SAME: )
// CHECK: call void @llvm.lifetime.end.p0(i64 -1, ptr [[FRAME]])
// CHECK: call void @llvm.coro.alloca.free(token [[ALLOCATION]])
// CHECK: }
@_silgen_name("increment_irm_yield_once_2")
modify {
increment(&irm)
var nothing: () = ()
yield &nothing
}
}
}
public func increment_async(_ int: inout Int) async {
int += 1
}
public func increment(_ int: inout Int) {
int += 1
}
Reference in New Issue View Git Blame Copy Permalink