diff options
Diffstat (limited to 'bench/Main.hs')
| -rw-r--r-- | bench/Main.hs | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/bench/Main.hs b/bench/Main.hs index 7f1cbad..6e83270 100644 --- a/bench/Main.hs +++ b/bench/Main.hs @@ -3,6 +3,7 @@ {-# LANGUAGE TypeApplications #-} module Main where +import Control.Exception (bracket) import Data.Array.RankedS qualified as RS import Data.Vector.Storable qualified as VS import Numeric.LinearAlgebra qualified as LA @@ -11,6 +12,7 @@ import Test.Tasty.Bench import Data.Array.Nested import Data.Array.Nested.Internal.Mixed (mliftPrim, mliftPrim2) import Data.Array.Nested.Internal.Ranked (liftRanked1, liftRanked2) +import qualified Data.Array.Mixed.Internal.Arith as Arith enableMisc :: Bool @@ -22,7 +24,14 @@ bgroupIf False = \name _ -> bgroup name [] main :: IO () -main = defaultMain +main = + bracket (Arith.statisticsEnable False) + (\() -> do Arith.statisticsEnable False + Arith.statisticsPrintAll) + (\() -> main_tests) + +main_tests :: IO () +main_tests = defaultMain [bgroup "Num" [bench "sum(+) Double [1e6]" $ let n = 1_000_000 |