add iai benchmark

.github/workflows/bench.yml

@@ -13,10 +13,14 @@ jobs:
         uses: actions/checkout@v4
       - name: Install Rust
         uses: actions-rust-lang/setup-rust-toolchain@v1
-      - name: Run benchmarks
-        run: cargo bench
+      - name: Run criterion benchmarks
+        run: cargo bench criterion
         env:
           AOC_SESSION: ${{secrets.AOC_SESSION}}
+      # Run iai after criterion to ensure private solutions have been downloaded,
+      # since the iai version cannot be built without private inputs.
+      - name: Run iai
+        run: cargo bench iai --features iai-bench
       - name: Publish results
         id: deployment
         uses: actions/upload-pages-artifact@v3

Cargo.toml

@@ -1,3 +1,6 @@
+[workspace]
+members = [".", "macros"]
+
 [package]
 name = "aoc2024"
 version = "0.1.0"
@@ -7,17 +10,29 @@ edition = "2021"
 anyhow = "1.0.93"
 bitvec = "1.0.1"
 clap = { version = "4.5.21", features = ["derive"] }
-criterion = "0.5.1"
 itertools = "0.13.0"
 jq-rs = { version = "0.4.1", features = ["bundled"] }
 paste = "1.0.15"
 reqwest = { version = "0.12.9", features = ["blocking"] }
 rustc-hash = "2.1.0"
 simd-json = "0.14.3"
+macros = { path = "macros" }
 
 [patch.crates-io]
 jq-src = { git = "https://github.com/SOF3/jq-src", rev = "refs/tags/jq-1.7.1" }
 
+[dev-dependencies]
+criterion = "0.5.1"
+iai = "0.1.1"
+
 [[bench]]
-name = "main"
+name = "criterion"
 harness = false
+
+[[bench]]
+name = "iai"
+harness = false
+
+
+[features]
+iai-bench = []

benches/criterion.rs

@@ -0,0 +1,22 @@
+use std::fmt;
+
+use aoc2024::*;
+use criterion::*;
+
+fn bench(criterion_manager: &mut Criterion) {
+    fn run(criterion_manager: &mut Criterion) -> anyhow::Result<()> {
+        aoc2024::bench!(criterion_manager);
+        Ok(())
+    }
+
+    run(criterion_manager).unwrap();
+}
+
+fn call_benched<In: Parse, Out: fmt::Display>(b: &mut Bencher, day: u32, f: impl FnMut(In) -> Out) {
+    let input = load_input(Mode::Private, day).unwrap();
+    let parsed: In = Parse::parse(&input);
+    b.iter_batched(|| parsed.clone(), f, BatchSize::LargeInput);
+}
+
+criterion_group!(benches, bench);
+criterion_main!(benches);

benches/iai.rs

@@ -0,0 +1,5 @@
+#[cfg(feature = "iai-bench")]
+aoc2024::iai!();
+
+#[cfg(not(feature = "iai-bench"))]
+fn main() {}

macros/Cargo.toml

@@ -0,0 +1,13 @@
+[package]
+name = "macros"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+proc-macro = true
+
+[dependencies]
+itertools = "0.13.0"
+proc-macro2 = "1.0.92"
+quote = "1.0.37"
+syn = "2.0.90"