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_PData             :: Module
  22         , dph_Prelude_PArr      :: Module
  23         , dph_Prelude_Int       :: Module
  24         , dph_Prelude_Word8     :: Module
  25         , dph_Prelude_Double    :: Module
  26         , dph_Prelude_Bool      :: Module
  27         , dph_Prelude_Tuple     :: Module
  28         }
  29
  30
  31 -- | The locations of builtins in the current DPH library.
  32 dph_Modules :: PackageId -> Modules
  33 dph_Modules pkg
  34         = Modules
  35         { dph_PArray         = mk (fsLit "Data.Array.Parallel.Lifted.PArray")
  36         , dph_Repr           = mk (fsLit "Data.Array.Parallel.Lifted.Repr")
  37         , dph_Closure        = mk (fsLit "Data.Array.Parallel.Lifted.Closure")
  38         , dph_Unboxed        = mk (fsLit "Data.Array.Parallel.Lifted.Unboxed")
  39         , dph_Instances      = mk (fsLit "Data.Array.Parallel.Lifted.Instances")
  40         , dph_Combinators    = mk (fsLit "Data.Array.Parallel.Lifted.Combinators")
  41         , dph_Scalar         = mk (fsLit "Data.Array.Parallel.Lifted.Scalar")
  42
  43         , dph_PData          = mk (fsLit "Data.Array.Parallel.PArray.PData")
  44
  45         , dph_Prelude_PArr   = mk (fsLit "Data.Array.Parallel.Prelude.Base.PArr")
  46         , dph_Prelude_Int    = mk (fsLit "Data.Array.Parallel.Prelude.Base.Int")
  47         , dph_Prelude_Word8  = mk (fsLit "Data.Array.Parallel.Prelude.Base.Word8")
  48         , dph_Prelude_Double = mk (fsLit "Data.Array.Parallel.Prelude.Base.Double")
  49         , dph_Prelude_Bool   = mk (fsLit "Data.Array.Parallel.Prelude.Base.Bool")
  50         , dph_Prelude_Tuple  = mk (fsLit "Data.Array.Parallel.Prelude.Base.Tuple")
  51         }
  52         where   mk = mkModule pkg . mkModuleNameFS
  53
  54
  55 -- | Project out ids of modules that contain orphan instances that we need to load.
  56 dph_Orphans :: [Modules -> Module]
  57 dph_Orphans = [dph_Repr, dph_Instances]