ghc-7.6.2: The GHC API

Safe HaskellNone

Vectorise.Monad.Global

Contents

Synopsis

Documentation

readGEnv :: (GlobalEnv -> a) -> VM a

Project something from the global environment.

setGEnv :: GlobalEnv -> VM ()

Set the value of the global environment.

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

globalScalarVars :: VM VarSet

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 TyCons 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 TyCons 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 DataCons 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