[Concurrency] Add more tests for optional isolated parameters.

Author: hborla

test/Concurrency/optional_isolated_parameters.swift

@@ -0,0 +1,27 @@
+// RUN: %empty-directory(%t)
+// RUN: %target-build-swift -Xfrontend -disable-availability-checking -enable-experimental-feature OptionalIsolatedParameters %s -o %t/main
+// RUN: %target-codesign %t/main
+// RUN: %target-run %t/main | %FileCheck %s
+
+// REQUIRES: executable_test
+// REQUIRES: concurrency
+
+actor MyActor: CustomStringConvertible {
+  let description = "MyActor"
+}
+
+func optionalIsolated(to actor: isolated (any Actor)?) {
+  actor?.assertIsolated()
+  if let actor {
+    print("isolated to \(actor)")
+  } else {
+    print("nonisolated")
+  }
+}
+
+// CHECK: nonisolated
+// CHECK: isolated to Swift.MainActor
+// CHECK: isolated to MyActor
+optionalIsolated(to: nil)
+await optionalIsolated(to: MainActor.shared)
+await optionalIsolated(to: MyActor())

test/SILGen/isolated_parameters.swift

@@ -1,4 +1,4 @@
-// RUN: %target-swift-frontend -emit-silgen %s -module-name test -swift-version 5 | %FileCheck %s
+// RUN: %target-swift-frontend -emit-silgen -enable-experimental-feature OptionalIsolatedParameters %s -module-name test -swift-version 5 | %FileCheck %s
 // REQUIRES: concurrency
 
 @available(SwiftStdlib 5.1, *)
@@ -34,3 +34,34 @@ public func testIsolatedExistential(_ a: isolated Actor) async {
   // CHECK: hop_to_executor [[ACTOR_BORROW]] : $any Actor
   // CHECK: return
 }
+
+@available(SwiftStdlib 5.1, *)
+nonisolated func suspend() async {}
+
+// CHECK-LABEL: sil{{.*}} [ossa] @$s4test0A16OptionalIsolatedyyAA1ACSgYiYaF
+// CHECK: bb0([[ACTOR:%.*]] : @guaranteed $Optional<A>)
+@available(SwiftStdlib 5.1, *)
+public func testOptionalIsolated(_ a: isolated A?) async {
+  await suspend()
+  // CHECK: [[ACTOR_COPY:%.*]] = copy_value [[ACTOR]] : $Optional<A>
+  // CHECK: [[ACTOR_BORROW:%.*]] = begin_borrow [[ACTOR_COPY]] : $Optional<A>
+  // CHECK: hop_to_executor [[ACTOR_BORROW]] : $Optional<A>
+  // CHECK: [[SUSPEND:%.*]] = function_ref @$s4test7suspendyyYaF : $@convention(thin) @async () -> ()
+  // CHECK: apply [[SUSPEND]]() : $@convention(thin) @async () -> ()
+  // CHECK: hop_to_executor [[ACTOR_BORROW]] : $Optional<A>
+  // CHECK: return
+}
+
+// CHECK-LABEL: sil{{.*}} [ossa] @$s4test0A27OptionalIsolatedExistentialyyScA_pSgYiYaF
+// CHECK: bb0([[ACTOR:%.*]] : @guaranteed $Optional<any Actor>)
+@available(SwiftStdlib 5.1, *)
+public func testOptionalIsolatedExistential(_ a: isolated (any Actor)?) async {
+  await suspend()
+  // CHECK: [[ACTOR_COPY:%.*]] = copy_value [[ACTOR]] : $Optional<any Actor>
+  // CHECK: [[ACTOR_BORROW:%.*]] = begin_borrow [[ACTOR_COPY]] : $Optional<any Actor>
+  // CHECK: hop_to_executor [[ACTOR_BORROW]] : $Optional<any Actor>
+  // CHECK: [[SUSPEND:%.*]] = function_ref @$s4test7suspendyyYaF : $@convention(thin) @async () -> ()
+  // CHECK: apply [[SUSPEND]]() : $@convention(thin) @async () -> ()
+  // CHECK: hop_to_executor [[ACTOR_BORROW]] : $Optional<any Actor>
+  // CHECK: return
+}

test/SILOptimizer/lower_hop_to_actor.sil

@@ -260,3 +260,33 @@ bb0(%0 : @guaranteed $Optional<MyActor>):
   %r = tuple ()
   return %r : $()
 }
+
+// CHECK-LABEL: sil [ossa] @optional_existential_actor :
+sil [ossa] @optional_existential_actor : $@async (@guaranteed Optional<any Actor>) -> () {
+bb0(%0 : @guaranteed $Optional<any Actor>):
+// CHECK:      bb0(%0 : @guaranteed $Optional<any Actor>):
+// CHECK-NEXT: switch_enum %0 : $Optional<any Actor>, case #Optional.some!enumelt: bb1, default bb2
+
+// CHECK: bb1([[ACTOR:%.*]] : @guaranteed $any Actor):
+// CHECK-NEXT: [[OPENED:%.*]] = open_existential_ref [[ACTOR]] : $any Actor to $@opened("[[ID:.*]]", any Actor) Self
+// CHECK-NEXT: [[WITNESS:%.*]] = witness_method $@opened("[[ID]]", any Actor) Self, #Actor.unownedExecutor!getter : <Self where Self : Actor> (Self) -> () -> UnownedSerialExecutor
+// CHECK-NEXT: [[SERIAL_EXEC:%.*]] = apply [[WITNESS]]<@opened("[[ID]]", any Actor) Self>([[OPENED]])
+// CHECK-NEXT: [[EXECUTOR:%.*]] = struct_extract [[SERIAL_EXEC]] : $UnownedSerialExecutor, #UnownedSerialExecutor.executor
+// CHECK-NEXT: [[SOME:%.*]] = enum $Optional<Builtin.Executor>, #Optional.some!enumelt, [[EXECUTOR]] : $Builtin.Executor
+// CHECK-NEXT: br bb3([[SOME]] : $Optional<Builtin.Executor>)
+
+// CHECK: bb2:
+// CHECK-NEXT: [[NONE:%.*]] = enum $Optional<Builtin.Executor>, #Optional.none!enumelt
+// CHECK-NEXT: br bb3([[NONE]] : $Optional<Builtin.Executor>)
+
+// CHECK: bb3([[OPT_EXEC:%.*]] : $Optional<Builtin.Executor>):
+// CHECK-NEXT:   [[DEPEND:%.*]] = mark_dependence [[OPT_EXEC]] : $Optional<Builtin.Executor> on %0 : $Optional<any Actor>
+// CHECK-NEXT:   hop_to_executor [[DEPEND]] : $Optional<Builtin.Executor>
+// CHECK-NEXT:   hop_to_executor [[DEPEND]] : $Optional<Builtin.Executor>
+// CHECK-NEXT:   [[RET:%.*]] = tuple ()
+// CHECK-NEXT:   return [[RET]] : $()
+  hop_to_executor %0 : $Optional<any Actor>
+  hop_to_executor %0 : $Optional<any Actor>
+  %r = tuple ()
+  return %r : $()
+}