exchange_bench: Input data as batch

Signed-off-by: Moritz Hoffmann <antiguru@gmail.com>
TimelyDataflow · Nov 20, 2021 · fa63c3a · fa63c3a
1 parent 92ca760
commit fa63c3a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/experiments/benches/exchange_bench.rs b/experiments/benches/exchange_bench.rs
@@ -77,10 +77,13 @@ fn experiment_exchange(
         let mut time = 0;
         let timer = Instant::now();
 
+        let buffer = (0..batch).collect();
+        let mut copy = Vec::new();
+
         for _round in 0..rounds {
-            for i in 0..batch {
-                input.send(i);
-            }
+            copy.clone_from(&buffer);
+            input.send_batch(&mut copy);
+            copy.clear();
             time += 1;
             input.advance_to(time);
             while probe.less_than(input.time()) {