lookupVar,
maybeCantVectoriseVarM,
dumpVar,
+ addGlobalScalar,
-- * Primitives
lookupPrimPArray,
import DsMonad
import Outputable
import Control.Monad
-
+import VarSet
-- | Run a vectorisation computation.
initV :: PackageId
| otherwise
= cantVectorise "Variable not vectorised:" (ppr var)
+-- local scalars --------------------------------------------------------------
+-- | Check if the variable is a locally defined scalar function
+
+addGlobalScalar :: Var -> VM ()
+addGlobalScalar var
+ = updGEnv $ \env -> pprTrace "addGLobalScalar" (ppr var) env{global_scalars = extendVarSet (global_scalars env) var}
+
-- Primitives -----------------------------------------------------------------
lookupPrimPArray :: TyCon -> VM (Maybe TyCon)
lookupPrimPArray = liftBuiltinDs . primPArray