composewell/streamly
1
1
Fork 0
mirror of https://github.com/composewell/streamly.git synced 2024-08-16 23:10:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Copy only the required slice while pinning/unpinning the array

Browse Source
This commit is contained in:
Adithya Kumar 2023-08-21 12:53:46 +05:30
parent aba1a5fe2c
commit fb51bf6b7d
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
core/src/Streamly/Internal/Data/MutArray/Type.hs
View File

@ -362,17 +362,19 @@ data MutArray a =
-- overhead is a copy if the input array is unpinned.
{-# INLINE pin #-}
pin :: MutArray a -> IO (MutArray a)
pin arr@MutArray{..} = do
contents <- Unboxed.pin arrContents
return $ arr {arrContents = contents}
pin arr@MutArray{..} =
if Unboxed.isPinned arrContents
then pure arr
else pinnedClone arr
-- | Copy the 'MutArray' to unpinned memory if pinned, else do nothing. The
-- overhead is a copy if the input array is pinned.
{-# INLINE unpin #-}
unpin :: MutArray a -> IO (MutArray a)
unpin arr@MutArray{..} = do
contents <- Unboxed.unpin arrContents
return $ arr {arrContents = contents}
unpin arr@MutArray{..} =
if Unboxed.isPinned arrContents
then clone arr
else pure arr
{-# INLINE isPinned #-}
isPinned :: MutArray a -> Bool