From 4087d405b51cf32363cb7507df6ffe1a170c0f7f Mon Sep 17 00:00:00 2001 From: Mikolaj Konarski Date: Fri, 25 Apr 2025 23:45:48 +0200 Subject: Add mstrideTree and StrideTree --- src/Data/Array/Nested/Internal/Shaped.hs | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/Data/Array/Nested/Internal/Shaped.hs') diff --git a/src/Data/Array/Nested/Internal/Shaped.hs b/src/Data/Array/Nested/Internal/Shaped.hs index d7a8ece..eebf66a 100644 --- a/src/Data/Array/Nested/Internal/Shaped.hs +++ b/src/Data/Array/Nested/Internal/Shaped.hs @@ -142,6 +142,8 @@ instance Elt a => Elt (Shaped sh a) where mshowShapeTree _ (sh, t) = "(" ++ show sh ++ ", " ++ mshowShapeTree (Proxy @a) t ++ ")" + mstrideTree (M_Shaped arr) = mstrideTree arr + mvecsWrite :: forall sh' s. IShX sh' -> IIxX sh' -> Shaped sh a -> MixedVecs s sh' (Shaped sh a) -> ST s () mvecsWrite sh idx (Shaped arr) vecs = mvecsWrite sh idx arr -- cgit v1.2.3-70-g09d2