Commits

Aleksey Khudyakov committed e510343

Eta-reduce/expand definitions to force inlining

Comments (0)

Files changed (1)

System/Random/MWC/Monad.hs

   {-# INLINE fmap #-}
 
 instance PrimMonad m => Monad (Rand m) where
-  return x         = Rand (\_ -> return x)
+  return           = Rand . const . return
   (Rand rnd) >>= f = Rand $ \g -> (\x -> runRand (f x) g) =<< rnd g
   {-# INLINE return #-}
   {-# INLINE (>>=)  #-}
 
 -- | Uniformly distributed values
 uniform :: (PrimMonad m, Variate a) => Rand m a
-uniform = Rand MWC.uniform
+uniform = Rand $ \g -> MWC.uniform g
 {-# INLINE uniform #-}
 
 -- | Uniformly distributed values in range
 uniformR :: (PrimMonad m, Variate a) => (a,a) -> Rand m a
-uniformR rng = Rand (MWC.uniformR rng)
+uniformR rng = Rand $ \g -> MWC.uniformR rng g
 {-# INLINE uniformR #-}