make distribution less biased to 0

ROCm · Oct 17, 2023 · 3a95ba3 · 3a95ba3
1 parent 75ee932
commit 3a95ba3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/random.hpp b/test/random.hpp
@@ -32,8 +32,7 @@ namespace prng {
 template <typename T>
 inline T gen_descreet_uniform_sign(double scale, int32_t range)
 {
-    return static_cast<T>((gen_canonical<int>() ? -scale : scale) *
-                          static_cast<double>(gen_0_to_B(range)));
+    return static_cast<T>(scale * prng::gen_A_to_B(-range + 1, range));
 }
 
 template <typename T>