summaryrefslogtreecommitdiff
path: root/src/AST/Count.hs
diff options
context:
space:
mode:
authorTom Smeding <t.j.smeding@uu.nl>2025-01-27 15:08:02 +0100
committerTom Smeding <t.j.smeding@uu.nl>2025-01-27 15:08:02 +0100
commit88fae8c2914b805a733b71de58ab672124e6069c (patch)
treec155fb1a83ace92aab376202ebc8b4b8a919da7c /src/AST/Count.hs
parent0bdc36d221703e5a2347d3d136d676a86bdb1b6a (diff)
Add ext field to remaining AST constructors
Diffstat (limited to 'src/AST/Count.hs')
-rw-r--r--src/AST/Count.hs10
1 files changed, 5 insertions, 5 deletions
diff --git a/src/AST/Count.hs b/src/AST/Count.hs
index 22a4da6..b7079ff 100644
--- a/src/AST/Count.hs
+++ b/src/AST/Count.hs
@@ -128,11 +128,11 @@ occCountGeneral onehot unpush alter many = go WId
EShape _ e -> re e
EOp _ _ e -> re e
ECustom _ _ _ _ _ _ _ a b -> re a <> re b
- EWith a b -> re a <> re1 b
- EAccum _ a b e -> re a <> re b <> re e
- EZero _ -> mempty
- EPlus _ a b -> re a <> re b
- EOneHot _ _ a b -> re a <> re b
+ EWith _ a b -> re a <> re1 b
+ EAccum _ _ a b e -> re a <> re b <> re e
+ EZero _ _ -> mempty
+ EPlus _ _ a b -> re a <> re b
+ EOneHot _ _ _ a b -> re a <> re b
EError{} -> mempty
where
re :: Monoid (r env') => Expr x env' t'' -> r env'