{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE TypeFamilies #-}
module Futhark.IR.SOACS
( SOACS,
Body,
Stm,
Pat,
Exp,
Lambda,
FParam,
LParam,
RetType,
PatElem,
module Futhark.IR.Prop,
module Futhark.IR.Traversals,
module Futhark.IR.Pretty,
module Futhark.IR.Syntax,
module Futhark.IR.SOACS.SOAC,
AST.LambdaT (Lambda),
AST.BodyT (Body),
AST.PatT (Pat),
AST.PatElemT (PatElem),
)
where
import Futhark.Builder
import Futhark.Construct
import Futhark.IR.Pretty
import Futhark.IR.Prop
import Futhark.IR.SOACS.SOAC
import Futhark.IR.Syntax hiding
( Body,
Exp,
FParam,
LParam,
Lambda,
Pat,
PatElem,
RetType,
Stm,
)
import qualified Futhark.IR.Syntax as AST
import Futhark.IR.Traversals
import qualified Futhark.TypeCheck as TypeCheck
data SOACS
instance RepTypes SOACS where
type Op SOACS = SOAC SOACS
instance ASTRep SOACS where
expTypesFromPat :: Pat SOACS -> m [BranchType SOACS]
expTypesFromPat = [ExtType] -> m [ExtType]
forall (m :: * -> *) a. Monad m => a -> m a
return ([ExtType] -> m [ExtType])
-> (PatT Type -> [ExtType]) -> PatT Type -> m [ExtType]
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PatT Type -> [ExtType]
forall dec. Typed dec => PatT dec -> [ExtType]
expExtTypesFromPat
type Exp = AST.Exp SOACS
type Body = AST.Body SOACS
type Stm = AST.Stm SOACS
type Pat = AST.Pat SOACS
type Lambda = AST.Lambda SOACS
type FParam = AST.FParam SOACS
type LParam = AST.LParam SOACS
type RetType = AST.RetType SOACS
type PatElem = AST.PatElem SOACS
instance TypeCheck.CheckableOp SOACS where
checkOp :: OpWithAliases (Op SOACS) -> TypeM SOACS ()
checkOp = OpWithAliases (Op SOACS) -> TypeM SOACS ()
forall rep. Checkable rep => SOAC (Aliases rep) -> TypeM rep ()
typeCheckSOAC
instance TypeCheck.Checkable SOACS
instance Buildable SOACS where
mkBody :: Stms SOACS -> Result -> Body SOACS
mkBody = BodyDec SOACS -> Stms SOACS -> Result -> Body SOACS
forall rep. BodyDec rep -> Stms rep -> Result -> BodyT rep
AST.Body ()
mkExpPat :: [Ident] -> Exp SOACS -> Pat SOACS
mkExpPat [Ident]
merge Exp SOACS
_ = [Ident] -> PatT Type
basicPat [Ident]
merge
mkExpDec :: Pat SOACS -> Exp SOACS -> ExpDec SOACS
mkExpDec Pat SOACS
_ Exp SOACS
_ = ()
mkLetNames :: [VName] -> Exp SOACS -> m (Stm SOACS)
mkLetNames = [VName] -> Exp SOACS -> m (Stm SOACS)
forall rep (m :: * -> *).
(ExpDec rep ~ (), LetDec rep ~ Type, MonadFreshNames m,
TypedOp (Op rep), HasScope rep m) =>
[VName] -> Exp rep -> m (Stm rep)
simpleMkLetNames
instance BuilderOps SOACS
instance PrettyRep SOACS