Feldspar.Optimize

Optimize Feldspar expressions

Documentation

witInteger :: ASTF FeldDomain a -> Maybe (Dict (Integral a, Ord a))

isExact :: ASTF FeldDomain a -> Bool

prj' :: sub :<: sup => sup sig -> Maybe (sub sig)

prj with a stronger constraint to allow using it in bidirectional patterns

pattern SymP :: forall (sub :: Type -> Type) (sup :: Type -> Type) sig. sub :<: sup => TypeRep (DenResult sig) -> sub sig -> AST (sup :&: TypeRepFun) sig

pattern VarP :: forall (sup :: Type -> Type) (info :: Type -> Type) sig. BindingT :<: sup => forall a. (sig ~# Full a, Typeable a) => info (DenResult sig) -> Name -> AST (sup :&: info) sig

pattern LamP :: forall (sup :: Type -> Type) (info :: Type -> Type) sig. BindingT :<: sup => forall a a1 b. ((a :-> sig) ~# (b :-> Full (a1 -> b)), Typeable a1) => info (DenResult (a :-> sig)) -> Name -> AST (sup :&: info) (Full a) -> AST (sup :&: info) sig

pattern LitP :: () => (Eq a, Show a) => TypeRep a -> a -> ASTF FeldDomain a

pattern AddP :: () => (Num a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern SubP :: () => (Num a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern MulP :: () => (Num a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern NegP :: () => (Num a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern QuotP :: () => (Integral a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern RemP :: () => (Integral a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern DivP :: () => (Integral a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern ModP :: () => (Integral a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

pattern DivBalancedP :: () => (Integral a, PrimType' a) => TypeRep a -> ASTF FeldDomain a -> ASTF FeldDomain a -> ASTF FeldDomain a

viewLit :: ASTF FeldDomain a -> Maybe a

pattern NonLitP :: forall a. ASTF FeldDomain a

simplifyUp :: ASTF FeldDomain a -> ASTF FeldDomain a

constFold :: ASTF FeldDomain a -> ASTF FeldDomain a

Reduce an expression to a literal if the following conditions are met:

• The top-most symbol can be evaluated statically (i.g. not a variable or a lifted side-effecting program)
• All immediate sub-terms are literals
• The type of the expression is an allowed type for literals (e.g. not a function)

Note that this function only folds the top-level node of an expressions (if possible). It does not reduce an expression like 1+(2+3) where the sub-expression 2+3 is not a literal.

constArgs :: AST FeldDomain sig -> Bool

Check whether all arguments of a symbol are literals

type OptEnv = Selection AssertionLabel

type Opt = ReaderT OptEnv (Writer (Set Name, Any))

tellVar :: Name -> Opt ()

deleteVar :: Name -> Opt a -> Opt a

tellUnsafe :: Opt ()

simplifyM :: ASTF FeldDomain a -> Opt (ASTF FeldDomain a)

simplify :: OptEnv -> ASTF FeldDomain a -> ASTF FeldDomain a

cmInterface :: CodeMotionInterface FeldDomain

Interface for controlling code motion

optimize :: OptEnv -> ASTF FeldDomain a -> ASTF FeldDomain a

Optimize a Feldspar expression