[Concurrency] Enable TaskGroup/DiscardingTaskGroup in Embedded Swift

This commit is contained in:
Kuba Mracek
2024-09-28 17:52:05 -07:00
parent f787e8f173
commit f1c2334c9a
11 changed files with 249 additions and 33 deletions

View File

@@ -0,0 +1,61 @@
// RUN: %empty-directory(%t)
// RUN: %target-swift-frontend -enable-experimental-feature Embedded -target %target-cpu-apple-macos14 -parse-as-library %s -c -o %t/a.o
// RUN: %target-clang -x c -c %S/Inputs/print.c -o %t/print.o
// RUN: %target-clang %t/a.o %t/print.o -o %t/a.out %swift_obj_root/lib/swift/embedded/%target-cpu-apple-macos/libswift_Concurrency.a -dead_strip
// RUN: %target-run %t/a.out | %FileCheck %s
// REQUIRES: executable_test
// REQUIRES: swift_in_compiler
// REQUIRES: optimized_stdlib
// REQUIRES: OS=macosx
import _Concurrency
protocol Go: Actor {
var name: String { get }
func go(times: Int) async -> Int
}
extension Go {
func go(times: Int) async -> Int {
for i in 0..<times {
print("\(name) @ \(i)")
await Task.yield()
}
return times
}
}
actor One: Go { var name = "One" }
actor Two: Go { var name = "Two" }
func yielding() async {
let one = One()
let two = Two()
await withDiscardingTaskGroup(returning: Void.self) { group in
group.addTask {
await one.go(times: 5)
}
group.addTask {
await two.go(times: 5)
}
}
}
@main struct Main {
static func main() async {
await yielding()
print("All done!")
// CHECK: One @ 0
// CHECK: Two @ 0
// CHECK: One @ 1
// CHECK: Two @ 1
// CHECK: One @ 2
// CHECK: Two @ 2
// CHECK: One @ 3
// CHECK: Two @ 3
// CHECK: One @ 4
// CHECK: Two @ 4
// CHECK: All done!
}
}