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/Simplify.hs | |
| parent | e7d7ac0fd8b81c1d6fae9ab7c1e4654133c631ea (diff) | |
Differentiate Replicate
Diffstat (limited to 'src/Simplify.hs')
| -rw-r--r-- | src/Simplify.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/Simplify.hs b/src/Simplify.hs index 2007585..3f4c8e3 100644 --- a/src/Simplify.hs +++ b/src/Simplify.hs @@ -87,7 +87,7 @@ simplify' = \case EConstArr _ n t v -> EConstArr ext n t v EBuild1 _ a b -> EBuild1 ext (simplify' a) (simplify' b) EBuild _ n a b -> EBuild ext n (simplify' a) (simplify' b) - EFold1Inner _ a b -> EFold1Inner ext (simplify' a) (simplify' b) + EFold1Inner _ a b c -> EFold1Inner ext (simplify' a) (simplify' b) (simplify' c) ESum1Inner _ e -> ESum1Inner ext (simplify' e) EUnit _ e -> EUnit ext (simplify' e) EReplicate1Inner _ a b -> EReplicate1Inner ext (simplify' a) (simplify' b) @@ -129,7 +129,7 @@ hasAdds = \case EConstArr _ _ _ _ -> False EBuild1 _ a b -> hasAdds a || hasAdds b EBuild _ _ a b -> hasAdds a || hasAdds b - EFold1Inner _ a b -> hasAdds a || hasAdds b + EFold1Inner _ a b c -> hasAdds a || hasAdds b || hasAdds c ESum1Inner _ e -> hasAdds e EUnit _ e -> hasAdds e EReplicate1Inner _ a b -> hasAdds a || hasAdds b |