diff options
Diffstat (limited to 'bench/Main.hs')
| -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 2058e77..8fe0fdc 100644 --- a/bench/Main.hs +++ b/bench/Main.hs @@ -9,7 +9,6 @@ import Control.Monad (when) import Data.Array.Internal qualified as OI import Data.Array.Internal.RankedG qualified as RG import Data.Array.Internal.RankedS qualified as RS -import Data.Foldable (toList) import Data.Vector.Storable qualified as VS import Numeric.LinearAlgebra qualified as LA import Test.Tasty.Bench @@ -19,6 +18,7 @@ import Data.Array.Nested import Data.Array.Nested.Mixed (Mixed(M_Primitive), mliftPrim, mliftPrim2, toPrimitive) import Data.Array.Nested.Mixed.Shape import Data.Array.Nested.Ranked (liftRanked1, liftRanked2) +import Data.Array.Nested.Ranked.Shape import Data.Array.Strided.Arith.Internal qualified as Arith import Data.Array.XArray (XArray(..)) @@ -40,7 +40,7 @@ main_tests = defaultMain let showSh l = showListWith (\n -> let ln = round (logBase 10 (fromIntegral n :: Double)) :: Int in if n > 1 && n == 10 ^ ln then showString ("1e" ++ show ln) else shows n) l "" - in bench (name ++ " " ++ showSh (toList (rshape inp1)) ++ + in bench (name ++ " " ++ showSh (shrToList (rshape inp1)) ++ " str " ++ showSh (stridesOf inp1) ++ " " ++ showSh (stridesOf inp2)) $ nf (\(a,b) -> rsumAllPrim (rdot1Inner a b)) (inp1, inp2) |