diff options
Diffstat (limited to 'bench')
| -rw-r--r-- | bench/Main.hs | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/bench/Main.hs b/bench/Main.hs index 7f4b4e3..45c1129 100644 --- a/bench/Main.hs +++ b/bench/Main.hs @@ -35,7 +35,7 @@ main = do let enable = False bracket (Arith.statisticsEnable enable) (\() -> do Arith.statisticsEnable False - when enable $ Arith.statisticsPrintAll) + when enable Arith.statisticsPrintAll) (\() -> main_tests) main_tests :: IO () @@ -97,17 +97,17 @@ main_tests = defaultMain ,bgroup "orthotope" [bench "normalize [1e6]" $ let n = 1_000_000 - in nf (\a -> RS.normalize a) + in nf RS.normalize (RS.rev [0] (RS.iota @Double n)) ,bench "normalize noop [1e6]" $ let n = 1_000_000 - in nf (\a -> RS.normalize a) + in nf RS.normalize (RS.rev [0] (RS.rev [0] (RS.iota @Double n))) ] ,bgroupIf enableMisc "misc" [let n = 1000 k = 1000 - in bgroup ("fusion [" ++ show k ++ "]*" ++ show n) $ + in bgroup ("fusion [" ++ show k ++ "]*" ++ show n) [bench "sum (concat)" $ nf (\as -> VS.sum (VS.concat as)) (replicate n (VS.enumFromTo (1::Int) k)) @@ -120,14 +120,14 @@ main_tests = defaultMain [bench ("LA.vjoin [500]*1e" ++ show ni) $ let n = 10 ^ ni k = 500 - in nf (\as -> LA.vjoin as) + in nf LA.vjoin (replicate n (VS.enumFromTo (1::Int) k)) | ni <- [1::Int ..5]] ,bgroup "vectorStorable" [bench ("VS.concat [500]*1e" ++ show ni) $ let n = 10 ^ ni k = 500 - in nf (\as -> VS.concat as) + in nf VS.concat (replicate n (VS.enumFromTo (1::Int) k)) | ni <- [1::Int ..5]] ] @@ -136,14 +136,14 @@ main_tests = defaultMain [bench ("LA.vjoin [1e" ++ show ki ++ "]*500") $ let n = 500 k = 10 ^ ki - in nf (\as -> LA.vjoin as) + in nf LA.vjoin (replicate n (VS.enumFromTo (1::Int) k)) | ki <- [1::Int ..5]] ,bgroup "vectorStorable" [bench ("VS.concat [1e" ++ show ki ++ "]*500") $ let n = 500 k = 10 ^ ki - in nf (\as -> VS.concat as) + in nf VS.concat (replicate n (VS.enumFromTo (1::Int) k)) | ki <- [1::Int ..5]] ] @@ -230,7 +230,7 @@ tests_compare = nf (\a -> LA.sumElements (sin a)) (LA.linspace @Double n (0.0, fromIntegral (n - 1))) ,bench "sum Double [1e6]" $ - nf (\a -> LA.sumElements a) + nf LA.sumElements (LA.linspace @Double n (0.0, fromIntegral (n - 1))) ,bench "dotprod Float [1e6]" $ nf (\(a, b) -> a LA.<.> b) |