composewell/streamly
1
1
Fork 0
mirror of https://github.com/composewell/streamly.git synced 2024-08-16 15:00:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use explicit buckets in classifySessions benchmarks

Browse Source 
Terminate the session fold after 10 elements
This commit is contained in:
Harendra Kumar 2022-03-31 16:05:13 +05:30
parent e0ad726fbd
commit ba05f939b9
2 changed files with 23 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
benchmark/Streamly/Benchmark/Prelude/Serial/NestedFold.hs
View File

@ -10,7 +10,6 @@
{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE TupleSections #-}
module Serial.NestedFold (benchmarks) where
@ -214,35 +213,44 @@ o_n_heap_buffering value =
-------------------------------------------------------------------------------
{-# INLINE classifySessionsOf #-}
classifySessionsOf :: (S.MonadAsync m) => SerialT m Int -> m ()
classifySessionsOf =
classifySessionsOf :: S.MonadAsync m => (Int -> Int) -> SerialT m Int -> m ()
classifySessionsOf getKey =
S.drain
. Internal.classifySessionsOf (const (return False)) 3 FL.drain
. Internal.classifySessionsOf
(const (return False)) 3 (FL.take 10 FL.sum)
. Internal.timestamped
. S.concatMap (\x -> S.map (x,) (S.enumerateFromTo 1 (10 :: Int)))
. S.map (\x -> (getKey x, x))
{-# INLINE classifySessionsOfHash #-}
classifySessionsOfHash :: (S.MonadAsync m) => SerialT m Int -> m ()
classifySessionsOfHash =
classifySessionsOfHash :: S.MonadAsync m =>
(Int -> Int) -> SerialT m Int -> m ()
classifySessionsOfHash getKey =
S.drain
. Internal.classifySessionsByGeneric
(Proxy :: Proxy (HashMap k))
1 False (const (return False)) 3 FL.drain
1 False (const (return False)) 3 (FL.take 10 FL.sum)
. Internal.timestamped
. S.concatMap (\x -> S.map (x,) (S.enumerateFromTo 1 (10 :: Int)))
. S.map (\x -> (getKey x, x))
o_n_space_grouping :: Int -> [Benchmark]
o_n_space_grouping value =
-- Buffering operations using heap proportional to group/window sizes.
[ bgroup "grouping"
-- We use 10 element stream per input, so div by 10 here
[ benchIOSink (value `div` 10) "classifySessionsOf"
classifySessionsOf
, benchIOSink (value `div` 10) "classifySessionsOfHash"
classifySessionsOfHash
[ benchIOSink value "classifySessionsOf (10000 buckets)"
(classifySessionsOf (getKey 10000))
, benchIOSink value "classifySessionsOf (64 buckets)"
(classifySessionsOf (getKey 64))
, benchIOSink value "classifySessionsOfHash (10000 buckets)"
(classifySessionsOfHash (getKey 10000))
, benchIOSink value "classifySessionsOfHash (64 buckets)"
(classifySessionsOfHash (getKey 64))
]
]
where
getKey n = (`mod` n)
-------------------------------------------------------------------------------
-- Mixed Transformation
-------------------------------------------------------------------------------

2
bin/bench-config.sh
View File

@ -68,7 +68,7 @@ bench_rts_options () {
Prelude.Serial/o-1-space/mixed/sum-product-fold) echo -n "-K64M" ;;
# XXX These should be moved to o-n-space?
Prelude.Serial/o-n-heap/grouping/classifySessionsOf*) echo -n "-K1M -M32M" ;;
Prelude.Serial/o-n-heap/grouping/classifySessionsOf*) echo -n "-K4M -M32M" ;;
Prelude.Serial/o-n-heap/Functor/*) echo -n "-K4M -M32M" ;;
Prelude.Serial/o-n-heap/transformer/*) echo -n "-K8M -M64M" ;;