diff options
Diffstat (limited to 'src/AST/Count.hs')
| -rw-r--r-- | src/AST/Count.hs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/AST/Count.hs b/src/AST/Count.hs index ad68685..dbec446 100644 --- a/src/AST/Count.hs +++ b/src/AST/Count.hs @@ -116,7 +116,7 @@ occCountGeneral onehot unpush alter many = go WId EConstArr{} -> mempty EBuild1 _ a b -> re a <> many (re1 b) EBuild _ _ a b -> re a <> many (re1 b) - EFold1Inner _ a b -> many (unpush (unpush (go (WSink .> WSink .> w) a))) <> re b + EFold1Inner _ a b c -> many (unpush (unpush (go (WSink .> WSink .> w) a))) <> re b <> re c ESum1Inner _ e -> re e EUnit _ e -> re e EReplicate1Inner _ a b -> re a <> re b |