diff options
Diffstat (limited to 'src/AST')
| -rw-r--r-- | src/AST/Count.hs | 2 | ||||
| -rw-r--r-- | src/AST/Pretty.hs | 8 |
2 files changed, 10 insertions, 0 deletions
diff --git a/src/AST/Count.hs b/src/AST/Count.hs index 364773a..22a4da6 100644 --- a/src/AST/Count.hs +++ b/src/AST/Count.hs @@ -119,6 +119,8 @@ occCountGeneral onehot unpush alter many = go WId ESum1Inner _ e -> re e EUnit _ e -> re e EReplicate1Inner _ a b -> re a <> re b + EMaximum1Inner _ e -> re e + EMinimum1Inner _ e -> re e EConst{} -> mempty EIdx0 _ e -> re e EIdx1 _ a b -> re a <> re b diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs index a2232ee..4d9aeec 100644 --- a/src/AST/Pretty.hs +++ b/src/AST/Pretty.hs @@ -141,6 +141,14 @@ ppExpr' d val = \case b' <- ppExpr' 11 val b return $ showParen (d > 10) $ showString "replicate1i " . a' . showString " " . b' + EMaximum1Inner _ e -> do + e' <- ppExpr' 11 val e + return $ showParen (d > 10) $ showString "maximum1i " . e' + + EMinimum1Inner _ e -> do + e' <- ppExpr' 11 val e + return $ showParen (d > 10) $ showString "minimum1i " . e' + EConst _ ty v | Dict <- scalRepIsShow ty -> return $ showsPrec d v |