diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-10-22 22:02:06 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-10-22 22:02:06 +0200 |
| commit | 79e072eddf0ec2a97ca455c27cb5ff6f2132bbab (patch) | |
| tree | 2099dc7e9d9a1109d844bca73277ca82983a02c2 /src/AST/Count.hs | |
| parent | e7d7ac0fd8b81c1d6fae9ab7c1e4654133c631ea (diff) | |
Differentiate Replicate
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 |