From 36732f84cfade5371248806328791d5066673fb7 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Thu, 12 Sep 2024 17:31:20 +0200 Subject: Interpreter, some operations --- src/AST/Count.hs | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'src/AST/Count.hs') 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 -- cgit v1.2.3-70-g09d2