summaryrefslogtreecommitdiff
path: root/src/AST/Pretty.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/AST/Pretty.hs')
-rw-r--r--src/AST/Pretty.hs5
1 files changed, 3 insertions, 2 deletions
diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs
index 8f1fe67..d811912 100644
--- a/src/AST/Pretty.hs
+++ b/src/AST/Pretty.hs
@@ -124,14 +124,15 @@ ppExpr' d val = \case
return $ showParen (d > 10) $
showString "build " . a' . showString (" (\\" ++ name ++ " -> ") . e' . showString ")"
- EFold1Inner _ a b -> do
+ EFold1Inner _ a b c -> do
name1 <- genNameIfUsedIn (typeOf a) (IS IZ) a
name2 <- genNameIfUsedIn (typeOf a) IZ a
a' <- ppExpr' 0 (Const name2 `SCons` Const name1 `SCons` val) a
b' <- ppExpr' 11 val b
+ c' <- ppExpr' 11 val c
return $ showParen (d > 10) $
showString ("fold1i (\\" ++ name1 ++ " " ++ name2 ++ " -> ") . a'
- . showString ") " . b'
+ . showString ") " . b' . showString " " . c'
ESum1Inner _ e -> do
e' <- ppExpr' 11 val e