actor_counters.swift: allow specifying of parameters

Julian Lettner · Julian Lettner · commit e023695bbc16 · 2021-03-01T13:35:03.000-08:00
This is useful during debugging when manually running the test.
diff --git a/test/Concurrency/Runtime/actor_counters.swift b/test/Concurrency/Runtime/actor_counters.swift
@@ -69,6 +69,12 @@ func runTest(numCounters: Int, numWorkers: Int, numIterations: Int) async {
 
 @main struct Main {
   static func main() async {
-    await runTest(numCounters: 10, numWorkers: 100, numIterations: 1000)
+    // Useful for debugging: specify counter/worker/iteration counts
+    let args = CommandLine.arguments
+    let counters = args.count >= 2 ? Int(args[1])! : 10
+    let workers = args.count >= 3 ? Int(args[2])! : 100
+    let iterations = args.count >= 4 ? Int(args[3])! : 1000
+    print("counters: \(counters), workers: \(workers), iterations: \(iterations)")
+    await runTest(numCounters: counters, numWorkers: workers, numIterations: iterations)
   }
 }
diff --git a/test/Sanitizers/tsan/actor_counters.swift b/test/Sanitizers/tsan/actor_counters.swift
@@ -70,6 +70,12 @@ func runTest(numCounters: Int, numWorkers: Int, numIterations: Int) async {
 
 @main struct Main {
   static func main() async {
-    await runTest(numCounters: 10, numWorkers: 100, numIterations: 1000)
+    // Useful for debugging: specify counter/worker/iteration counts
+    let args = CommandLine.arguments
+    let counters = args.count >= 2 ? Int(args[1])! : 10
+    let workers = args.count >= 3 ? Int(args[2])! : 100
+    let iterations = args.count >= 4 ? Int(args[3])! : 1000
+    print("counters: \(counters), workers: \(workers), iterations: \(iterations)")
+    await runTest(numCounters: counters, numWorkers: workers, numIterations: iterations)
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -69,6 +69,12 @@ func runTest(numCounters: Int, numWorkers: Int, numIterations: Int) async {`
`69`	`69`
`70`	`70`	`@main struct Main {`
`71`	`71`	`static func main() async {`
`72`		`- await runTest(numCounters: 10, numWorkers: 100, numIterations: 1000)`
	`72`	`+ // Useful for debugging: specify counter/worker/iteration counts`
	`73`	`+ let args = CommandLine.arguments`
	`74`	`+ let counters = args.count >= 2 ? Int(args[1])! : 10`
	`75`	`+ let workers = args.count >= 3 ? Int(args[2])! : 100`
	`76`	`+ let iterations = args.count >= 4 ? Int(args[3])! : 1000`
	`77`	`+ print("counters: \(counters), workers: \(workers), iterations: \(iterations)")`
	`78`	`+ await runTest(numCounters: counters, numWorkers: workers, numIterations: iterations)`
`73`	`79`	`}`
`74`	`80`	`}`
Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,12 @@ func runTest(numCounters: Int, numWorkers: Int, numIterations: Int) async {`
`70`	`70`
`71`	`71`	`@main struct Main {`
`72`	`72`	`static func main() async {`
`73`		`- await runTest(numCounters: 10, numWorkers: 100, numIterations: 1000)`
	`73`	`+ // Useful for debugging: specify counter/worker/iteration counts`
	`74`	`+ let args = CommandLine.arguments`
	`75`	`+ let counters = args.count >= 2 ? Int(args[1])! : 10`
	`76`	`+ let workers = args.count >= 3 ? Int(args[2])! : 100`
	`77`	`+ let iterations = args.count >= 4 ? Int(args[3])! : 1000`
	`78`	`+ print("counters: \(counters), workers: \(workers), iterations: \(iterations)")`
	`79`	`+ await runTest(numCounters: counters, numWorkers: workers, numIterations: iterations)`
`74`	`80`	`}`
`75`	`81`	`}`