depot/third_party/tvl/users/grfn/xanthous/bench/Xanthous/RandomBench.hs
Default email a291c8690a Project import generated by Copybara.
GitOrigin-RevId: e6e19f3d81a982a62e1bba08f0b4f7fdc21b4ea0
2022-05-19 16:39:52 +02:00

32 lines
1.1 KiB
Haskell

--------------------------------------------------------------------------------
module Xanthous.RandomBench (benchmark, main) where
--------------------------------------------------------------------------------
import Bench.Prelude
--------------------------------------------------------------------------------
import Control.Parallel.Strategies
import Control.Monad.Random
--------------------------------------------------------------------------------
import Xanthous.Random
--------------------------------------------------------------------------------
main :: IO ()
main = defaultMain [benchmark]
--------------------------------------------------------------------------------
benchmark :: Benchmark
benchmark = bgroup "Random"
[ bgroup "chooseSubset"
[ bench "serially" $
nf (evalRand $ chooseSubset (0.5 :: Double) [1 :: Int ..1000000])
(mkStdGen 1234)
]
, bgroup "choose weightedBy"
[ bench "serially" $
nf (evalRand
. choose
. weightedBy (\n -> product [n, pred n .. 1])
$ [1 :: Int ..1000000])
(mkStdGen 1234)
]
]