diff options
Diffstat (limited to 'bench')
-rw-r--r-- | bench/Main.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/bench/Main.hs b/bench/Main.hs index ef03b1a..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.XArray (XArray(..)) import Data.Array.Nested 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 @@ -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 |