Language.Hakaru.CodeGen.CodeGenMonad

as some useful helper functions for manipulating it

Documentation

type CodeGen = State CG

data CG

Constructors

CG

Fields

• freshNames :: [String]

  fresh names for code generations

• reservedNames :: Set String

  reserve names during code generations

• extDecls :: [CExtDecl]

  total external declarations

• declarations :: [CDecl]

  declarations in local block

• statements :: [CStat]

  statements can include assignments as well as other side-effects

• varEnv :: Env

  mapping between Hakaru vars and codegeneration vars

• managedMem :: Bool

  garbage collected block

• sharedMem :: Bool

  shared memory supported block (OpenMP)

• simd :: Bool

  support single instruction multiple data instructions (OpenMP)

• distributed :: Bool

  distributed supported block

• logProbs :: Bool

  true by default, but might not matter in some situations

runCodeGen :: CodeGen a -> ([CExtDecl], [CDecl], [CStat])

runCodeGenBlock :: CodeGen a -> CodeGen CStat

runCodeGenWith :: CodeGen a -> CG -> [CExtDecl]

emptyCG :: CG

declare :: Sing (a :: Hakaru) -> Ident -> CodeGen ()

types like SData and SMeasure are impure in that they will produce extra code in the CodeGenMonad while literal types SReal, SInt, SNat, and SProb do not

declare' :: CDecl -> CodeGen ()

assign :: Ident -> CExpr -> CodeGen ()

putStat :: CStat -> CodeGen ()

putExprStat :: CExpr -> CodeGen ()

extDeclare :: CExtDecl -> CodeGen ()

extDeclareTypes :: Sing (a :: Hakaru) -> CodeGen ()

for types that contain subtypes we need to recursively traverse them and build up a list of external type declarations. For example: Measure (Array Nat) will need to have structures for arrays declared before the top level type

funCG :: [CTypeSpec] -> Ident -> [CDecl] -> CodeGen () -> CodeGen ()

whenPar :: CodeGen () -> CodeGen ()

parDo :: CodeGen a -> CodeGen a

seqDo :: CodeGen a -> CodeGen a

reserveIdent :: String -> CodeGen Ident

genIdent :: CodeGen Ident

genIdent' :: String -> CodeGen Ident

createIdent :: Variable (a :: Hakaru) -> CodeGen Ident

createIdent' :: String -> Variable (a :: Hakaru) -> CodeGen Ident

lookupIdent :: Variable (a :: Hakaru) -> CodeGen Ident

ifCG :: CExpr -> CodeGen () -> CodeGen () -> CodeGen ()

whileCG :: CExpr -> CodeGen () -> CodeGen ()

doWhileCG :: CExpr -> CodeGen () -> CodeGen ()

forCG :: CExpr -> CExpr -> CExpr -> CodeGen () -> CodeGen ()

reductionCG :: Either CBinaryOp (Sing (a :: Hakaru), CExpr -> CodeGen (), CExpr -> CExpr -> CodeGen ()) -> CExpr -> CExpr -> CExpr -> CExpr -> CodeGen () -> CodeGen ()

codeBlockCG :: CodeGen () -> CodeGen ()

Takes a cexpression for the location and size and a hakaru type, and generates a statement for allocating the memory

putMallocStat :: CExpr -> CExpr -> Sing (a :: Hakaru) -> CodeGen ()