Merge pull request #7121 from airspeedswift/map-reduce-benchmarks

[benchmarks] add more tests of map/reduce

Author: airspeedswift

benchmark/single-source/MapReduce.swift

@@ -19,9 +19,139 @@ public func run_MapReduce(_ N: Int) {
 
   var c = 0
   for _ in 1...N*100 {
-    numbers = numbers.map({$0 &+ 5})
+    numbers = numbers.map { $0 &+ 5 }
     c += numbers.reduce(0, &+)
   }
   CheckResults(c != 0, "IncorrectResults in MapReduce")
 }
 
+@inline(never)
+public func run_MapReduceAnyCollection(_ N: Int) {
+  let numbers = AnyCollection([Int](0..<1000))
+
+  var c = 0
+  for _ in 1...N*100 {
+    let mapped = numbers.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceAnyCollectionShort(_ N: Int) {
+  let numbers = AnyCollection([Int](0..<10))
+
+  var c = 0
+  for _ in 1...N*10000 {
+    let mapped = numbers.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceShort(_ N: Int) {
+  var numbers = [Int](0..<10)
+
+  var c = 0
+  for _ in 1...N*10000 {
+    numbers = numbers.map { $0 &+ 5 }
+    c += numbers.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceSequence(_ N: Int) {
+  let numbers = sequence(first: 0) { $0 < 1000 ? $0 &+ 1 : nil }
+
+  var c = 0
+  for _ in 1...N*100 {
+    let mapped = numbers.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceLazySequence(_ N: Int) {
+  let numbers = sequence(first: 0) { $0 < 1000 ? $0 &+ 1 : nil }
+
+  var c = 0
+  for _ in 1...N*100 {
+    let mapped = numbers.lazy.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceLazyCollection(_ N: Int) {
+  let numbers = [Int](0..<1000)
+
+  var c = 0
+  for _ in 1...N*100 {
+    let mapped = numbers.lazy.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceLazyCollectionShort(_ N: Int) {
+  let numbers = [Int](0..<10)
+
+  var c = 0
+  for _ in 1...N*10000 {
+    let mapped = numbers.lazy.map { $0 &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceString(_ N: Int) {
+  let s = "thequickbrownfoxjumpsoverthelazydogusingasmanycharacteraspossible123456789"
+
+  var c: UInt64 = 0
+  for _ in 1...N*100 {
+    c += s.utf8.map { UInt64($0 &+ 5) }.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceShortString(_ N: Int) {
+  let s = "12345"
+
+  var c: UInt64 = 0
+  for _ in 1...N*100 {
+    c += s.utf8.map { UInt64($0 &+ 5) }.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceClass(_ N: Int) {
+  let numbers = (0..<1000).map { NSDecimalNumber(value: $0) }
+
+  var c = 0
+  for _ in 1...N*100 {
+    let mapped = numbers.map { $0.intValue &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+
+@inline(never)
+public func run_MapReduceClassShort(_ N: Int) {
+  let numbers = (0..<10).map { NSDecimalNumber(value: $0) }
+
+  var c = 0
+  for _ in 1...N*10000 {
+    let mapped = numbers.map { $0.intValue &+ 5 }
+    c += mapped.reduce(0, &+)
+  }
+  CheckResults(c != 0, "IncorrectResults in MapReduce")
+}
+

benchmark/utils/main.swift

@@ -168,6 +168,16 @@ precommitTests = [
   "Join": run_Join,
   "LinkedList": run_LinkedList,
   "MapReduce": run_MapReduce,
+  "MapReduceAnyCollection": run_MapReduceAnyCollection,
+  "MapReduceShort": run_MapReduceShort,
+  "MapReduceSequence": run_MapReduceSequence,
+  "MapReduceLazySequence": run_MapReduceLazySequence,
+  "MapReduceLazyCollection": run_MapReduceLazyCollection,
+  "MapReduceLazyCollectionShort": run_MapReduceLazyCollectionShort,
+  "MapReduceString": run_MapReduceString,
+  "MapReduceShortString": run_MapReduceShortString,
+  "MapReduceClass": run_MapReduceClass,
+  "MapReduceClassShort": run_MapReduceClassShort,
   "Memset": run_Memset,
   "MonteCarloE": run_MonteCarloE,
   "MonteCarloPi": run_MonteCarloPi,