From f2cec69969a68e8feed3dceacef5186b1debdda5 Mon Sep 17 00:00:00 2001 From: Mikolaj Konarski Date: Tue, 16 Dec 2025 09:51:51 +0100 Subject: Make ShR a newtype over ShX --- bench/Main.hs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'bench/Main.hs') 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) -- cgit v1.2.3-70-g09d2