diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-09-12 17:31:20 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-09-12 17:31:42 +0200 |
| commit | 36732f84cfade5371248806328791d5066673fb7 (patch) | |
| tree | 68cf208fca197a48e6b0506e783c1bdaf98d2e42 /src/AST/Count.hs | |
| parent | 1f53cea6a1352db125e1897ca574360180be2550 (diff) | |
Interpreter, some operations
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 |