From 2140d9b568ac0c43f89d24eb48b63791222e0c11 Mon Sep 17 00:00:00 2001
From: Louis Fortier-Dubois <louisfd94@gmail.com>
Date: Sun, 21 Apr 2024 09:48:11 -0400
Subject: [PATCH] remove JIT subsequent RNG tests (#1652)

---
 crates/burn-jit/src/tests/bernoulli.rs | 21 ---------------------
 crates/burn-jit/src/tests/normal.rs    | 16 ----------------
 crates/burn-jit/src/tests/uniform.rs   | 14 --------------
 3 files changed, 51 deletions(-)

diff --git a/crates/burn-jit/src/tests/bernoulli.rs b/crates/burn-jit/src/tests/bernoulli.rs
index 4487cb0e04..2c0defd435 100644
--- a/crates/burn-jit/src/tests/bernoulli.rs
+++ b/crates/burn-jit/src/tests/bernoulli.rs
@@ -11,27 +11,6 @@ mod tests {
     use burn_jit::kernel::prng::tests_utils::calculate_bin_stats;
     use burn_tensor::{backend::Backend, Distribution, Shape, Tensor};
 
-    #[test]
-    #[serial]
-    fn subsequent_calls_give_different_tensors() {
-        TestBackend::seed(0);
-        let shape: Shape<2> = [40, 40].into();
-        let device = Default::default();
-
-        let tensor_1 =
-            Tensor::<TestBackend, 2>::random(shape.clone(), Distribution::Bernoulli(0.5), &device);
-        let tensor_2 =
-            Tensor::<TestBackend, 2>::random(shape.clone(), Distribution::Bernoulli(0.5), &device);
-        let mut diff_exists = false;
-        for i in 0..shape.num_elements() {
-            if tensor_1.to_data().value[i] != tensor_2.to_data().value[i] {
-                diff_exists = true;
-                break;
-            }
-        }
-        assert!(diff_exists);
-    }
-
     #[test]
     #[serial]
     fn number_of_1_proportional_to_prob() {
diff --git a/crates/burn-jit/src/tests/normal.rs b/crates/burn-jit/src/tests/normal.rs
index dd31300bcc..ca6bc0b751 100644
--- a/crates/burn-jit/src/tests/normal.rs
+++ b/crates/burn-jit/src/tests/normal.rs
@@ -5,22 +5,6 @@ mod tests {
     use burn_tensor::{backend::Backend, Data, Distribution, Shape, Tensor};
     use serial_test::serial;
 
-    #[test]
-    #[serial]
-    fn subsequent_calls_give_different_tensors() {
-        TestBackend::seed(0);
-        let shape = [4, 5];
-        let device = Default::default();
-
-        let tensor_1 =
-            Tensor::<TestBackend, 2>::random(shape, Distribution::Normal(0., 1.), &device);
-        let tensor_2 =
-            Tensor::<TestBackend, 2>::random(shape, Distribution::Normal(0., 1.), &device);
-        for i in 0..20 {
-            assert!(tensor_1.to_data().value[i] != tensor_2.to_data().value[i]);
-        }
-    }
-
     #[test]
     #[serial]
     fn empirical_mean_close_to_expectation() {
diff --git a/crates/burn-jit/src/tests/uniform.rs b/crates/burn-jit/src/tests/uniform.rs
index 0401a1ed2b..05cae3616a 100644
--- a/crates/burn-jit/src/tests/uniform.rs
+++ b/crates/burn-jit/src/tests/uniform.rs
@@ -8,20 +8,6 @@ mod tests {
     use burn_jit::kernel::prng::tests_utils::calculate_bin_stats;
     use serial_test::serial;
 
-    #[test]
-    #[serial]
-    fn subsequent_calls_give_different_tensors() {
-        TestBackend::seed(0);
-        let shape = [4, 5];
-        let device = Default::default();
-
-        let tensor_1 = Tensor::<TestBackend, 2>::random(shape, Distribution::Default, &device);
-        let tensor_2 = Tensor::<TestBackend, 2>::random(shape, Distribution::Default, &device);
-        for i in 0..20 {
-            assert!(tensor_1.to_data().value[i] != tensor_2.to_data().value[i]);
-        }
-    }
-
     #[test]
     #[serial]
     fn values_all_within_interval_default() {