Skip to content

Commit 740eee0

Browse files
antoinewdgLuthaf
authored andcommitted
Use fixed RNG seeds in benchmarks
1 parent fd0a9ef commit 740eee0

File tree

5 files changed

+15
-9
lines changed

5 files changed

+15
-9
lines changed

benches/argon.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ fn cache_move_particle(bencher: &mut Bencher) {
4242
let mut cache = EnergyCache::new();
4343
cache.init(&system);
4444

45-
let mut rng = rand::weak_rng();
45+
let mut rng = utils::get_rng(654646);
4646

4747
let particle: usize = rng.gen_range(0, system.size());
4848
let mut delta = system[particle].position;

benches/nacl.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ fn cache_move_particle_ewald(bencher: &mut Bencher) {
7878
let mut cache = EnergyCache::new();
7979
cache.init(&system);
8080

81-
let mut rng = rand::weak_rng();
81+
let mut rng = utils::get_rng(41201154);
8282

8383
let particle: usize = rng.gen_range(0, system.size());
8484
let mut delta = system[particle].position;
@@ -96,7 +96,7 @@ fn cache_move_particle_wolf(bencher: &mut Bencher) {
9696
let mut cache = EnergyCache::new();
9797
cache.init(&system);
9898

99-
let mut rng = rand::weak_rng();
99+
let mut rng = utils::get_rng(474114);
100100

101101
let particle: usize = rng.gen_range(0, system.size());
102102
let mut delta = system[particle].position;

benches/propane.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ fn cache_move_particles(bencher: &mut Bencher) {
4242
let mut cache = EnergyCache::new();
4343
cache.init(&system);
4444

45-
let mut rng = rand::weak_rng();
45+
let mut rng = utils::get_rng(84541545);
4646

4747
let molecule = rng.choose(system.molecules()).unwrap();
4848
let mut delta = vec![];
@@ -61,7 +61,7 @@ fn cache_move_all_rigid_molecules(bencher: &mut Bencher) {
6161
let mut cache = EnergyCache::new();
6262
cache.init(&system);
6363

64-
let mut rng = rand::weak_rng();
64+
let mut rng = utils::get_rng(7012121);
6565
for molecule in system.molecules().to_owned() {
6666
let delta = Vector3D::new(rng.gen(), rng.gen(), rng.gen());
6767
for i in molecule {

benches/utils/mod.rs

+6
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@ use lumol::sys::{System, Trajectory};
22
use lumol_input::InteractionsInput;
33
use std::path::Path;
44

5+
use rand::{XorShiftRng, SeedableRng};
6+
57
pub fn get_system(name: &str) -> System {
68
let data = Path::new(file!()).parent().unwrap().join("..").join("data");
79

@@ -15,3 +17,7 @@ pub fn get_system(name: &str) -> System {
1517

1618
system
1719
}
20+
21+
pub fn get_rng(seed: u32) -> XorShiftRng {
22+
XorShiftRng::from_seed([seed, 784, 71255487, 5824])
23+
}

benches/water.rs

+4-4
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ fn cache_move_particles_wolf(bencher: &mut Bencher) {
8989
let mut cache = EnergyCache::new();
9090
cache.init(&system);
9191

92-
let mut rng = rand::weak_rng();
92+
let mut rng = utils::get_rng(454548784);
9393

9494
let molecule = rng.choose(system.molecules()).unwrap();
9595
let mut delta = vec![];
@@ -110,7 +110,7 @@ fn cache_move_particles_ewald(bencher: &mut Bencher) {
110110
let mut cache = EnergyCache::new();
111111
cache.init(&system);
112112

113-
let mut rng = rand::weak_rng();
113+
let mut rng = utils::get_rng(9886565);
114114

115115
let molecule = rng.choose(system.molecules()).unwrap();
116116
let mut delta = vec![];
@@ -131,7 +131,7 @@ fn cache_move_all_rigid_molecules_wolf(bencher: &mut Bencher) {
131131
let mut cache = EnergyCache::new();
132132
cache.init(&system);
133133

134-
let mut rng = rand::weak_rng();
134+
let mut rng = utils::get_rng(3);
135135
for molecule in system.molecules().to_owned() {
136136
let delta = Vector3D::new(rng.gen(), rng.gen(), rng.gen());
137137
for i in molecule {
@@ -151,7 +151,7 @@ fn cache_move_all_rigid_molecules_ewald(bencher: &mut Bencher) {
151151
let mut cache = EnergyCache::new();
152152
cache.init(&system);
153153

154-
let mut rng = rand::weak_rng();
154+
let mut rng = utils::get_rng(2121);
155155
for molecule in system.molecules().to_owned() {
156156
let delta = Vector3D::new(rng.gen(), rng.gen(), rng.gen());
157157
for i in molecule {

0 commit comments

Comments
 (0)