Files
swift-mirror/test/SILOptimizer/optimize_keypath.swift
Tim Kientzle 8eabeeb8ca [SE-0474] Read2/Modify2 => YieldingBorrow/YieldingMutate
This updates a large number of internal symbols, function names,
and types to match the final approved terminology.  Matching the
surface language terminology and the compiler internals should
make the code easier for people to understand into the future.
2026-01-03 16:05:12 -08:00

736 lines
23 KiB
Swift

// RUN: %empty-directory(%t)
// RUN: %target-swift-frontend -primary-file %s -O -sil-verify-all -Xllvm -sil-print-types -emit-sil >%t/output.sil
// RUN: %FileCheck %s < %t/output.sil
// RUN: %FileCheck -check-prefix=CHECK-ALL %s < %t/output.sil
// RUN: %target-swift-frontend -primary-file %s -O -sil-verify-all -swift-version 6 -Xllvm -sil-print-types -emit-sil >%t/output6.sil
// RUN: %FileCheck %s < %t/output6.sil
// RUN: %FileCheck -check-prefix=CHECK-ALL %s < %t/output6.sil
// RUN: %target-build-swift -O %s -o %t/a.out
// RUN: %target-run %t/a.out | %FileCheck %s -check-prefix=CHECK-OUTPUT -check-prefix=CHECK5-OUTPUT
// RUN: %target-build-swift -swift-version 6 -O %s -o %t/a6.out
// RUN: %target-run %t/a6.out | %FileCheck %s -check-prefix=CHECK-OUTPUT
// REQUIRES: executable_test,optimized_stdlib
// REQUIRES: CPU=arm64 || CPU=x86_64
// REQUIRES: swift_in_compiler
protocol P {
mutating func modifyIt()
var computed: Int { get set }
}
struct GenStruct<T : P> : P {
var st: T
var computed: Int { get { st.computed } set { st.computed = newValue } }
var computedGeneric: T { get { st} set { st = newValue} }
init(_ st: T) { self.st = st }
mutating func modifyIt() {
st.modifyIt()
}
}
#if !swift(>=6)
var numGenClassObjs = 0
#endif
final class GenClass<T : P> : P {
var ct: T
var computed: Int { get { ct.computed } set { ct.computed = newValue } }
var gs: GenStruct<T>
init(_ ct: T) {
self.ct = ct
self.gs = .init(ct)
#if !swift(>=6)
numGenClassObjs += 1
#endif
}
deinit {
#if !swift(>=6)
numGenClassObjs -= 1
#endif
}
func modifyIt() {
ct.modifyIt()
}
}
class Base<T> {
final var i: Int = 12
}
class DerivedClass<T> : Base<T> {
}
final class DerivedClass2 : DerivedClass<Int> {
}
final class SimpleClass : P {
var i: Int
#if !swift(>=6)
static var numObjs = 0
#endif
var tuple = (0, 1)
struct Nested {
var i: Int = 0
@inline(never)
var computedGenClass: GenClass<SimpleStruct> { GenClass(SimpleStruct(i: i)) }
}
var opt: Nested?
init(_ i: Int, nested: Int? = nil) {
self.i = i
self.opt = nested.map { Nested(i: $0) }
#if !swift(>=6)
Self.numObjs += 1
#endif
}
deinit {
#if !swift(>=6)
Self.numObjs -= 1
#endif
}
func modifyIt() {
i += 10
}
var computed: Int { get { i + 1 } set { i = newValue - 1} }
}
struct SimpleStruct: P {
var tuple = (0, 1)
struct Nested {
var i: Int
}
var opt: Nested?
struct Nested2 {
var opt: Nested?
}
var opt2: Nested2?
var i = 0
init(i: Int = 0) { self.i = i }
mutating func modifyIt() {
i += 10
}
var computed: Int { get { i + 1 } set { i = newValue - 1} }
}
// Check if all keypath instructions have been optimized away
// CHECK-ALL-NOT: = keypath
// CHECK-LABEL: sil {{.*}}testGenStructRead
// CHECK: [[A:%[0-9]+]] = struct_element_addr %1
// CHECK: copy_addr [[A]] to [init] %0
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenStructRead<T>(_ s: GenStruct<T>) -> T {
let kp = \GenStruct<T>.st
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testGenStructWrite
// CHECK: [[A:%[0-9]+]] = struct_element_addr %0
// CHECK: destroy_addr [[A]]
// CHECK: copy_addr {{.*}} to [init] [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenStructWrite<T>(_ s: inout GenStruct<T>, _ t: T) {
let kp = \GenStruct<T>.st
s[keyPath: kp] = t
}
// CHECK-LABEL: sil {{.*}}testGenClassRead
// CHECK: [[E:%[0-9]+]] = ref_element_addr %1
// CHECK: [[A:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E]]
// CHECK: copy_addr [[A]] to [init] %0
// CHECK: end_access [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenClassRead<T>(_ c: GenClass<T>) -> T {
let kp = \GenClass<T>.ct
return c[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testDerivedClassRead
// CHECK: [[C:%[0-9]+]] = upcast %0
// CHECK: [[E:%[0-9]+]] = ref_element_addr [[C]]
// CHECK: [[A:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E]]
// CHECK: [[V:%[0-9]+]] = load [[A]]
// CHECK: end_access [[A]]
// CHECK: return [[V]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testDerivedClassRead<T>(_ c: DerivedClass<T>) -> Int {
let kp = \DerivedClass<T>.i
return c[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testDerivedClass2Read
// CHECK: [[C:%[0-9]+]] = upcast %0
// CHECK: [[E:%[0-9]+]] = ref_element_addr [[C]]
// CHECK: [[A:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E]]
// CHECK: [[V:%[0-9]+]] = load [[A]]
// CHECK: end_access [[A]]
// CHECK: return [[V]]
@inline(never)
func testDerivedClass2Read(_ c: DerivedClass2) -> Int {
let kp = \DerivedClass2.i
return c[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testGenClassWrite
// CHECK: [[E:%[0-9]+]] = ref_element_addr %0
// CHECK: [[A:%[0-9]+]] = begin_access [modify] [dynamic] [[E]]
// CHECK: destroy_addr [[A]]
// CHECK: copy_addr %1 to [init] [[A]]
// CHECK: end_access [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenClassWrite<T>(_ c: GenClass<T>, _ t: T) {
let kp = \GenClass<T>.ct
c[keyPath: kp] = t
}
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func modifyGeneric<T : P>(_ t: inout T) {
t.modifyIt()
}
// CHECK-LABEL: sil {{.*}}testGenStructModify
// CHECK: [[A:%[0-9]+]] = struct_element_addr %0
// CHECK: [[F:%[0-9]+]] = function_ref {{.*}}modifyGeneric
// CHECK: apply [[F]]<T>([[A]])
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenStructModify<T : P>(_ s: inout GenStruct<T>) {
let kp = \GenStruct<T>.st
modifyGeneric(&s[keyPath: kp])
}
// CHECK-LABEL: sil {{.*}}testGenClassModify
// CHECK: [[E:%[0-9]+]] = ref_element_addr %0
// CHECK: [[A:%[0-9]+]] = begin_access [modify] [dynamic] [[E]]
// CHECK: [[F:%[0-9]+]] = function_ref {{.*}}modifyGeneric
// CHECK: apply [[F]]<T>([[A]])
// CHECK: end_access [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGenClassModify<T : P>(_ c: GenClass<T>) {
let kp = \GenClass<T>.ct
modifyGeneric(&c[keyPath: kp])
}
// CHECK-LABEL: sil {{.*}}testNestedRead1
// CHECK: [[R1:%[0-9]+]] = struct_extract %0
// CHECK: [[E1:%[0-9]+]] = ref_element_addr [[R1]]
// CHECK: [[A1:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E1]]
// CHECK: [[E2:%[0-9]+]] = struct_element_addr [[A1]]
// CHECK: [[R2:%[0-9]+]] = load [[E2]]
// CHECK: end_access [[A1]]
// CHECK: [[E3:%[0-9]+]] = ref_element_addr [[R2]]
// CHECK: [[A2:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E3]]
// CHECK: [[I:%[0-9]+]] = load [[A2]]
// CHECK: end_access [[A2]]
// CHECK: return [[I]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testNestedRead1(_ s: GenStruct<GenClass<GenStruct<SimpleClass>>>) -> Int {
let kp = \GenStruct<GenClass<GenStruct<SimpleClass>>>.st.ct.st.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testNestedYieldingBorrow
// CHECK: [[R:%[0-9]+]] = struct_extract %1
// CHECK: [[E1:%[0-9]+]] = ref_element_addr [[R]]
// CHECK: [[A:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E1]]
// CHECK: [[E2:%[0-9]+]] = struct_element_addr [[A]]
// CHECK: copy_addr [[E2]] to [init] %0
// CHECK: end_access [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testNestedYieldingBorrow<T>(_ s: GenStruct<GenClass<GenStruct<T>>>) -> T {
let kp = \GenStruct<GenClass<GenStruct<T>>>.st.ct.st
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testNestedWrite
// CHECK: [[E1:%[0-9]+]] = ref_element_addr %0
// CHECK: [[A1:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E1]]
// CHECK: [[E2:%[0-9]+]] = struct_element_addr [[A1]]
// CHECK: [[R1:%[0-9]+]] = load [[E2]]
// CHECK: end_access [[A1]]
// CHECK: [[E3:%[0-9]+]] = ref_element_addr [[R1]]
// CHECK: [[A2:%[0-9]+]] = begin_access [modify] [dynamic] [no_nested_conflict] [[E3]]
// CHECK: store %1 to [[A2]]
// CHECK: end_access [[A2]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testNestedWrite(_ s: GenClass<GenStruct<SimpleClass>>, _ i: Int) {
let kp = \GenClass<GenStruct<SimpleClass>>.ct.st.i
s[keyPath: kp] = i
}
// CHECK-LABEL: sil {{.*}}testNestedModify
// CHECK: [[E1:%[0-9]+]] = struct_element_addr %0
// CHECK: [[R1:%[0-9]+]] = load [[E1]]
// CHECK: [[E2:%[0-9]+]] = ref_element_addr [[R1]]
// CHECK: [[A1:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E2]]
// CHECK: [[R2:%[0-9]+]] = load [[A1]]
// CHECK: end_access [[A1]]
// CHECK: [[E3:%[0-9]+]] = ref_element_addr [[R2]]
// CHECK: [[A2:%[0-9]+]] = begin_access [modify] [dynamic] [[E3]]
// CHECK: [[F:%[0-9]+]] = function_ref {{.*}}modifyGeneric
// CHECK: apply [[F]]<T>([[A2]])
// CHECK: end_access [[A2]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testNestedModify<T : P>(_ s: inout GenStruct<GenClass<GenClass<T>>>) {
let kp = \GenStruct<GenClass<GenClass<T>>>.st.ct.ct
modifyGeneric(&s[keyPath: kp])
}
// CHECK-LABEL: sil {{.*}}testTuple
// CHECK: [[E:%[0-9]+]] = struct_element_addr
// CHECK: [[T1:%[0-9]+]] = tuple_element_addr [[E]]
// CHECK: [[I:%[0-9]+]] = load [[T1]]
// CHECK: [[T2:%[0-9]+]] = tuple_element_addr [[E]]
// CHECK: store [[I]] to [[T2]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testTuple(_ s: inout SimpleStruct) {
let first = \SimpleStruct.tuple.0
let second = \SimpleStruct.tuple.1
s[keyPath: first] = s[keyPath: second]
}
// CHECK-LABEL: sil {{.*}} [noinline] {{.*}}testGetter
// CHECK: [[A:%[0-9]+]] = alloc_stack $Int
// CHECK: [[F:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[F]]<T>([[A]], %0)
// destroy_addr gets optimized out
// CHECK: dealloc_stack [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGetter<T : P>(_ s: GenStruct<T>) -> Int {
let kp = \GenStruct<T>.computed
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}} [noinline] {{.*}}testClassMemberGetter
// CHECK: [[A:%[0-9]+]] = alloc_stack $Int
// CHECK: [[E:%[0-9]+]] = ref_element_addr
// CHECK: [[M:%[0-9]+]] = begin_access [read] [dynamic] [[E]]
// CHECK: [[F:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[F]]<T>([[A]], [[M]])
// CHECK: end_access
// destroy_addr gets optimized out
// CHECK: dealloc_stack [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testClassMemberGetter<T : P>(_ c: GenClass<T>) -> Int {
let kp = \GenClass<T>.gs.computed
return c[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testComputedModify
// CHECK: [[A:%[0-9]+]] = alloc_stack $Int
// CHECK: [[G:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[G]]<T>([[A]], %0)
// CHECK: store {{%[0-9]+}} to [[A]]
// CHECK: [[S:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[S]]<T>([[A]], %0)
// destroy_addr gets optimized out
// CHECK: dealloc_stack [[A]]
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testComputedModify<T : P>(_ s: inout GenStruct<T>) {
let kp = \GenStruct<T>.computed
s[keyPath: kp] += 10
}
// CHECK-LABEL: sil {{.*}}testComputedModify
// CHECK: [[E:%[0-9]+]] = ref_element_addr
// CHECK: [[M:%[0-9]+]] = begin_access [modify] [dynamic] [[E]]
// CHECK: [[A:%[0-9]+]] = alloc_stack $Int
// CHECK: [[G:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[G]]<T>([[A]], [[M]])
// CHECK: store {{%[0-9]+}} to [[A]]
// CHECK: [[S:%[0-9]+]] = function_ref {{.*}}computed
// CHECK: apply [[S]]<T>([[A]], [[M]])
// destroy_addr gets optimized out
// CHECK: dealloc_stack [[A]]
// CHECK: end_access
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testClassMemberComputedModify<T : P>(_ s: inout GenClass<T>) {
let kp = \GenClass<T>.gs.computed
s[keyPath: kp] += 10
}
// CHECK-LABEL: sil {{.*}}testModifyOptionalForce
// CHECK: [[F:%[0-9]+]] = select_enum [[O:%[0-9]+]]
// CHECK: cond_fail [[F]]
// CHECK: unchecked_enum_data [[O]]
// CHECK: [[E2:%[0-9]+]] = enum $Optional<SimpleStruct.Nested>
// CHECK: store [[E2]] to {{%[0-9]+}}
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testModifyOptionalForce(_ s: inout SimpleStruct) {
let kp = \SimpleStruct.opt!.i
s[keyPath: kp] += 10
}
// CHECK-LABEL: sil {{.*}}testModifyOptionalForceClass
// CHECK: [[O:%[0-9]+]] = ref_element_addr
// CHECK: begin_access [modify] [dynamic] [no_nested_conflict] [[O]]
// CHECK: [[F:%[0-9]+]] = select_enum
// CHECK: cond_fail [[F]]
// CHECK: unchecked_enum_data [[E1:%[0-9]+]]
// CHECK: [[E2:%[0-9]+]] = enum $Optional<SimpleClass.Nested>
// CHECK: store [[E2]] to {{%[0-9]+}}
// CHECK: end_access
// CHECK: return
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testModifyOptionalForceClass(_ s: inout SimpleClass) {
let kp = \SimpleClass.opt!.i
s[keyPath: kp] += 10
}
// CHECK-LABEL: sil {{.*}}testOptionalChain
// By the time the test gets run, lots of stack
// allocations have been promoted to registers.
//
// Check if value is null
// CHECK: switch_enum [[O:%[0-9]+]]
// CHECK: bb{{.*}}:
// Unwrap value
// CHECK: [[A1:%[0-9]+]] = alloc_stack
// CHECK: store [[O]] to [[A1]]
// CHECK: [[U:%[0-9]+]] = unchecked_take_enum_data_addr [[A1]]
// Access stored property & re-wrap result
// CHECK: [[I:%[0-9]+]] = struct_element_addr [[U]]
// CHECK: [[R1:%[0-9]+]] = enum
// CHECK: dealloc_stack [[A1]]
// CHECK: br [[CONTINUATION:bb.]]([[R1]] : $Optional<Int>)
// CHECK: {{bb.}}:
// Store nil in result
// CHECK: [[R2:%[0-9]+]] = enum
// CHECK: br [[CONTINUATION]]([[R2]] : $Optional<Int>)
// CHECK: [[CONTINUATION]]([[R:%[0-9]+]] : $Optional<Int>):
// CHECK: return [[R]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testOptionalChain(_ s: SimpleStruct) -> Int? {
let kp = \SimpleStruct.opt?.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testOptionalChainClass
// By the time the test gets run, lots of stack
// allocations have been promoted to registers.
//
// CHECK: [[E1:%[0-9]+]] = ref_element_addr
// CHECK: [[E2:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E1]]
// Check if value is null
// CHECK: switch_enum [[O:%[0-9]+]]
// CHECK: bb{{.*}}:
// Unwrap value
// CHECK: [[A1:%[0-9]+]] = alloc_stack
// CHECK: store [[O]] to [[A1]]
// CHECK: [[U:%[0-9]+]] = unchecked_take_enum_data_addr [[A1]]
// Access stored property & re-wrap result
// CHECK: [[I:%[0-9]+]] = struct_element_addr [[U]]
// CHECK: [[R1:%[0-9]+]] = enum
// CHECK: dealloc_stack [[A1]]
// CHECK: br [[CONTINUATION:bb.]]([[R1]] : $Optional<Int>)
// CHECK: {{bb.}}:
// Store nil in result
// CHECK: [[R2:%[0-9]+]] = enum
// CHECK: br [[CONTINUATION]]([[R2]] : $Optional<Int>)
// CHECK: [[CONTINUATION]]([[R:%[0-9]+]] : $Optional<Int>):
// CHECK: end_access [[E2]]
// CHECK: return [[R]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testOptionalChainClass(_ s: SimpleClass) -> Int? {
let kp = \SimpleClass.opt?.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testNestedOptionalChain
// By the time the test gets run, lots of stack
// allocations have been promoted to registers.
//
// Check if value is null
// CHECK: switch_enum [[O:%[0-9]+]]
// CHECK: bb{{.*}}:
// Unwrap value
// CHECK: [[A1:%[0-9]+]] = alloc_stack
// CHECK: store [[O]] to [[A1]]
// CHECK: [[U:%[0-9]+]] = unchecked_take_enum_data_addr [[A1]]
//
// Unwrap nested optional
// CHECK: switch_enum [[O2:%[0-9]+]]
// CHECK: bb{{.*}}:
// CHECK: [[A2:%[0-9]+]] = alloc_stack
// CHECK: store [[O2]] to [[A2]]
// CHECK: [[U2:%[0-9]+]] = unchecked_take_enum_data_addr [[A2]]
// Access stored property & re-wrap result
// CHECK: [[I:%[0-9]+]] = struct_element_addr [[U2]]
// CHECK: [[R1:%[0-9]+]] = enum
// CHECK: dealloc_stack [[A2]]
// CHECK: br [[CONT2:bb.]]([[R1]] : $Optional<Int>
// CHECK: {{bb.}}:
// Store nil in result
// CHECK: [[R2:%[0-9]+]] = enum
// CHECK: br [[CONT2]]([[R2]] : $Optional<Int>
// CHECK: [[CONT2]]([[R3:%[0-9]+]] : $Optional<Int>):
// CHECK: dealloc_stack [[A1]]
// CHECK: br [[CONT1:bb.]]([[R3]] : $Optional<Int>)
// CHECK: {{bb.}}:
// Store nil in result
// CHECK: [[R2:%[0-9]+]] = enum
// CHECK: br [[CONT1]]([[R2]] : $Optional<Int>)
// CHECK: [[CONT1]]([[R:%[0-9]+]] : $Optional<Int>):
// CHECK: return [[R]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testNestedOptionalChain(_ s: SimpleStruct) -> Int? {
let kp = \SimpleStruct.opt2?.opt?.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testGetOptionalForce
// CHECK: [[F:%[0-9]+]] = select_enum [[O:%[0-9]+]]
// CHECK: cond_fail [[F]]
// CHECK: [[A:%[0-9]+]] = alloc_stack
// CHECK: store [[O]] to [[A]]
// CHECK: [[E2:%[0-9]+]] = unchecked_take_enum_data_addr [[A]]
// CHECK: [[E3:%[0-9]+]] = struct_element_addr [[E2]]
// CHECK: [[I:%[0-9]+]] = load [[E3]]
// CHECK: dealloc_stack [[A]]
// CHECK: return [[I]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGetOptionalForce(_ s: SimpleStruct) -> Int {
let kp = \SimpleStruct.opt!.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testGetOptionalForceClass
// CHECK: [[R1:%[0-9]+]] = ref_element_addr
// CHECK: [[R2:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[R1]]
// CHECK: [[F:%[0-9]+]] = select_enum [[O:%[0-9]+]]
// CHECK: cond_fail [[F]]
// CHECK: [[A:%[0-9]+]] = alloc_stack
// CHECK: store [[O]] to [[A]]
// CHECK: [[E2:%[0-9]+]] = unchecked_take_enum_data_addr [[A]]
// CHECK: [[E3:%[0-9]+]] = struct_element_addr [[E2]]
// CHECK: [[I:%[0-9]+]] = load [[E3]]
// CHECK: dealloc_stack [[A]]
// CHECK: end_access [[R2]]
// CHECK: return [[I]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGetOptionalForceClass(_ s: SimpleClass) -> Int {
let kp = \SimpleClass.opt!.i
return s[keyPath: kp]
}
// CHECK-LABEL: sil {{.*}}testGetComplex
// opt
// CHECK: [[E1:%[0-9]+]] = ref_element_addr
// CHECK: [[B1:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E1]]
// !
// CHECK: [[F:%[0-9]+]] = select_enum [[O:%[0-9]+]]
// CHECK: cond_fail [[F]]
// computedGenClass
// CHECK: [[F:%[0-9]+]] = function_ref
// CHECK: apply [[F]]
// CHECK: end_access [[B1]]
// ct
// CHECK: [[E3:%[0-9]+]] = ref_element_addr
// CHECK: [[B2:%[0-9]+]] = begin_access [read] [dynamic] [no_nested_conflict] [[E3]]
// tuple
// CHECK: [[E4:%[0-9]+]] = struct_element_addr
// 0
// CHECK: [[E5:%[0-9]+]] = tuple_element_addr [[E4]]
// CHECK: load [[E5]]
// CHECK: end_access [[B2]]
@inline(never)
@_semantics("optimize.sil.specialize.generic.never")
func testGetComplex(_ s: SimpleClass) -> Int {
let kp = \SimpleClass.opt!.computedGenClass.ct.tuple.1
return s[keyPath: kp]
}
// allow exactly one unoptimized key path instruction, in this function
// CHECK-ALL: sil {{.*}}makeKeyPathInGenericContext
// CHECK-ALL: = keypath
func makeKeyPathInGenericContext<T: P>(of: T.Type) -> WritableKeyPath<GenStruct<T>, T> {
\.computedGeneric
}
// CHECK-ALL-NOT: = keypath
func testGenericResult(_ s: inout GenStruct<SimpleStruct>) {
let kp = makeKeyPathInGenericContext(of: SimpleStruct.self)
s[keyPath: kp].i += 1
}
func testit() {
// CHECK-OUTPUT: GenStructRead: 27
print("GenStructRead: \(testGenStructRead(GenStruct(SimpleClass(27))).i)")
// CHECK-OUTPUT: GenStructWrite: 28
var s = GenStruct(SimpleClass(0))
testGenStructWrite(&s, SimpleClass(28))
print("GenStructWrite: \(s.st.i)")
// CHECK-OUTPUT: GenStructModify: 38
testGenStructModify(&s)
print("GenStructModify: \(s.st.i)")
// CHECK-OUTPUT: GenClassRead: 29
print("GenClassRead: \(testGenClassRead(GenClass(SimpleClass(29))).i)")
// CHECK-OUTPUT: DerivedClassRead: 12
print("DerivedClassRead: \(testDerivedClassRead(DerivedClass<Int>())))")
// CHECK-OUTPUT: DerivedClass2Read: 12
print("DerivedClass2Read: \(testDerivedClass2Read(DerivedClass2())))")
// CHECK-OUTPUT: GenClassWrite: 30
let c = GenClass(SimpleClass(0))
testGenClassWrite(c, SimpleClass(30))
print("GenClassWrite: \(c.ct.i)")
// CHECK-OUTPUT: GenClassModify: 40
testGenClassModify(c)
print("GenClassModify: \(c.ct.i)")
// CHECK-OUTPUT: NestedRead1: 31
print("NestedRead1: \(testNestedRead1(GenStruct(GenClass(GenStruct(SimpleClass(31))))))")
// CHECK-OUTPUT: NestedYieldingBorrow: 32
print("NestedYieldingBorrow: \(testNestedYieldingBorrow(GenStruct(GenClass(GenStruct(SimpleClass(32))))).i)")
// CHECK-OUTPUT: NestedWrite: 33
let c2 = GenClass(GenStruct(SimpleClass(0)))
testNestedWrite(c2, 33)
print("NestedWrite: \(c2.ct.st.i)")
// CHECK-OUTPUT: NestedModify: 44
var s2 = GenStruct(GenClass(GenClass(SimpleClass(34))))
testNestedModify(&s2)
print("NestedModify: \(s2.st.ct.ct.i)")
// CHECK-OUTPUT: Getter: 51
var s3 = GenStruct(SimpleClass(50))
print("Getter: \(testGetter(s3))")
// CHECK-OUTPUT: ClassMemberGetter: 52
var c3 = GenClass(SimpleClass(51))
print("ClassMemberGetter: \(testClassMemberGetter(c3))")
// CHECK-OUTPUT: ComputedModify: 61
testComputedModify(&s3)
print("ComputedModify: \(s3.computed)")
// CHECK-OUTPUT: ClassComputedModify: 62
testClassMemberComputedModify(&c3)
print("ClassComputedModify: \(c3.computed)")
var s4 = SimpleStruct()
// CHECK-OUTPUT: Tuple: 1
testTuple(&s4)
print("Tuple: \(s4.tuple.0)")
var c4 = SimpleClass(0)
// CHECK-OUTPUT: OptionalChain1: nil
print("OptionalChain1: \(String(describing: testOptionalChain(s4)))")
// CHECK-OUTPUT: ClassOptionalChain1: nil
print("ClassOptionalChain1: \(String(describing: testOptionalChainClass(c4)))")
// CHECK-OUTPUT: OptionalChain2: Optional(70)
s4.opt = .init(i: 70)
print("OptionalChain2: \(String(describing: testOptionalChain(s4)))")
// CHECK-OUTPUT: ClassOptionalChain2: Optional(71)
c4.opt = .init(i: 71)
print("ClassOptionalChain2: \(String(describing: testOptionalChainClass(c4)))")
// CHECK-OUTPUT: OptionalForce: 80
testModifyOptionalForce(&s4)
print("OptionalForce: \(testGetOptionalForce(s4))")
// CHECK-OUTPUT: ClassOptionalForce: 81
testModifyOptionalForceClass(&c4)
print("ClassOptionalForce: \(testGetOptionalForceClass(c4))")
// CHECK-OUTPUT: NestedOptionalChain1: nil
print("NestedOptionalChain1: \(String(describing: testNestedOptionalChain(s4)))")
// CHECK-OUTPUT: NestedOptionalChain2: nil
s4.opt2 = .init()
print("NestedOptionalChain2: \(String(describing: testNestedOptionalChain(s4)))")
// CHECK-OUTPUT: NestedOptionalChain3: Optional(90)
s4.opt2!.opt = .init(i: 90)
print("NestedOptionalChain3: \(String(describing: testNestedOptionalChain(s4)))")
// CHECK-OUTPUT: testGetComplex: 1
print("testGetComplex: \(testGetComplex(c4))")
// CHECK-OUTPUT: testGenericResult: 2
var s5 = GenStruct(SimpleStruct(i: 1))
testGenericResult(&s5)
print("testGenericResult: \(s5.st.i)")
}
testit()
#if !swift(>=6)
// CHECK5-OUTPUT: SimpleClass obj count: 0
print("SimpleClass obj count: \(SimpleClass.numObjs)")
// CHECK5-OUTPUT: GenClass obj count: 0
print("GenClass obj count: \(numGenClassObjs)")
#endif