Allow primMapListBounded to inline with one arg

haskell · Feb 15, 2024 · cd02c61 · cd02c61
1 parent 2603009
commit cd02c61
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Data/ByteString/Builder/Prim.hs b/Data/ByteString/Builder/Prim.hs
@@ -579,7 +579,10 @@ primBounded w x =
 -- because it moves several variables out of the inner loop.
 {-# INLINE primMapListBounded #-}
 primMapListBounded :: BoundedPrim a -> [a] -> Builder
-primMapListBounded w xs0 =
+primMapListBounded w = \xs0 ->
+  -- We want this to inline when there is one arg, so that we can
+  -- specialise on the BoundedPrim "w".  So we move the \xs0 after the
+  -- "=" sign so that the INLINE pragma doesn't interfere with this.
     builder $ step xs0
   where
     step xs1 k (BufferRange op0 ope0) =