summaryrefslogtreecommitdiff
path: root/src/Analysis/Identity.hs
diff options
context:
space:
mode:
Diffstat (limited to 'src/Analysis/Identity.hs')
-rw-r--r--src/Analysis/Identity.hs4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/Analysis/Identity.hs b/src/Analysis/Identity.hs
index 5e36dde..095d0fa 100644
--- a/src/Analysis/Identity.hs
+++ b/src/Analysis/Identity.hs
@@ -145,7 +145,7 @@ idana env expr = case expr of
res <- VIArr <$> genId <*> shidsToVec dim shids
pure (res, EBuild res dim e1' e2')
- EFold1Inner _ e1 e2 e3 -> do
+ EFold1Inner _ cm e1 e2 e3 -> do
let t1 = typeOf e1
x1 <- genIds t1
x2 <- genIds t1
@@ -154,7 +154,7 @@ idana env expr = case expr of
(v3, e3') <- idana env e3
let VIArr _ (_ :< sh) = v3
res <- VIArr <$> genId <*> pure sh
- pure (res, EFold1Inner res e1' e2' e3')
+ pure (res, EFold1Inner res cm e1' e2' e3')
ESum1Inner _ e1 -> do
(v1, e1') <- idana env e1