Make ShR a newtype over ShX

1 files changed, 2 insertions, 2 deletions

diff --git a/bench/Main.hs b/bench/Main.hs
index ce3a9df..185bef0 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)