compiler/vectorise/Vectorise/Builtins/Modules.hs

   1
   2 -- | Modules that contain builtin functions used by the vectoriser.
   3 module Vectorise.Builtins.Modules
   4         ( Modules(..)
   5         , dph_Modules
   6         , dph_Orphans)
   7 where
   8 import Module
   9 import FastString
  10
  11 -- | Ids of the modules that contain our DPH builtins.
  12 data Modules
  13         = Modules
  14         { dph_PArray            :: Module
  15         , dph_Repr              :: Module
  16         , dph_Closure           :: Module
  17         , dph_Unboxed           :: Module
  18         , dph_Instances         :: Module
  19         , dph_Combinators       :: Module
  20         , dph_Scalar            :: Module
  21         , dph_PArray_PData      :: Module
  22         , dph_PArray_Scalar     :: Module
  23         , dph_Prelude_PArr      :: Module
  24         , dph_Prelude_Int       :: Module
  25         , dph_Prelude_Word8     :: Module
  26         , dph_Prelude_Double    :: Module
  27         , dph_Prelude_Bool      :: Module
  28         , dph_Prelude_Tuple     :: Module
  29         }
  30
  31
  32 -- | The locations of builtins in the current DPH library.
  33 dph_Modules :: PackageId -> Modules
  34 dph_Modules pkg
  35         = Modules
  36         { dph_PArray         = mk (fsLit "Data.Array.Parallel.Lifted.PArray")
  37         , dph_Repr           = mk (fsLit "Data.Array.Parallel.Lifted.Repr")
  38         , dph_Closure        = mk (fsLit "Data.Array.Parallel.Lifted.Closure")
  39         , dph_Unboxed        = mk (fsLit "Data.Array.Parallel.Lifted.Unboxed")
  40         , dph_Instances      = mk (fsLit "Data.Array.Parallel.Lifted.Instances")
  41         , dph_Combinators    = mk (fsLit "Data.Array.Parallel.Lifted.Combinators")
  42         , dph_Scalar         = mk (fsLit "Data.Array.Parallel.Lifted.Scalar")
  43
  44         , dph_PArray_PData   = mk (fsLit "Data.Array.Parallel.PArray.PData")
  45         , dph_PArray_Scalar  = mk (fsLit "Data.Array.Parallel.PArray.Scalar")
  46
  47         , dph_Prelude_PArr   = mk (fsLit "Data.Array.Parallel.Prelude.Base.PArr")
  48         , dph_Prelude_Int    = mk (fsLit "Data.Array.Parallel.Prelude.Base.Int")
  49         , dph_Prelude_Word8  = mk (fsLit "Data.Array.Parallel.Prelude.Base.Word8")
  50         , dph_Prelude_Double = mk (fsLit "Data.Array.Parallel.Prelude.Base.Double")
  51         , dph_Prelude_Bool   = mk (fsLit "Data.Array.Parallel.Prelude.Base.Bool")
  52         , dph_Prelude_Tuple  = mk (fsLit "Data.Array.Parallel.Prelude.Base.Tuple")
  53         }
  54         where   mk = mkModule pkg . mkModuleNameFS
  55
  56
  57 -- | Project out ids of modules that contain orphan instances that we need to load.
  58 dph_Orphans :: [Modules -> Module]
  59 dph_Orphans = [dph_Repr, dph_Instances]