diff options
| author | Mikolaj Konarski <mikolaj.konarski@gmail.com> | 2025-04-25 23:45:48 +0200 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@gmail.com> | 2025-04-25 23:45:48 +0200 |
| commit | 4087d405b51cf32363cb7507df6ffe1a170c0f7f (patch) | |
| tree | fa9ec2db7f8d02c1e5cb4e10978e1871632e0441 /src/Data/Array/Nested/Internal/Shaped.hs | |
| parent | 0121c2480b2ea8d34c9d293941283e6c8e2e09dc (diff) | |
Add mstrideTree and StrideTree
Diffstat (limited to 'src/Data/Array/Nested/Internal/Shaped.hs')
| -rw-r--r-- | src/Data/Array/Nested/Internal/Shaped.hs | 2 |
1 files changed, 2 insertions, 0 deletions
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 |