Split the Parser module into "main" and "benchmarks"

Author: harendra-kumar

benchmark/Streamly/Benchmark/Data/Parser.hs

@@ -16,9 +16,10 @@
 {-# LANGUAGE ScopedTypeVariables #-}
 {-# OPTIONS_GHC -Wno-orphans #-}
 
-module Main
+module Streamly.Benchmark.Data.Parser
   (
-    main
+    benchmarks
+  , sourceUnfoldrM
   ) where
 
 import Control.Applicative ((<|>))
@@ -703,9 +704,6 @@ concatSequence =
 -- Benchmarks
 -------------------------------------------------------------------------------
 
-moduleName :: String
-moduleName = "Data.Parser"
-
 instance NFData ParseError where
     {-# INLINE rnf #-}
     rnf (ParseError x) = rnf x
@@ -823,46 +821,3 @@ benchmarks value env arrays =
     {-# NOINLINE parseManyGroupsRollingEitherAlt1 #-}
     parseManyGroupsRollingEitherAlt1 =
         parseManyGroupsRollingEitherAlt (>) value
-
--------------------------------------------------------------------------------
--- Driver
--------------------------------------------------------------------------------
-
-main :: IO ()
-main = do
-#ifndef FUSION_CHECK
-    env <- mkHandleBenchEnv
-    runWithCLIOptsEnv defaultStreamSize alloc (allBenchmarks env)
-
-    where
-
-    alloc value = Stream.fold Fold.toList $ Array.chunksOf 100 $ sourceUnfoldrM value 0
-
-    allBenchmarks env arrays value =
-        let allBenches = benchmarks value env arrays
-            get x = map snd $ filter ((==) x . fst) allBenches
-            o_1_space = get SpaceO_1
-            o_n_heap = get HeapO_n
-            o_n_space = get SpaceO_n
-        in
-        [ bgroup (o_1_space_prefix moduleName) o_1_space
-        , bgroup (o_n_heap_prefix moduleName) o_n_heap
-        , bgroup (o_n_space_prefix moduleName) o_n_space
-        ]
-#else
-    -- Enable FUSION_CHECK macro at the beginning of the file
-    -- Enable one benchmark below, and run the benchmark
-    -- Check the .dump-simpl output
-    let value = 100000
-    -- let input = sourceUnfoldrM value 1
-    -- manyTill value input
-    -- deintercalate value input
-    -- deintercalate1 value input
-    -- deintercalateAll value input
-    -- sepByWords input
-    -- sepByAllWords input
-    -- sepBy1 input
-    -- sepByWords1 input
-    takeFramedByEsc_ value (sourceEscapedFrames value 1)
-    return ()
-#endif

benchmark/Streamly/Benchmark/Data/ParserMain.hs

@@ -0,0 +1,79 @@
+#undef FUSION_CHECK
+#ifdef FUSION_CHECK
+{-# OPTIONS_GHC -ddump-simpl -ddump-to-file -dsuppress-all #-}
+#endif
+
+-- |
+-- Module      : Streamly.Benchmark.Data.Parser
+-- Copyright   : (c) 2020 Composewell Technologies
+--
+-- License     : BSD-3-Clause
+-- Maintainer  : [email protected]
+
+{-# LANGUAGE CPP #-}
+{-# LANGUAGE FlexibleContexts #-}
+{-# LANGUAGE ScopedTypeVariables #-}
+{-# OPTIONS_GHC -Wno-orphans #-}
+
+module Main
+  (
+    main
+  ) where
+
+import Prelude hiding
+    (any, all, take, sequence, sequence_, sequenceA, takeWhile, dropWhile, span)
+
+import qualified Streamly.Internal.Data.Array as Array
+import qualified Streamly.Internal.Data.Fold as Fold
+import qualified Streamly.Data.Stream as Stream
+
+import Test.Tasty.Bench hiding (env)
+import Streamly.Benchmark.Common
+import Streamly.Benchmark.Common.Handle
+import Streamly.Benchmark.Data.Parser
+
+moduleName :: String
+moduleName = "Data.Parser"
+
+-------------------------------------------------------------------------------
+-- Driver
+-------------------------------------------------------------------------------
+
+main :: IO ()
+main = do
+#ifndef FUSION_CHECK
+    env <- mkHandleBenchEnv
+    runWithCLIOptsEnv defaultStreamSize alloc (allBenchmarks env)
+
+    where
+
+    alloc value = Stream.fold Fold.toList $ Array.chunksOf 100 $ sourceUnfoldrM value 0
+
+    allBenchmarks env arrays value =
+        let allBenches = benchmarks value env arrays
+            get x = map snd $ filter ((==) x . fst) allBenches
+            o_1_space = get SpaceO_1
+            o_n_heap = get HeapO_n
+            o_n_space = get SpaceO_n
+        in
+        [ bgroup (o_1_space_prefix moduleName) o_1_space
+        , bgroup (o_n_heap_prefix moduleName) o_n_heap
+        , bgroup (o_n_space_prefix moduleName) o_n_space
+        ]
+#else
+    -- Enable FUSION_CHECK macro at the beginning of the file
+    -- Enable one benchmark below, and run the benchmark
+    -- Check the .dump-simpl output
+    let value = 100000
+    -- let input = sourceUnfoldrM value 1
+    -- manyTill value input
+    -- deintercalate value input
+    -- deintercalate1 value input
+    -- deintercalateAll value input
+    -- sepByWords input
+    -- sepByAllWords input
+    -- sepBy1 input
+    -- sepByWords1 input
+    takeFramedByEsc_ value (sourceEscapedFrames value 1)
+    return ()
+#endif

benchmark/streamly-benchmarks.cabal

@@ -329,8 +329,10 @@ benchmark Data.MutArray
 benchmark Data.Parser
   import: bench-options
   type: exitcode-stdio-1.0
-  hs-source-dirs: Streamly/Benchmark/Data
-  main-is: Parser.hs
+  hs-source-dirs: .
+  main-is: Streamly/Benchmark/Data/ParserMain.hs
+  other-modules:
+    Streamly.Benchmark.Data.Parser
   if impl(ghcjs)
     buildable: False
   else

hie.yaml

@@ -32,7 +32,7 @@ cradle:
               component: "bench:Data.Fold.Window"
             - path: "./benchmark/Streamly/Benchmark/Data/MutArray.hs"
               component: "bench:Data.MutArray"
-            - path: "./benchmark/Streamly/Benchmark/Data/Parser.hs"
+            - path: "./benchmark/Streamly/Benchmark/Data/"
               component: "bench:Data.Parser"
             - path: "./benchmark/Streamly/Benchmark/Data/ParserK.hs"
               component: "bench:Data.ParserK"