Handling of recursive scalar functions in isScalarLam

[ghc-hetmet.git] / compiler / vectorise / Vectorise / Monad.hs

diff --git a/compiler/vectorise/Vectorise/Monad.hs b/compiler/vectorise/Vectorise/Monad.hs
index 77b9b7f..2597430 100644 (file)
--- a/compiler/vectorise/Vectorise/Monad.hs
+++ b/compiler/vectorise/Vectorise/Monad.hs
@@ -17,7 +17,8 @@ module Vectorise.Monad (
        maybeCantVectoriseVarM,
        dumpVar,
        addGlobalScalar, 
-
+    deleteGlobalScalar,
+    
        -- * Primitives
        lookupPrimPArray,
        lookupPrimMethod
@@ -146,6 +147,11 @@ addGlobalScalar :: Var -> VM ()
 addGlobalScalar var 
   = updGEnv $ \env -> pprTrace "addGLobalScalar" (ppr var) env{global_scalars = extendVarSet (global_scalars env) var}
      
+deleteGlobalScalar :: Var -> VM ()
+deleteGlobalScalar var 
+  = updGEnv $ \env -> pprTrace "deleteGLobalScalar" (ppr var) env{global_scalars = delVarSet (global_scalars env) var}
+     
+     
 -- Primitives -----------------------------------------------------------------
 lookupPrimPArray :: TyCon -> VM (Maybe TyCon)
 lookupPrimPArray = liftBuiltinDs . primPArray