fmapSvL.hs

{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE ScopedTypeVariables #-}

{-# OPTIONS_GHC -O2 -fforce-recomp #-}

module Main (main) where

import Criterion.Main
import Control.Monad
import Data.Int
import Data.Vector (Vector)
import qualified Data.Vector as Vector
import Control.DeepSeq ( ($!!) )

main :: IO ()
main = do
  bs :: Vector Int <- pure $!! Vector.fromList [1..5000]
  defaultMain
    [ bgroup "fmap bench"
      [ bench "FMAP (LAZY)"   $ nf (fmap (+1)) bs
      , bench "FMAP (STRICT)" $ nf (fmap' (+1)) bs
      ]
    ]

fmap' :: Monad m => (a -> b) -> m a -> m b
fmap' f m = m >>= \ !x -> pure $! f x
{-# INLINE fmap' #-}