From 4c9ae47dd5bbd27b1acb6dc5d4a55657ac1f026f Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Thu, 30 Oct 2025 15:58:08 +0100
Subject: Simplify foldD2 to not sum x0 contributions

---
 src/AST/Pretty.hs | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

(limited to 'src/AST/Pretty.hs')

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
-- 
cgit v1.2.3-70-g09d2