From: Matthijs Kooijman <m.kooijman@student.utwente.nl>
Date: Mon, 17 Aug 2009 09:42:37 +0000 (+0200)
Subject: Revert "Limit flattenLets to non-recursive lets only."
X-Git-Url: https://git.stderr.nl/gitweb?a=commitdiff_plain;h=285f404a1d7c8026aeb6e52f4ec78be905dd9c74;p=matthijs%2Fmaster-project%2Fc%CE%BBash.git

Revert "Limit flattenLets to non-recursive lets only."

This reverts commit de2b45772572a90429ffe41a86a7ba08e7de6d4b.
---

diff --git "a/c\316\273ash/CLasH/Utils/Core/CoreTools.hs" "b/c\316\273ash/CLasH/Utils/Core/CoreTools.hs"
index 3ba2622..094b702 100644
--- "a/c\316\273ash/CLasH/Utils/Core/CoreTools.hs"
+++ "b/c\316\273ash/CLasH/Utils/Core/CoreTools.hs"
@@ -299,17 +299,19 @@ hasStateType expr = case getType expr of
   Just ty -> isStateType ty
 
 
--- | Flattens nested non-recursive lets into a single list of bindings. The
--- expression passed does not have to be a let expression, if it isn't an
--- empty list of bindings is returned.
+-- | Flattens nested lets into a single list of bindings. The expression
+--   passed does not have to be a let expression, if it isn't an empty list of
+--   bindings is returned.
 flattenLets ::
   CoreSyn.CoreExpr -- ^ The expression to flatten.
   -> ([Binding], CoreSyn.CoreExpr) -- ^ The bindings and resulting expression.
-flattenLets (CoreSyn.Let (CoreSyn.NonRec bndr expr) res) =
-  ((bndr, expr):bindings, res')
+flattenLets (CoreSyn.Let binds expr) = 
+  (bindings ++ bindings', expr')
   where
     -- Recursively flatten the contained expression
-    (bindings, res') = flattenLets res
+    (bindings', expr') =flattenLets expr
+    -- Flatten our own bindings to remove the Rec / NonRec constructors
+    bindings = CoreSyn.flattenBinds [binds]
 flattenLets expr = ([], expr)
 
 -- | Create bunch of nested non-recursive let expressions from the given