diff options
Diffstat (limited to 'src/AST/Count.hs')
| -rw-r--r-- | src/AST/Count.hs | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/AST/Count.hs b/src/AST/Count.hs index 39d26c2..6a00e83 100644 --- a/src/AST/Count.hs +++ b/src/AST/Count.hs @@ -109,11 +109,13 @@ occCountGeneral onehot unpush alter many = go WId EInl _ _ e -> re e EInr _ _ e -> re e ECase _ e a b -> re e <> (re1 a `alter` re1 b) + EConstArr{} -> mempty EBuild1 _ a b -> re a <> many (re1 b) EBuild _ _ a b -> re a <> many (re1 b) - EFold1 _ a b -> many (unpush (unpush (go (WSink .> WSink .> w) a))) <> re b + EFold1Inner _ a b -> many (unpush (unpush (go (WSink .> WSink .> w) a))) <> re b + ESum1Inner _ e -> re e EUnit _ e -> re e - -- EReplicate _ e -> re e + EReplicate1Inner _ a b -> re a <> re b EConst{} -> mempty EIdx0 _ e -> re e EIdx1 _ a b -> re a <> re b |