test_aggregation_throughput_per_num_xmss

Author: TomWambsgans

crates/rec_aggregation/src/benchmark.rs

@@ -304,7 +304,7 @@ pub fn run_aggregation_benchmark(topology: &AggregationTopology, overlap: usize,
 
     let n_sigs = count_signers(topology, overlap);
 
-    let cache = read_benchmark_signers_cache();
+    let cache = get_benchmark_signers_cache();
     assert!(cache.len() >= n_sigs);
     let paired: Vec<_> = (0..n_sigs)
         .into_par_iter()
@@ -334,3 +334,50 @@ pub fn run_aggregation_benchmark(topology: &AggregationTopology, overlap: usize,
     let message = message_for_benchmark();
     crate::verify_aggregation(&aggregated_sigs, &message, BENCHMARK_SLOT).unwrap();
 }
+
+#[test]
+#[ignore]
+fn test_aggregation_throughput_per_num_xmss() {
+    let log_inv_rate = 1;
+    precompute_dft_twiddles::<F>(1 << 24);
+    init_aggregation_bytecode();
+    let _ = get_aggregation_bytecode();
+    let mut num_xmss_and_time = vec![];
+    let mut indexes = vec![];
+    for i in 1..100 {
+        indexes.push(i * 10);
+    }
+    for i in 50..100 {
+        indexes.push(i * 20);
+    }
+    for i in 40..60 {
+        indexes.push(i * 50);
+    }
+    for num_xmss in indexes {
+        let topology = AggregationTopology {
+            raw_xmss: num_xmss,
+            children: vec![],
+            log_inv_rate,
+        };
+        let instant = Instant::now();
+        run_aggregation_benchmark(&topology, 0, false);
+        let elapsed = instant.elapsed();
+        num_xmss_and_time.push((num_xmss, elapsed.as_secs_f64()));
+        println!(
+            "{} XMSS -> {} XMSS/s",
+            num_xmss,
+            (num_xmss as f64 / elapsed.as_secs_f64()).round() as usize
+        );
+
+        std::thread::sleep(std::time::Duration::from_millis(1000));
+
+        let mut csv = String::from("num_sigs,throughput (XMSS/s)\n");
+        for &(n, t) in &num_xmss_and_time {
+            csv.push_str(&format!("{},{:.1}\n", n, n as f64 / t));
+        }
+        let path = std::path::PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("benchmarks/xmss_throughput.csv");
+        std::fs::create_dir_all(path.parent().unwrap()).unwrap();
+        std::fs::write(&path, &csv).unwrap();
+        println!("\nWrote {}", path.display());
+    }
+}

crates/xmss/src/signers_cache.rs

@@ -1,11 +1,18 @@
 use std::path::PathBuf;
+use std::sync::OnceLock;
 
 use backend::*;
 use rand::rngs::StdRng;
 use rand::{Rng, SeedableRng};
 
 use crate::*;
 
+static SIGNERS_CACHE: OnceLock<Vec<[F; RANDOMNESS_LEN_FE]>> = OnceLock::new();
+
+pub fn get_benchmark_signers_cache() -> &'static Vec<[F; RANDOMNESS_LEN_FE]> {
+    SIGNERS_CACHE.get_or_init(read_benchmark_signers_cache)
+}
+
 pub const BENCHMARK_SLOT: u32 = 1111;
 
 pub fn message_for_benchmark() -> [F; MESSAGE_LEN_FE] {