initial commit

Author: Tomasz Tylec

README.md

@@ -0,0 +1,47 @@
+module Main where
+
+import Criterion.Main
+
+import Control.Monad
+import Data.IORef
+import Data.Random
+import Data.Random.Distribution.Uniform
+import Data.Vector (Vector, fromList)
+import System.Random.MWC (GenIO, uniformVector, createSystemRandom, create)
+import qualified System.Random.MWC as MWC
+import Statistics.Sample (mean)
+
+import Data.Random.Source.PureMT (newPureMT)
+import Data.Random.Source.DevRandom
+
+
+
+k = 10000
+
+randomListFu :: RVar (Vector Double)
+randomListFu = fromList <$> replicateM k stdUniform
+
+randomListMWC :: GenIO -> IO (Vector Double)
+randomListMWC gen = fromList <$> replicateM k (MWC.uniform gen)
+
+main :: IO ()
+main = do
+  mwc <- createSystemRandom
+  puremt <- newPureMT >>= newIORef
+
+  defaultMain
+    [ bgroup "MWC vs randomu-fu abstraction"
+      [ bench "random-fu"
+        $ nfIO (mean <$> sampleFrom mwc randomListFu)
+      , bench "mw'"
+        $ nfIO (mean <$> randomListMWC mwc)
+      ]
+    , bgroup "Different sources"
+      [ bench "pureMT"
+        $ nfIO (mean <$> sampleFrom puremt randomListFu)
+      , bench "std"
+        $ nfIO (mean <$> sample randomListFu)
+      , bench "urandom"
+        $ nfIO (mean <$> sampleFrom DevURandom randomListFu)
+      ]
+    ]

examples/benchmark.hs

@@ -0,0 +1,34 @@
+module Main where
+
+import System.Environment
+import Control.Monad
+import Data.Random
+
+inCircle :: Double -> Double -> Double
+inCircle x y | x^2 + y^2 <= 1 = 1
+             | otherwise = 0
+
+point = inCircle <$> stdUniform <*> stdUniform
+
+areaMC :: [Bool] -> Double
+areaMC ps = let l = fromIntegral . length
+            in l (filter id ps) / l ps
+
+areaMC' :: [Double] -> Double
+areaMC' ps = sum ps
+
+-- piMC :: Int -> IO Double
+-- piMC n = (*4) . areaMC' <$> sample (replicateM n point)
+
+piMC :: Int -> IO Double
+piMC n = (*4) . areaMC' <$> sample (replicateM n point)
+
+piMC' :: Int -> RVar Double
+piMC' n = (*4) . areaMC' <$> replicateM n point
+
+main = do
+  (ns:_) <- getArgs
+  let n = read ns :: Int
+  p <- piMC n
+  -- p <- sample $ piMC' n
+  print p

examples/memory.hs

@@ -0,0 +1,39 @@
+{pkgs ? import <nixpkgs> { }, ghc ? pkgs.haskell.compiler.ghc822}:
+
+with pkgs;
+
+let
+  # Uncomment the line below to build HaskellR against a version of R with
+  # the --enable-strict-barrier configure flag enabled for better memory
+  # diagnostics.
+
+  # R = pkgs.R.override { enableStrictBarrier = true; };
+
+  # XXX Workaround https://ghc.haskell.org/trac/ghc/ticket/11042.
+  libHack = if stdenv.isDarwin then {
+      DYLD_LIBRARY_PATH = ["${R}/lib/R/lib"];
+    } else {
+      LD_LIBRARY_PATH = ["${R}/lib/R"];
+    };
+in
+
+haskell.lib.buildStackProject ({
+  name = "HaskellR";
+  inherit ghc;
+  src = pkgs.fetchFromGitHub {
+    owner  = "tweag";
+    repo   = "HaskellR";
+    rev    = "59e8540f6fd6c84acba1655d758bab637b675219";
+    sha256 = "0v9yxj4dsnl3c7qyjiwsj367z0ksy936fks9db5z3gcb5339fwrm";
+  };
+  buildInputs =
+    [ python36Packages.ipython
+      python36Packages.jupyter_client
+      python36Packages.notebook
+      R
+      zeromq
+      zlib
+    ];
+  LANG = "en_US.UTF-8";
+  LD_LIBRARY_PATH = ["${R}/lib/R/"];
+} // libHack)

haskellr.nix

@@ -0,0 +1 @@
+import (fetchTarball "https://github.com/nixos/nixpkgs/archive/1354099daf98b7a1f79e6c41ce6bfda5c40177ae.tar.gz")

nixpkgs.nix

@@ -0,0 +1,35 @@
+name:                ft14-random
+version:             0.1.0.0
+license:             AllRightsReserved
+author:              "Tomasz Tylec"
+maintainer:          "[email protected]"
+copyright:           "2018 Tomasz Tylec"
+
+extra-source-files:
+- README.md
+
+description:         Please see the README
+
+dependencies:
+- base >= 4.7 && < 5
+- mtl
+- containers
+- extra
+- random-fu
+- random-source
+- vector
+- criterion
+- mwc-random
+- statistics
+- ihaskell
+- ihaskell-inline-r
+
+
+executables:
+  benchmark:
+    main: examples/benchmark.hs
+    ghc-options: -O1
+
+  memory:
+    main: examples/memory.hs
+    ghc-options: -O1