composewell/streamly
1
1
Fork 0
mirror of https://github.com/composewell/streamly.git synced 2024-09-17 14:37:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add a benchmark for foldBreak in Fold module

Browse Source
This commit is contained in:
Harendra Kumar 2022-05-17 22:38:05 +05:30
parent 107bebd771
commit 61b907cf8f
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
benchmark/Streamly/Benchmark/Data/Fold.hs
View File

@ -12,6 +12,7 @@
module Main (main) where
import Control.DeepSeq (NFData(..))
import Control.Monad (when)
import Control.Monad.IO.Class (MonadIO(..))
import Data.Functor.Identity (Identity)
import Data.Map.Strict (Map)
@ -140,6 +141,12 @@ shortest = IP.fold (FL.shortest FL.sum FL.length)
longest :: Monad m => SerialT m Int -> m (Either Int Int)
longest = IP.fold (FL.longest FL.sum FL.length)
{-# INLINE foldBreak #-}
foldBreak :: Monad m => SerialT m Int -> m ()
foldBreak s = do
(r, s1) <- IP.foldBreak (FL.take 1 FL.length) s
when (r /= 0) $ foldBreak s1
-------------------------------------------------------------------------------
-- Distributing by parallel application
-------------------------------------------------------------------------------
@ -388,6 +395,7 @@ o_1_space_serial_composition value =
[ benchIOSink value "filter even" $ filter value
, benchIOSink value "foldFilter even" $ foldFilter value
, benchIOSink value "foldFilter even, odd" $ foldFilter2 value
, benchIOSink value "foldBreak (recursive)" foldBreak
, benchIOSink value "serialWith (all, any)" $ splitAllAny value
, benchIOSink value "serial_ (all, any)" $ serial_ value
, benchIOSink value "tee (all, any)" $ teeAllAny value