diff options
Diffstat (limited to 'src/Simplify.hs')
| -rw-r--r-- | src/Simplify.hs | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/src/Simplify.hs b/src/Simplify.hs index 0ce5594..f8b4b63 100644 --- a/src/Simplify.hs +++ b/src/Simplify.hs @@ -74,13 +74,28 @@ simplify' = \case EIdx0 _ (ELet _ rhs body) -> simplify' (ELet ext rhs (EIdx0 ext body)) EIdx1 _ (ELet _ rhs body) e -> simplify' (ELet ext rhs (EIdx1 ext body (weakenExpr WSink e))) + -- projection down-commuting + EFst _ (ECase _ e1 (EPair _ e2 _) (EPair _ e3 _)) -> + acted $ simplify' $ + ECase ext e1 e2 e3 + ESnd _ (ECase _ e1 (EPair _ _ e2) (EPair _ _ e3)) -> + acted $ simplify' $ + ECase ext e1 e2 e3 + -- TODO: array indexing (index of build, index of fold) -- TODO: beta rules for maybe -- TODO: constant folding for operations - -- TODO: accum of zero, plus of zero + -- TODO: properly concatenate accum/onehot + EAccum SZ _ (EOneHot _ i idx val) acc -> + acted $ simplify' $ + EAccum i idx val acc + EAccum _ _ (EZero _) _ -> (Any True, ENil ext) + EPlus _ (EZero _) e -> acted $ simplify' e + EPlus _ e (EZero _) -> acted $ simplify' e + EOneHot _ SZ _ e -> acted $ simplify' e EVar _ t i -> pure $ EVar ext t i ELet _ a b -> ELet ext <$> simplify' a <*> simplify' b |