summaryrefslogtreecommitdiff
path: root/src/AST/Count.hs
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2024-11-08 12:37:51 +0100
committerTom Smeding <tom@tomsmeding.com>2024-11-08 12:37:51 +0100
commit83692cf41f76272423445c9cbbad65561ee3b50c (patch)
tree49f56f498a68722a7302b4ce0b41402a9b9da9ef /src/AST/Count.hs
parent58d4d0b47f5e609e21132f48b727de37d06b6777 (diff)
WIP custom derivatives
Diffstat (limited to 'src/AST/Count.hs')
-rw-r--r--src/AST/Count.hs1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/AST/Count.hs b/src/AST/Count.hs
index 71b38b1..d365218 100644
--- a/src/AST/Count.hs
+++ b/src/AST/Count.hs
@@ -125,6 +125,7 @@ occCountGeneral onehot unpush alter many = go WId
EIdx _ a b -> re a <> re b
EShape _ e -> re e
EOp _ _ e -> re e
+ ECustom _ _ _ _ _ _ a b -> re a <> re b
EWith a b -> re a <> re1 b
EAccum _ a b e -> re a <> re b <> re e
EZero _ -> mempty