projects
/
ghc-hetmet.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
a6b396b
)
Utility functions for vectorisation
author
Roman Leshchinskiy
<rl@cse.unsw.edu.au>
Tue, 10 Jul 2007 04:23:19 +0000
(
04:23
+0000)
committer
Roman Leshchinskiy
<rl@cse.unsw.edu.au>
Tue, 10 Jul 2007 04:23:19 +0000
(
04:23
+0000)
compiler/vectorise/Vectorise.hs
patch
|
blob
|
history
diff --git
a/compiler/vectorise/Vectorise.hs
b/compiler/vectorise/Vectorise.hs
index
607a44c
..
71514e1
100644
(file)
--- a/
compiler/vectorise/Vectorise.hs
+++ b/
compiler/vectorise/Vectorise.hs
@@
-209,6
+209,12
@@
setLEnv lenv = VM $ \_ genv _ -> return (Yes genv lenv ())
updLEnv :: (LocalEnv -> LocalEnv) -> VM ()
updLEnv f = VM $ \_ genv lenv -> return (Yes genv (f lenv) ())
updLEnv :: (LocalEnv -> LocalEnv) -> VM ()
updLEnv f = VM $ \_ genv lenv -> return (Yes genv (f lenv) ())
+newLocalVar :: FastString -> Type -> VM Var
+newLocalVar fs ty
+ = do
+ u <- liftDs newUnique
+ return $ mkSysLocal fs u ty
+
newTyVar :: FastString -> Kind -> VM Var
newTyVar fs k
= do
newTyVar :: FastString -> Kind -> VM Var
newTyVar fs k
= do
@@
-218,6
+224,10
@@
newTyVar fs k
lookupTyCon :: TyCon -> VM (Maybe TyCon)
lookupTyCon tc = readGEnv $ \env -> lookupNameEnv (global_tycons env) (tyConName tc)
lookupTyCon :: TyCon -> VM (Maybe TyCon)
lookupTyCon tc = readGEnv $ \env -> lookupNameEnv (global_tycons env) (tyConName tc)
+
+extendTyVarPA :: Var -> CoreExpr -> VM ()
+extendTyVarPA tv pa = updLEnv $ \env -> env { local_tyvar_pa = extendVarEnv (local_tyvar_pa env) tv pa }
+
-- ----------------------------------------------------------------------------
-- Bindings
-- ----------------------------------------------------------------------------
-- Bindings