diff options
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 |