Make sure GHC.List.last is memory-efficient
authorJoachim Breitner <mail@joachim-breitner.de>
Thu, 16 Apr 2015 12:49:23 +0000 (14:49 +0200)
committerJoachim Breitner <mail@joachim-breitner.de>
Fri, 24 Apr 2015 09:14:26 +0000 (11:14 +0200)
by eta-expanding its definition so that GHC optmizes the foldl here.
Also make sure that other uses of last go via foldl as well, to allow
list fusion (tested in T9339). Fixes #10260.

libraries/base/GHC/List.hs

index a712f9e..fcc89d3 100644 (file)
@@ -84,8 +84,15 @@ last [x]                =  x
 last (_:xs)             =  last xs
 last []                 =  errorEmptyList "last"
 #else
--- use foldl to allow fusion
-last = foldl (\_ x -> x) (errorEmptyList "last")
+-- Use foldl to make last a good consumer.
+-- This will compile to good code for the actual GHC.List.last.
+-- (At least as long it is eta-expaned, otherwise it does not, #10260.)
+last xs = foldl (\_ x -> x) lastError xs
+{-# INLINE last #-}
+-- The inline pragma is required to make GHC remember the implementation via
+-- foldl.
+lastError :: a
+lastError = errorEmptyList "last"
 #endif
 
 -- | Return all the elements of a list except the last one.