From 9560d0f26420409afd2230fb7e5e111eafcced06 Mon Sep 17 00:00:00 2001
From: Mikolaj Konarski <mikolaj.konarski@funktory.com>
Date: Mon, 1 Dec 2025 01:44:05 +0100
Subject: Expose the unfolding of the indexing operations

---
 src/Data/Array/Nested/Shaped.hs | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/Data/Array/Nested/Shaped.hs')

diff --git a/src/Data/Array/Nested/Shaped.hs b/src/Data/Array/Nested/Shaped.hs
index 8957549..82dfc91 100644
--- a/src/Data/Array/Nested/Shaped.hs
+++ b/src/Data/Array/Nested/Shaped.hs
@@ -52,6 +52,7 @@ srank = shsRank . sshape
 ssize :: Elt a => Shaped sh a -> Int
 ssize = shsSize . sshape
 
+{-# INLINEABLE sindex #-}
 sindex :: Elt a => Shaped sh a -> IIxS sh -> a
 sindex (Shaped arr) idx = mindex arr (ixxFromIxS idx)
 
@@ -59,6 +60,7 @@ shsTakeIx :: Proxy sh' -> ShS (sh ++ sh') -> IIxS sh -> ShS sh
 shsTakeIx _ _ ZIS = ZSS
 shsTakeIx p sh (_ :.$ idx) = case sh of n :$$ sh' -> n :$$ shsTakeIx p sh' idx
 
+{-# INLINEABLE sindexPartial #-}
 sindexPartial :: forall sh1 sh2 a. Elt a => Shaped (sh1 ++ sh2) a -> IIxS sh1 -> Shaped sh2 a
 sindexPartial sarr@(Shaped arr) idx =
   Shaped (mindexPartial @a @(MapJust sh1) @(MapJust sh2)
-- 
cgit v1.2.3-70-g09d2