[commit: ghc] wip/T13351: Add foldr fusion rules for short lists (c55ead7)

[git at git.haskell.org]

Repository : ssh://git@git.haskell.org/ghc

On branch  : wip/T13351
Link       : http://ghc.haskell.org/trac/ghc/changeset/c55ead70523d1458e3e5fe1c976b148547dea3bb/ghc

>---------------------------------------------------------------

commit c55ead70523d1458e3e5fe1c976b148547dea3bb
Author: Joachim Breitner <mail at joachim-breitner.de>
Date:   Tue Feb 28 12:20:02 2017 -0800

    Add foldr fusion rules for short lists
    
    and make the comment there more useful.
    
    Differential Revision: https://phabricator.haskell.org/D3246


>---------------------------------------------------------------

c55ead70523d1458e3e5fe1c976b148547dea3bb
 libraries/base/GHC/Base.hs                    | 19 +++++++++++++------
 testsuite/tests/simplCore/should_run/T2110.hs |  1 +
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/libraries/base/GHC/Base.hs b/libraries/base/GHC/Base.hs
index e07c077..b5fa91c 100644
--- a/libraries/base/GHC/Base.hs
+++ b/libraries/base/GHC/Base.hs
@@ -897,16 +897,23 @@ augment g xs = g (:) xs
         -- Only activate this from phase 1, because that's
         -- when we disable the rule that expands (++) into foldr
 
+"foldr/nil"    forall k z.             foldr k z []             = z
+"foldr/single" forall k z x.           foldr k z [x]            = k x z
+"foldr/short2" forall k z x1 x2.       foldr k z [x1,x2]        = k x1 (k x2 z)
+"foldr/short3" forall k z x1 x2 x3.    foldr k z [x1,x2,x3]     = k x1 (k x2 (k x3 z))
+"foldr/short4" forall k z x1 x2 x3 x4. foldr k z [x1,x2,x3,x4]  = k x1 (k x2 (k x3 (k x4 z)))
+
+-- "foldr/cons" forall k z x xs. foldr k z (x:xs) = k x (foldr k z xs)
 -- The foldr/cons rule looks nice, but it can give disastrously
 -- bloated code when commpiling
 --      array (a,b) [(1,2), (2,2), (3,2), ...very long list... ]
 -- i.e. when there are very very long literal lists
--- So I've disabled it for now. We could have special cases
--- for short lists, I suppose.
--- "foldr/cons" forall k z x xs. foldr k z (x:xs) = k x (foldr k z xs)
-
-"foldr/single"  forall k z x. foldr k z [x] = k x z
-"foldr/nil"     forall k z.   foldr k z []  = z
+-- So we disabled it, but have special cases for short lists up
+-- to a completely arbitrary limit of 4.
+--
+-- Note that static lists that are explicitly entered as such in the source,
+-- the compiler desugars them to build (if they are short), and then normal
+-- foldr/build rule fires, see note [Desugaring explicit lists] in DsExpr
 
 "foldr/cons/build" forall k z x (g::forall b. (a->b->b) -> b -> b) .
                            foldr k z (x:build g) = k x (g k z)
diff --git a/testsuite/tests/simplCore/should_run/T2110.hs b/testsuite/tests/simplCore/should_run/T2110.hs
index 610be09..7cde608 100644
--- a/testsuite/tests/simplCore/should_run/T2110.hs
+++ b/testsuite/tests/simplCore/should_run/T2110.hs
@@ -19,6 +19,7 @@ same x y = case reallyUnsafePtrEquality# (unsafeCoerce x) y of
 
 main = do
     let l = [1,2,3]
+        {-# NOINLINE l #-}
     same (fooAge l) l
     same (fooCoerce l) l
     same (fooUnsafeCoerce l) l