chore(csprng): cleanup conditional imports

zama-ai · Dec 19, 2024 · ae832c1 · ae832c1
1 parent 8504d79
commit ae832c1
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/tfhe-csprng/src/generators/default.rs b/tfhe-csprng/src/generators/default.rs
@@ -1,19 +1,9 @@
 #[cfg(all(target_arch = "x86_64", not(feature = "software-prng")))]
-use super::AesniRandomGenerator;
+pub type DefaultRandomGenerator = super::AesniRandomGenerator;
 #[cfg(all(target_arch = "aarch64", not(feature = "software-prng")))]
-use super::NeonAesRandomGenerator;
+pub type DefaultRandomGenerator = super::NeonAesRandomGenerator;
 #[cfg(any(
     feature = "software-prng",
     not(any(target_arch = "x86_64", target_arch = "aarch64"))
 ))]
-use super::SoftwareRandomGenerator;
-
-#[cfg(all(target_arch = "x86_64", not(feature = "software-prng")))]
-pub type DefaultRandomGenerator = AesniRandomGenerator;
-#[cfg(all(target_arch = "aarch64", not(feature = "software-prng")))]
-pub type DefaultRandomGenerator = NeonAesRandomGenerator;
-#[cfg(any(
-    feature = "software-prng",
-    not(any(target_arch = "x86_64", target_arch = "aarch64"))
-))]
-pub type DefaultRandomGenerator = SoftwareRandomGenerator;
+pub type DefaultRandomGenerator = super::SoftwareRandomGenerator;