diff options
Diffstat (limited to 'src/AST/Pretty.hs')
| -rw-r--r-- | src/AST/Pretty.hs | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs index 67197f9..68fc629 100644 --- a/src/AST/Pretty.hs +++ b/src/AST/Pretty.hs @@ -250,23 +250,18 @@ ppExpr' d val expr = case expr of return $ ppParen (d > 10) $ ppApp (annotate AHighlight (ppString opname) <> ppX expr) [ppLam [ppString name1, ppString name2] a', b', c'] - EFold1InnerD2 _ cm ef ez eplus ebog ed -> do + EFold1InnerD2 _ cm ef ebog ed -> do let STArr _ tB = typeOf ebog - t2 = typeOf ez + STArr _ t2 = typeOf ed namef1 <- genNameIfUsedIn tB (IS IZ) ef namef2 <- genNameIfUsedIn t2 IZ ef ef' <- ppExpr' 0 (Const namef2 `SCons` Const namef1 `SCons` val) ef - ez' <- ppExpr' 11 val ez - namep1 <- genNameIfUsedIn t2 (IS IZ) eplus - namep2 <- genNameIfUsedIn t2 IZ eplus - eplus' <- ppExpr' 0 (Const namep2 `SCons` Const namep1 `SCons` val) eplus ebog' <- ppExpr' 11 val ebog ed' <- ppExpr' 11 val ed let opname = "fold1iD2" ++ ppCommut cm return $ ppParen (d > 10) $ ppApp (annotate AHighlight (ppString opname) <> ppX expr) - [ppLam [ppString namef1, ppString namef2] ef', ez' - ,ppLam [ppString namep1, ppString namep2] eplus', ebog', ed'] + [ppLam [ppString namef1, ppString namef2] ef', ebog', ed'] EConst _ ty v | Dict <- scalRepIsShow ty -> return $ ppString (showsPrec d v "") <> ppX expr |