aboutsummaryrefslogtreecommitdiff
path: root/rand/benches/distributions/gamma.rs
diff options
context:
space:
mode:
Diffstat (limited to 'rand/benches/distributions/gamma.rs')
-rw-r--r--rand/benches/distributions/gamma.rs31
1 files changed, 31 insertions, 0 deletions
diff --git a/rand/benches/distributions/gamma.rs b/rand/benches/distributions/gamma.rs
new file mode 100644
index 0000000..bf3fd36
--- /dev/null
+++ b/rand/benches/distributions/gamma.rs
@@ -0,0 +1,31 @@
+use std::mem::size_of;
+use test::Bencher;
+use rand;
+use rand::distributions::IndependentSample;
+use rand::distributions::gamma::Gamma;
+
+#[bench]
+fn bench_gamma_large_shape(b: &mut Bencher) {
+ let gamma = Gamma::new(10., 1.0);
+ let mut rng = rand::weak_rng();
+
+ b.iter(|| {
+ for _ in 0..::RAND_BENCH_N {
+ gamma.ind_sample(&mut rng);
+ }
+ });
+ b.bytes = size_of::<f64>() as u64 * ::RAND_BENCH_N;
+}
+
+#[bench]
+fn bench_gamma_small_shape(b: &mut Bencher) {
+ let gamma = Gamma::new(0.1, 1.0);
+ let mut rng = rand::weak_rng();
+
+ b.iter(|| {
+ for _ in 0..::RAND_BENCH_N {
+ gamma.ind_sample(&mut rng);
+ }
+ });
+ b.bytes = size_of::<f64>() as u64 * ::RAND_BENCH_N;
+}