summaryrefslogtreecommitdiff
path: root/compiler/GHC/Utils
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/GHC/Utils')
-rw-r--r--compiler/GHC/Utils/Misc.hs14
1 files changed, 9 insertions, 5 deletions
diff --git a/compiler/GHC/Utils/Misc.hs b/compiler/GHC/Utils/Misc.hs
index 1d827e7902..50ac5dc9e8 100644
--- a/compiler/GHC/Utils/Misc.hs
+++ b/compiler/GHC/Utils/Misc.hs
@@ -6,6 +6,7 @@
{-# LANGUAGE BangPatterns #-}
{-# LANGUAGE TupleSections #-}
{-# LANGUAGE ScopedTypeVariables #-}
+{-# LANGUAGE MagicHash #-}
{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-}
@@ -774,15 +775,18 @@ dropList _ xs@[] = xs
dropList (_:xs) (_:ys) = dropList xs ys
--- | Given two lists xs=x0..xn and ys=y0..ym, return `splitAt n ys`.
+-- | Given two lists xs and ys, return `splitAt (length xs) ys`.
splitAtList :: [b] -> [a] -> ([a], [a])
-splitAtList xs ys = go 0 xs ys
+splitAtList xs ys = go 0# xs ys
where
-- we are careful to avoid allocating when there are no leftover
-- arguments: in this case we can return "ys" directly (cf #18535)
- go _ _ [] = (ys, []) -- len(ys) <= len(xs)
- go n [] bs = (take n ys, bs) -- = splitAt n ys
- go n (_:as) (_:bs) = go (n+1) as bs
+ --
+ -- We make `xs` strict because in the general case `ys` isn't `[]` so we
+ -- will have to evaluate `xs` anyway.
+ go _ !_ [] = (ys, []) -- length ys <= length xs
+ go n [] bs = (take (I# n) ys, bs) -- = splitAt n ys
+ go n (_:as) (_:bs) = go (n +# 1#) as bs
-- drop from the end of a list
dropTail :: Int -> [a] -> [a]