Strict accumulators

idontgetoutmuch · Apr 7, 2020 · 6f1abb9 · 6f1abb9
1 parent ef71573
commit 6f1abb9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/System/Random.hs b/System/Random.hs
@@ -1,3 +1,4 @@
+{-# LANGUAGE BangPatterns #-}
 {-# LANGUAGE CPP #-}
 {-# LANGUAGE DefaultSignatures #-}
 {-# LANGUAGE FlexibleContexts #-}
@@ -1124,7 +1125,7 @@ boundedExclusiveIntegerM s gen = go
 integerWordSize :: Integer -> Int
 integerWordSize = go 0
   where
-    go acc i
+    go !acc i
       | i == 0 = acc
       | otherwise = go (acc + 1) (i `shiftR` WORD_SIZE_IN_BITS)
 {-# INLINE integerWordSize #-}
@@ -1134,7 +1135,7 @@ integerWordSize = go 0
 uniformIntegerWords :: (MonadRandom g m) => Int -> g -> m Integer
 uniformIntegerWords n gen = go 0 n
   where
-    go acc i
+    go !acc i
       | i == 0 = return acc
       | otherwise = do
         (w :: Word) <- uniform gen