diff options
Diffstat (limited to 'bench')
| -rw-r--r-- | bench/Main.hs | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/bench/Main.hs b/bench/Main.hs index 17a93e0..b604eb9 100644 --- a/bench/Main.hs +++ b/bench/Main.hs @@ -15,11 +15,11 @@ import Numeric.LinearAlgebra qualified as LA import Test.Tasty.Bench import Text.Show (showListWith) -import Data.Array.Mixed.XArray (XArray(..)) import Data.Array.Nested -import Data.Array.Nested.Internal.Mixed (mliftPrim, mliftPrim2, Mixed(M_Primitive), toPrimitive) -import Data.Array.Nested.Internal.Ranked (liftRanked1, liftRanked2) -import qualified Data.Array.Strided.Arith.Internal as Arith +import Data.Array.Nested.Mixed (Mixed(M_Primitive), mliftPrim, mliftPrim2, toPrimitive) +import Data.Array.Nested.Ranked (liftRanked1, liftRanked2) +import Data.Array.Strided.Arith.Internal qualified as Arith +import Data.Array.XArray (XArray(..)) enableMisc :: Bool @@ -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 () @@ -51,7 +51,7 @@ main_tests = defaultMain " str " ++ showSh (stridesOf inp1) ++ " " ++ showSh (stridesOf inp2)) $ nf (\(a,b) -> rsumAllPrim (rdot1Inner a b)) (inp1, inp2) - iota n = riota @Double n + iota = riota @Double in [dotprodBench "dot 1D" (iota 10_000_000 @@ -107,7 +107,7 @@ main_tests = defaultMain ,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)) |