diff options
| author | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2025-12-08 20:09:31 +0100 |
|---|---|---|
| committer | Mikolaj Konarski <mikolaj.konarski@funktory.com> | 2026-01-31 10:52:00 +0100 |
| commit | c20f930f21c60f91d01009fc7e16fa4ccc345828 (patch) | |
| tree | 6856d722eab318af94272f59d62cd3d7b5ff2196 /src/Data/Array/Nested/Shaped/Base.hs | |
| parent | 91cb6b86e46054e75a5b3506aaa2b262a3387c8e (diff) | |
Express mvecsWrite and mvecsWritePartial using the new methods
and change the type of the latter to make it possible.
This slightly improves performance of horde-ad tests,
before horde-ad even starts using the Linear methods, which improves
performance even more.
Diffstat (limited to 'src/Data/Array/Nested/Shaped/Base.hs')
| -rw-r--r-- | src/Data/Array/Nested/Shaped/Base.hs | 20 |
1 files changed, 0 insertions, 20 deletions
diff --git a/src/Data/Array/Nested/Shaped/Base.hs b/src/Data/Array/Nested/Shaped/Base.hs index d262383..e2ec416 100644 --- a/src/Data/Array/Nested/Shaped/Base.hs +++ b/src/Data/Array/Nested/Shaped/Base.hs @@ -142,32 +142,12 @@ instance Elt a => Elt (Shaped sh a) where marrayStrides (M_Shaped arr) = marrayStrides 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 - (coerce @(MixedVecs s sh' (Shaped sh a)) @(MixedVecs s sh' (Mixed (MapJust sh) a)) - vecs) - mvecsWriteLinear :: forall sh' s. Int -> Shaped sh a -> MixedVecs s sh' (Shaped sh a) -> ST s () mvecsWriteLinear idx (Shaped arr) vecs = mvecsWriteLinear idx arr (coerce @(MixedVecs s sh' (Shaped sh a)) @(MixedVecs s sh' (Mixed (MapJust sh) a)) vecs) - mvecsWritePartial - :: forall sh1 sh2 s. - IShX (sh1 ++ sh2) -> IIxX sh1 -> Mixed sh2 (Shaped sh a) - -> MixedVecs s (sh1 ++ sh2) (Shaped sh a) - -> ST s () - mvecsWritePartial sh idx arr vecs = - mvecsWritePartial sh idx - (coerce @(Mixed sh2 (Shaped sh a)) - @(Mixed sh2 (Mixed (MapJust sh) a)) - arr) - (coerce @(MixedVecs s (sh1 ++ sh2) (Shaped sh a)) - @(MixedVecs s (sh1 ++ sh2) (Mixed (MapJust sh) a)) - vecs) - mvecsWritePartialLinear :: forall sh1 sh2 s. Proxy sh1 -> Int -> Mixed sh2 (Shaped sh a) |