Skip to content

Commit 1020b0b

Browse files
authored
Merge pull request #2 from alphal00p/float_refactor
Float refactor
2 parents 9d25e9e + 9a3a0f7 commit 1020b0b

13 files changed

+872
-638
lines changed

Cargo.toml

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,9 +23,9 @@ lto = "fat"
2323
smallvec = "1.13.2"
2424
ahash = "0.8.11"
2525
num = { version = "0.4.1", features = ["serde"] }
26-
f128 = { git = "https://github.com/benruijl/f128" }
2726
statrs = "0.16.0"
2827
itertools = "0.12.1"
2928
rand = "0.8.5"
3029
criterion = "0.5.1"
3130
serde = { version = "1.0", features = ["derive"] }
31+
ref-ops = "0.2.5"

benches/gamma_bench.rs

+3-3
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
use criterion::{criterion_group, criterion_main, Criterion};
2-
use momtrop::gamma::inverse_gamma_lr;
2+
use momtrop::gamma::inverse_gamma_lr_impl;
33

44
fn criterion_benchmark(c: &mut Criterion) {
55
let mut group = c.benchmark_group("gamma sampling benchmarks");
66

77
for omega in [0.5, 1.0, 2.0, 10.0] {
88
for p in [0.1, 0.3, 0.5, 0.7, 0.9] {
9-
group.bench_function(&format!("omega = {omega}, p = {p}"), |b| {
10-
b.iter(|| inverse_gamma_lr(omega, p, 50, 5.0))
9+
group.bench_function(format!("omega = {omega}, p = {p}"), |b| {
10+
b.iter(|| inverse_gamma_lr_impl(omega, p, 50, 5.0))
1111
});
1212
}
1313
}

benches/graph_bench.rs

+6-2
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,15 @@ fn criterion_benchmark(c: &mut Criterion) {
3434
let logger = momtrop::log::DummyLogger {};
3535

3636
let loop_signature = vec![vec![1]; 3];
37-
let sampler = graph.build_sampler(loop_signature, 3).unwrap();
37+
let sampler = graph.build_sampler(loop_signature).unwrap();
3838
let mut rng = rand::rngs::StdRng::seed_from_u64(69);
3939
let p1 = Vector::from_array([3.0, 4.0, 5.0]);
4040
let p2 = Vector::from_array([6.0, 7.0, 8.0]);
41-
let edge_data = vec![(None, Vector::new()), (None, p1), (None, (&p1 + &p2))];
41+
let edge_data = vec![
42+
(None, Vector::new_from_num(&0.0)),
43+
(None, p1),
44+
(None, (&p1 + &p2)),
45+
];
4246

4347
let x_space_point = vec![rng.r#gen(); sampler.get_dimension()];
4448

0 commit comments

Comments
 (0)