fix benchmarks being broken for every target except wam32

diff --git a/benches/run_iai.rs b/benches/run_iai.rs
index b852542e97cbede943753e22aabfcf15cd02d5e1..2ce4731f65e5a4a288de3bb7b9e28e9e1cf11edf 100644 (file)
--- a/benches/run_iai.rs
+++ b/benches/run_iai.rs
@@ -1,27 +1,29 @@
-#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
-use iai_callgrind::{library_benchmark, library_benchmark_group, main};
-#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
-use scryer_prolog::machine::parsed_results::QueryResolution;
-
 #[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
 mod setup;
 
 #[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
-#[library_benchmark]
-#[bench::count_edges(setup::prolog_benches()["count_edges"].setup())]
-#[bench::numlist(setup::prolog_benches()["numlist"].setup())]
-#[bench::csv_codename(setup::prolog_benches()["csv_codename"].setup())]
-fn bench(mut run: impl FnMut() -> QueryResolution) -> QueryResolution {
-    run()
+mod iai {
+    use iai_callgrind::{library_benchmark, library_benchmark_group, main};
+
+    use scryer_prolog::machine::parsed_results::QueryResolution;
+
+    use super::setup;
+
+    #[library_benchmark]
+    #[bench::count_edges(setup::prolog_benches()["count_edges"].setup())]
+    #[bench::numlist(setup::prolog_benches()["numlist"].setup())]
+    #[bench::csv_codename(setup::prolog_benches()["csv_codename"].setup())]
+    fn bench(mut run: impl FnMut() -> QueryResolution) -> QueryResolution {
+        run()
+    }
+
+    library_benchmark_group!(
+        name = benches;
+        benchmarks = bench
+    );
+
+    main!(library_benchmark_groups = benches);
 }
 
 #[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
-library_benchmark_group!(
-    name = benches;
-    benchmarks = bench
-);
-#[cfg(not(all(target_arch = "wasm32", target_os = "unknown")))]
-main!(library_benchmark_groups = benches);
-
-#[cfg(all(target_arch = "wasm32", target_os = "unknown"))]
 fn main() {}