Safe Haskell | None |
---|
Vectorise.Monad.Global
- readGEnv :: (GlobalEnv -> a) -> VM a
- setGEnv :: GlobalEnv -> VM ()
- updGEnv :: (GlobalEnv -> GlobalEnv) -> VM ()
- defGlobalVar :: Var -> Var -> VM ()
- lookupVectDecl :: Var -> VM (Maybe (Type, CoreExpr))
- noVectDecl :: Var -> VM Bool
- globalScalarVars :: VM VarSet
- isGlobalScalarVar :: Var -> VM Bool
- globalScalarTyCons :: VM NameSet
- lookupTyCon :: TyCon -> VM (Maybe TyCon)
- defTyConName :: TyCon -> Name -> TyCon -> VM ()
- defTyCon :: TyCon -> TyCon -> VM ()
- globalVectTyCons :: VM (NameEnv TyCon)
- lookupDataCon :: DataCon -> VM (Maybe DataCon)
- defDataCon :: DataCon -> DataCon -> VM ()
- lookupTyConPA :: TyCon -> VM (Maybe Var)
- defTyConPAs :: [(TyCon, Var)] -> VM ()
- lookupTyConPR :: TyCon -> VM (Maybe Var)
Documentation
updGEnv :: (GlobalEnv -> GlobalEnv) -> VM ()
Update the global environment using the provided function.
Vars
defGlobalVar :: Var -> Var -> VM ()
Add a mapping between a global var and its vectorised version to the state.
Vectorisation declarations
lookupVectDecl :: Var -> VM (Maybe (Type, CoreExpr))
Check whether a variable has a (non-scalar) vectorisation declaration.
noVectDecl :: Var -> VM Bool
Check whether a variable has a NOVECTORISE
declaration.
Scalars
Get the set of global scalar variables.
isGlobalScalarVar :: Var -> VM Bool
Check whether a given variable is in the set of global scalar variables.
globalScalarTyCons :: VM NameSet
Get the set of global scalar type constructors including both those scalar type constructors declared in an imported module and those declared in the current module.
TyCons
lookupTyCon :: TyCon -> VM (Maybe TyCon)
Lookup the vectorised version of a TyCon
from the global environment.
defTyConName :: TyCon -> Name -> TyCon -> VM ()
Add a mapping between plain and vectorised TyCon
s to the global environment.
The second argument is only to enable tracing for (mutually) recursively defined type constructors, where we must not pull at the vectorised type constructors (because that would pull too early at the recursive knot).
defTyCon :: TyCon -> TyCon -> VM ()
Add a mapping between plain and vectorised TyCon
s to the global environment.
globalVectTyCons :: VM (NameEnv TyCon)
Get the set of all vectorised type constructors.
Datacons
lookupDataCon :: DataCon -> VM (Maybe DataCon)
Lookup the vectorised version of a DataCon
from the global environment.
defDataCon :: DataCon -> DataCon -> VM ()
Add the mapping between plain and vectorised DataCon
s to the global environment.
PA Dictionaries
lookupTyConPA :: TyCon -> VM (Maybe Var)
Lookup the PA
dfun of a vectorised type constructor in the global environment.
defTyConPAs :: [(TyCon, Var)] -> VM ()
Associate vectorised type constructors with the dfun of their PA
instances in the global
environment.
PR Dictionaries
lookupTyConPR :: TyCon -> VM (Maybe Var)