Vectorisation of method types

[ghc-hetmet.git] / compiler / vectorise / VectBuiltIn.hs

diff --git a/compiler/vectorise/VectBuiltIn.hs b/compiler/vectorise/VectBuiltIn.hs
index 65579b4..059b575 100644 (file)
--- a/compiler/vectorise/VectBuiltIn.hs
+++ b/compiler/vectorise/VectBuiltIn.hs
@@ -42,7 +42,7 @@ import TysWiredIn      ( unitDataCon,
                          doubleTyCon,
                          boolTyCon, trueDataCon, falseDataCon,
                          parrTyConName )
-import PrelNames       ( word8TyConName, gHC_PARR )
+import PrelNames       ( word8TyConName, gHC_PARR, gHC_CLASSES )
 import BasicTypes      ( Boxity(..) )
 
 import FastString
@@ -77,7 +77,6 @@ data Modules
         , dph_Instances                :: Module
         , dph_Combinators      :: Module
         , dph_Scalar           :: Module
-        , dph_Selector         :: Module
         , dph_Prelude_PArr     :: Module
         , dph_Prelude_Int      :: Module
         , dph_Prelude_Word8    :: Module
@@ -98,7 +97,6 @@ dph_Modules pkg
        , dph_Instances      = mk (fsLit "Data.Array.Parallel.Lifted.Instances")
        , dph_Combinators    = mk (fsLit "Data.Array.Parallel.Lifted.Combinators")
        , dph_Scalar         = mk (fsLit "Data.Array.Parallel.Lifted.Scalar")
-       , dph_Selector       = mk (fsLit "Data.Array.Parallel.Lifted.Selector")
 
        , dph_Prelude_PArr   = mk (fsLit "Data.Array.Parallel.Prelude.Base.PArr")
        , dph_Prelude_Int    = mk (fsLit "Data.Array.Parallel.Prelude.Base.Int")
@@ -251,17 +249,17 @@ initBuiltins pkg
       voidTyCon                <- externalTyCon dph_Repr (fsLit "Void")
       wrapTyCon                <- externalTyCon dph_Repr (fsLit "Wrap")
 
-      -- From dph-common:Data.Array.Parallel.Lifted.Selector
-      sel_tys      <- mapM (externalType dph_Selector)
+      -- From dph-common:Data.Array.Parallel.Lifted.Unboxed
+      sel_tys      <- mapM (externalType dph_Unboxed)
                            (numbered "Sel" 2 mAX_DPH_SUM)
 
-      sel_replicates <- mapM (externalFun dph_Selector)
-                             (numbered "replicate" 2 mAX_DPH_SUM)
+      sel_replicates <- mapM (externalFun dph_Unboxed)
+                             (numbered_hash "replicateSel" 2 mAX_DPH_SUM)
 
-      sel_picks    <- mapM (externalFun dph_Selector)
-                           (numbered "pick" 2 mAX_DPH_SUM)
+      sel_picks    <- mapM (externalFun dph_Unboxed)
+                           (numbered_hash "pickSel" 2 mAX_DPH_SUM)
 
-      sel_tags     <- mapM (externalFun dph_Selector)
+      sel_tags     <- mapM (externalFun dph_Unboxed)
                            (numbered "tagsSel" 2 mAX_DPH_SUM)
 
       sel_els      <- mapM mk_elements
@@ -350,8 +348,8 @@ initBuiltins pkg
                dph_PArray         = dph_PArray
              , dph_Repr           = dph_Repr
              , dph_Closure        = dph_Closure
-             , dph_Selector       = dph_Selector
              , dph_Scalar         = dph_Scalar
+             , dph_Unboxed        = dph_Unboxed
              })
       = dph_Modules pkg
 
@@ -364,10 +362,13 @@ initBuiltins pkg
     numbered :: String -> Int -> Int -> [FastString]
     numbered pfx m n = [mkFastString (pfx ++ show i) | i <- [m..n]]
 
+    numbered_hash :: String -> Int -> Int -> [FastString]
+    numbered_hash pfx m n = [mkFastString (pfx ++ show i ++ "#") | i <- [m..n]]
+
     mk_elements :: (Int, Int) -> DsM ((Int, Int), CoreExpr)
     mk_elements (i,j)
       = do
-          v <- externalVar dph_Selector
+          v <- externalVar dph_Unboxed
              $ mkFastString ("elementsSel" ++ show i ++ "_" ++ show j ++ "#")
           return ((i,j), Var v)
 
@@ -459,11 +460,18 @@ preludeVars (Modules { dph_Combinators    = dph_Combinators
     [ mk dph_Prelude_Bool  (fsLit "andP")  dph_Prelude_Bool (fsLit "andPA")
     , mk dph_Prelude_Bool  (fsLit "orP")   dph_Prelude_Bool (fsLit "orPA")
 
+    , mk gHC_CLASSES (fsLit "not")         dph_Prelude_Bool (fsLit "notV")
+    , mk gHC_CLASSES (fsLit "&&")          dph_Prelude_Bool (fsLit "andV")
+    , mk gHC_CLASSES (fsLit "||")          dph_Prelude_Bool (fsLit "orV")
+
     -- FIXME: temporary
     , mk dph_Prelude_PArr (fsLit "fromPArrayP")       dph_Prelude_PArr (fsLit "fromPArrayPA")
     , mk dph_Prelude_PArr (fsLit "toPArrayP")         dph_Prelude_PArr (fsLit "toPArrayPA")
     , mk dph_Prelude_PArr (fsLit "fromNestedPArrayP") dph_Prelude_PArr (fsLit "fromNestedPArrayPA")
     , mk dph_Prelude_PArr (fsLit "combineP")          dph_Combinators  (fsLit "combine2PA")
+    , mk dph_Prelude_PArr (fsLit "updateP")           dph_Combinators  (fsLit "updatePA")
+    , mk dph_Prelude_PArr (fsLit "bpermuteP")         dph_Combinators  (fsLit "bpermutePA")
+    , mk dph_Prelude_PArr (fsLit "indexedP")          dph_Combinators  (fsLit "indexedPA")
     ]
   where
     mk  = (,,,)