第38回　書き換え規則を強力にサポートする段階制御

#define INLINE_U       INLINE
#define INLINE_UP      INLINE
#define INLINE_STREAM  INLINE [1]
#define INLINE_DIST    INLINE [1]
#define INLINE_PA      INLINE
#define INLINE_BACKEND INLINE [2]

#define PHASE_PA
#define PHASE_BACKEND   [2]
#define PHASE_DIST      [1]
#define PHASE_STREAM    [1]

{-# LANGUAGE CPP #-}
#include "fusion-phases.h"
module Data.Array.Parallel.Unlifted.Sequential.Flat.Permute (
  permuteU, permuteMU, bpermuteU, bpermuteDftU, reverseU, updateU,
  atomicUpdateMU
) where
～ 略 ～
bpermuteDftU :: UA e
             => Int                             -- ^ length of result array
             -> (Int -> e)                      -- ^ initialiser function
             -> UArr (Int :*: e)                -- ^ index-value pairs
             -> UArr e
{-# INLINE_U bpermuteDftU #-}
bpermuteDftU n init = updateU (mapU init . enumFromToU 0 $ n-1)
atomicUpdateMU :: UA e => MUArr e s -> UArr (Int :*: e) -> ST s ()
{-# INLINE_U atomicUpdateMU #-}
atomicUpdateMU marr upd = updateM writeMU marr (streamU upd)

～ 略 ～

updateU :: UA e => UArr e -> UArr (Int :*: e) -> UArr e
{-# INLINE_U updateU #-}
updateU arr upd = update (streamU arr) (streamU upd)
update :: UA e => Stream e -> Stream (Int :*: e) -> UArr e
{-# INLINE_STREAM update #-}
update s1@(Stream _ _ n) !s2 = newDynU n (\marr ->
  do
    i <- unstreamMU marr s1
    updateM writeMU marr s2
    return i
  )

{-# LANGUAGE CPP #-}

#include "fusion-phases.h"

module Data.Array.Parallel.Lifted.Selector
where

～ 略 ～

indicesSel2 :: Sel2 -> U.Array Int
{-# INLINE_BACKEND indicesSel2 #-}
indicesSel2 (Sel2 tags idxs na nb) = idxs

elementsSel2_0 :: Sel2 -> Int
{-# INLINE_BACKEND elementsSel2_0 #-}
elementsSel2_0 (Sel2 tags idxs na nb) = na

elementsSel2_0# :: Sel2 -> Int#
{-# INLINE elementsSel2_0# #-}
elementsSel2_0# (Sel2 _ _ (I# na#) _) = na#

elementsSel2_1 :: Sel2 -> Int
{-# INLINE_BACKEND elementsSel2_1 #-}
elementsSel2_1 (Sel2 tags idxs na nb) = nb