Merge branch 'sparse'HEADmaster

1 files changed, 9 insertions, 2 deletions

diff --git a/src/Analysis/Identity.hs b/src/Analysis/Identity.hs
index 4501c32..b54946b 100644
--- a/src/Analysis/Identity.hs
+++ b/src/Analysis/Identity.hs
@@ -307,11 +307,11 @@ idana env expr = case expr of
     let res = VIPair v2 x2
     pure (res, EWith res t e1' e2')
 
-  EAccum _ t prj e1 e2 e3 -> do
+  EAccum _ t prj e1 sp e2 e3 -> do
     (_, e1') <- idana env e1
     (_, e2') <- idana env e2
     (_, e3') <- idana env e3
-    pure (VINil, EAccum VINil t prj e1' e2' e3')
+    pure (VINil, EAccum VINil t prj e1' sp e2' e3')
 
   EZero _ t e1 -> do
     -- Approximate the result of EZero to be independent from the zero info
@@ -320,6 +320,13 @@ idana env expr = case expr of
     res <- genIds (fromSMTy t)
     pure (res, EZero res t e1')
 
+  EDeepZero _ t e1 -> do
+    -- Approximate the result of EDeepZero to be independent from the zero info
+    -- expression; not quite true for shape variables
+    (_, e1') <- idana env e1
+    res <- genIds (fromSMTy t)
+    pure (res, EDeepZero res t e1')
+
   EPlus _ t e1 e2 -> do
     (_, e1') <- idana env e1
     (_, e2') <- idana env e2