Merge pull request swiftlang#36586 from jckarter/owned-task-spawn-closures

Allow closures to be `__owned` or `__shared`, and make task spawning closures `__owned`

Author: jckarter

lib/Sema/TypeCheckType.cpp

@@ -3481,8 +3481,9 @@ TypeResolver::resolveSpecifierTypeRepr(SpecifierTypeRepr *repr,
     return ErrorType::get(getASTContext());
   }
 
-  if (!isa<ImplicitlyUnwrappedOptionalTypeRepr>(repr->getBase())) {
-    // Anything within the inout isn't a parameter anymore.
+  if (isa<InOutTypeRepr>(repr)
+      && !isa<ImplicitlyUnwrappedOptionalTypeRepr>(repr->getBase())) {
+    // Anything within an inout isn't a parameter anymore.
     options.setContext(None);
   }
 

stdlib/public/Concurrency/Task.swift

@@ -377,7 +377,7 @@ extension Task {
   @discardableResult
   public static func runDetached<T>(
     priority: Priority = .default,
-    operation: @Sendable @escaping () async -> T
+    operation: __owned @Sendable @escaping () async -> T
     // TODO: Allow inheriting task-locals?
   ) -> Handle<T, Never> {
     // Set up the job flags for a new task.
@@ -430,7 +430,7 @@ extension Task {
   @discardableResult
   public static func runDetached<T, Failure>(
     priority: Priority = .default,
-    operation: @Sendable @escaping () async throws -> T
+    operation: __owned @Sendable @escaping () async throws -> T
   ) -> Handle<T, Failure> {
     // Set up the job flags for a new task.
     var flags = JobFlags()

stdlib/public/Concurrency/TaskGroup.swift

@@ -124,7 +124,7 @@ extension Task {
     @discardableResult
     public mutating func add(
       overridingPriority priorityOverride: Priority? = nil,
-      operation: @Sendable @escaping () async throws -> TaskResult
+      operation: __owned @Sendable @escaping () async throws -> TaskResult
     ) async -> Bool {
       let canAdd = _taskGroupAddPendingTask(group: _group)