#include "MachDeps.h"
module Data.Primitive.SIMD.Word64X8 (Word64X8) where
import Data.Primitive.SIMD.Class
import GHC.Word
import GHC.Types
import GHC.Prim
import GHC.Ptr
import GHC.ST
import Foreign.Storable
import Control.Monad.Primitive
import Data.Primitive.Types
import Data.Primitive.ByteArray
import Data.Primitive.Addr
import Data.Monoid
import Data.Typeable
import qualified Data.Vector.Primitive as PV
import qualified Data.Vector.Primitive.Mutable as PMV
import Data.Vector.Unboxed (Unbox)
import qualified Data.Vector.Unboxed as UV
import Data.Vector.Generic (Vector(..))
import Data.Vector.Generic.Mutable (MVector(..))
#if WORD_SIZE_IN_BITS == 64
type RealWord64# = Word#
#elif WORD_SIZE_IN_BITS == 32
type RealWord64# = Word64#
#else
#error "WORD_SIZE_IN_BITS is neither 64 or 32"
#endif
data Word64X8 = Word64X8 Word64X2# Word64X2# Word64X2# Word64X2# deriving Typeable
abs' :: Word64 -> Word64
abs' (W64# x) = W64# (abs# x)
abs# :: RealWord64# -> RealWord64#
abs# x = case abs (W64# x) of
W64# y -> y
signum' :: Word64 -> Word64
signum' (W64# x) = W64# (signum# x)
signum# :: RealWord64# -> RealWord64#
signum# x = case signum (W64# x) of
W64# y -> y
instance Eq Word64X8 where
a == b = case unpackWord64X8 a of
(x1, x2, x3, x4, x5, x6, x7, x8) -> case unpackWord64X8 b of
(y1, y2, y3, y4, y5, y6, y7, y8) -> x1 == y1 && x2 == y2 && x3 == y3 && x4 == y4 && x5 == y5 && x6 == y6 && x7 == y7 && x8 == y8
instance Ord Word64X8 where
a `compare` b = case unpackWord64X8 a of
(x1, x2, x3, x4, x5, x6, x7, x8) -> case unpackWord64X8 b of
(y1, y2, y3, y4, y5, y6, y7, y8) -> x1 `compare` y1 <> x2 `compare` y2 <> x3 `compare` y3 <> x4 `compare` y4 <> x5 `compare` y5 <> x6 `compare` y6 <> x7 `compare` y7 <> x8 `compare` y8
instance Show Word64X8 where
showsPrec _ a s = case unpackWord64X8 a of
(x1, x2, x3, x4, x5, x6, x7, x8) -> "Word64X8 (" ++ shows x1 (", " ++ shows x2 (", " ++ shows x3 (", " ++ shows x4 (", " ++ shows x5 (", " ++ shows x6 (", " ++ shows x7 (", " ++ shows x8 (")" ++ s))))))))
instance Num Word64X8 where
(+) = plusWord64X8
() = minusWord64X8
(*) = timesWord64X8
negate = mapVector negate
abs = mapVector abs'
signum = mapVector signum'
fromInteger = broadcastVector . fromInteger
instance Bounded Word64X8 where
minBound = broadcastVector minBound
maxBound = broadcastVector maxBound
instance Storable Word64X8 where
sizeOf x = vectorSize x * elementSize x
alignment = sizeOf
peek (Ptr a) = readOffAddr (Addr a) 0
poke (Ptr a) = writeOffAddr (Addr a) 0
instance SIMDVector Word64X8 where
type Elem Word64X8 = Word64
type ElemTuple Word64X8 = (Word64, Word64, Word64, Word64, Word64, Word64, Word64, Word64)
nullVector = broadcastVector 0
vectorSize _ = 8
elementSize _ = 8
broadcastVector = broadcastWord64X8
unsafeInsertVector = unsafeInsertWord64X8
packVector = packWord64X8
unpackVector = unpackWord64X8
mapVector = mapWord64X8
zipVector = zipWord64X8
foldVector = foldWord64X8
sumVector = sumWord64X8
instance SIMDIntVector Word64X8 where
quotVector = quotWord64X8
remVector = remWord64X8
instance Prim Word64X8 where
sizeOf# a = let !(I# x) = sizeOf a in x
alignment# a = let !(I# x) = alignment a in x
indexByteArray# ba i = indexWord64X8Array (ByteArray ba) (I# i)
readByteArray# mba i s = let (ST r) = readWord64X8Array (MutableByteArray mba) (I# i) in r s
writeByteArray# mba i v s = let (ST r) = writeWord64X8Array (MutableByteArray mba) (I# i) v in case r s of { (# s', _ #) -> s' }
setByteArray# mba off n v s = let (ST r) = setByteArrayGeneric (MutableByteArray mba) (I# off) (I# n) v in case r s of { (# s', _ #) -> s' }
indexOffAddr# addr i = indexWord64X8OffAddr (Addr addr) (I# i)
readOffAddr# addr i s = let (ST r) = readWord64X8OffAddr (Addr addr) (I# i) in r s
writeOffAddr# addr i v s = let (ST r) = writeWord64X8OffAddr (Addr addr) (I# i) v in case r s of { (# s', _ #) -> s' }
setOffAddr# addr off n v s = let (ST r) = setOffAddrGeneric (Addr addr) (I# off) (I# n) v in case r s of { (# s', _ #) -> s' }
newtype instance UV.Vector Word64X8 = V_Word64X8 (PV.Vector Word64X8)
newtype instance UV.MVector s Word64X8 = MV_Word64X8 (PMV.MVector s Word64X8)
instance Vector UV.Vector Word64X8 where
basicUnsafeFreeze (MV_Word64X8 v) = V_Word64X8 <$> PV.unsafeFreeze v
basicUnsafeThaw (V_Word64X8 v) = MV_Word64X8 <$> PV.unsafeThaw v
basicLength (V_Word64X8 v) = PV.length v
basicUnsafeSlice start len (V_Word64X8 v) = V_Word64X8(PV.unsafeSlice start len v)
basicUnsafeIndexM (V_Word64X8 v) = PV.unsafeIndexM v
basicUnsafeCopy (MV_Word64X8 m) (V_Word64X8 v) = PV.unsafeCopy m v
elemseq _ = seq
instance MVector UV.MVector Word64X8 where
basicLength (MV_Word64X8 v) = PMV.length v
basicUnsafeSlice start len (MV_Word64X8 v) = MV_Word64X8(PMV.unsafeSlice start len v)
basicOverlaps (MV_Word64X8 v) (MV_Word64X8 w) = PMV.overlaps v w
basicUnsafeNew len = MV_Word64X8 <$> PMV.unsafeNew len
#if MIN_VERSION_vector(0,11,0)
basicInitialize (MV_Word64X8 v) = basicInitialize v
#endif
basicUnsafeRead (MV_Word64X8 v) = PMV.unsafeRead v
basicUnsafeWrite (MV_Word64X8 v) = PMV.unsafeWrite v
instance Unbox Word64X8
broadcastWord64X8 :: Word64 -> Word64X8
broadcastWord64X8 (W64# x) = case broadcastWord64X2# x of
v -> Word64X8 v v v v
packWord64X8 :: (Word64, Word64, Word64, Word64, Word64, Word64, Word64, Word64) -> Word64X8
packWord64X8 (W64# x1, W64# x2, W64# x3, W64# x4, W64# x5, W64# x6, W64# x7, W64# x8) = Word64X8 (packWord64X2# (# x1, x2 #)) (packWord64X2# (# x3, x4 #)) (packWord64X2# (# x5, x6 #)) (packWord64X2# (# x7, x8 #))
unpackWord64X8 :: Word64X8 -> (Word64, Word64, Word64, Word64, Word64, Word64, Word64, Word64)
unpackWord64X8 (Word64X8 m1 m2 m3 m4) = case unpackWord64X2# m1 of
(# x1, x2 #) -> case unpackWord64X2# m2 of
(# x3, x4 #) -> case unpackWord64X2# m3 of
(# x5, x6 #) -> case unpackWord64X2# m4 of
(# x7, x8 #) -> (W64# x1, W64# x2, W64# x3, W64# x4, W64# x5, W64# x6, W64# x7, W64# x8)
unsafeInsertWord64X8 :: Word64X8 -> Word64 -> Int -> Word64X8
unsafeInsertWord64X8 (Word64X8 m1 m2 m3 m4) (W64# y) _i@(I# ip) | _i < 2 = Word64X8 (insertWord64X2# m1 y (ip -# 0#)) m2 m3 m4
| _i < 4 = Word64X8 m1 (insertWord64X2# m2 y (ip -# 2#)) m3 m4
| _i < 6 = Word64X8 m1 m2 (insertWord64X2# m3 y (ip -# 4#)) m4
| otherwise = Word64X8 m1 m2 m3 (insertWord64X2# m4 y (ip -# 6#))
mapWord64X8 :: (Word64 -> Word64) -> Word64X8 -> Word64X8
mapWord64X8 f = mapWord64X8# (\ x -> case f (W64# x) of { W64# y -> y})
mapWord64X8# :: (RealWord64# -> RealWord64#) -> Word64X8 -> Word64X8
mapWord64X8# f = \ v -> case unpackWord64X8 v of
(W64# x1, W64# x2, W64# x3, W64# x4, W64# x5, W64# x6, W64# x7, W64# x8) -> packWord64X8 (W64# (f x1), W64# (f x2), W64# (f x3), W64# (f x4), W64# (f x5), W64# (f x6), W64# (f x7), W64# (f x8))
zipWord64X8 :: (Word64 -> Word64 -> Word64) -> Word64X8 -> Word64X8 -> Word64X8
zipWord64X8 f = \ v1 v2 -> case unpackWord64X8 v1 of
(x1, x2, x3, x4, x5, x6, x7, x8) -> case unpackWord64X8 v2 of
(y1, y2, y3, y4, y5, y6, y7, y8) -> packWord64X8 (f x1 y1, f x2 y2, f x3 y3, f x4 y4, f x5 y5, f x6 y6, f x7 y7, f x8 y8)
foldWord64X8 :: (Word64 -> Word64 -> Word64) -> Word64X8 -> Word64
foldWord64X8 f' = \ v -> case unpackWord64X8 v of
(x1, x2, x3, x4, x5, x6, x7, x8) -> x1 `f` x2 `f` x3 `f` x4 `f` x5 `f` x6 `f` x7 `f` x8
where f !x !y = f' x y
sumWord64X8 :: Word64X8 -> Word64
sumWord64X8 (Word64X8 x1 x2 x3 x4) = case unpackWord64X2# (plusWord64X2# x1 (plusWord64X2# x2 (plusWord64X2# x3 x4))) of
(# y1, y2 #) -> W64# y1 + W64# y2
plusWord64X8 :: Word64X8 -> Word64X8 -> Word64X8
plusWord64X8 (Word64X8 m1_1 m2_1 m3_1 m4_1) (Word64X8 m1_2 m2_2 m3_2 m4_2) = Word64X8 (plusWord64X2# m1_1 m1_2) (plusWord64X2# m2_1 m2_2) (plusWord64X2# m3_1 m3_2) (plusWord64X2# m4_1 m4_2)
minusWord64X8 :: Word64X8 -> Word64X8 -> Word64X8
minusWord64X8 (Word64X8 m1_1 m2_1 m3_1 m4_1) (Word64X8 m1_2 m2_2 m3_2 m4_2) = Word64X8 (minusWord64X2# m1_1 m1_2) (minusWord64X2# m2_1 m2_2) (minusWord64X2# m3_1 m3_2) (minusWord64X2# m4_1 m4_2)
timesWord64X8 :: Word64X8 -> Word64X8 -> Word64X8
timesWord64X8 (Word64X8 m1_1 m2_1 m3_1 m4_1) (Word64X8 m1_2 m2_2 m3_2 m4_2) = Word64X8 (timesWord64X2# m1_1 m1_2) (timesWord64X2# m2_1 m2_2) (timesWord64X2# m3_1 m3_2) (timesWord64X2# m4_1 m4_2)
quotWord64X8 :: Word64X8 -> Word64X8 -> Word64X8
quotWord64X8 (Word64X8 m1_1 m2_1 m3_1 m4_1) (Word64X8 m1_2 m2_2 m3_2 m4_2) = Word64X8 (quotWord64X2# m1_1 m1_2) (quotWord64X2# m2_1 m2_2) (quotWord64X2# m3_1 m3_2) (quotWord64X2# m4_1 m4_2)
remWord64X8 :: Word64X8 -> Word64X8 -> Word64X8
remWord64X8 (Word64X8 m1_1 m2_1 m3_1 m4_1) (Word64X8 m1_2 m2_2 m3_2 m4_2) = Word64X8 (remWord64X2# m1_1 m1_2) (remWord64X2# m2_1 m2_2) (remWord64X2# m3_1 m3_2) (remWord64X2# m4_1 m4_2)
indexWord64X8Array :: ByteArray -> Int -> Word64X8
indexWord64X8Array (ByteArray a) (I# i) = Word64X8 (indexWord64X2Array# a ((i *# 4#) +# 0#)) (indexWord64X2Array# a ((i *# 4#) +# 1#)) (indexWord64X2Array# a ((i *# 4#) +# 2#)) (indexWord64X2Array# a ((i *# 4#) +# 3#))
readWord64X8Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> m Word64X8
readWord64X8Array (MutableByteArray a) (I# i) = primitive (\ s0 -> case readWord64X2Array# a ((i *# 4#) +# 0#) s0 of
(# s1, m1 #) -> case readWord64X2Array# a ((i *# 4#) +# 1#) s1 of
(# s2, m2 #) -> case readWord64X2Array# a ((i *# 4#) +# 2#) s2 of
(# s3, m3 #) -> case readWord64X2Array# a ((i *# 4#) +# 3#) s3 of
(# s4, m4 #) -> (# s4, Word64X8 m1 m2 m3 m4 #))
writeWord64X8Array :: PrimMonad m => MutableByteArray (PrimState m) -> Int -> Word64X8 -> m ()
writeWord64X8Array (MutableByteArray a) (I# i) (Word64X8 m1 m2 m3 m4) = primitive_ (writeWord64X2Array# a ((i *# 4#) +# 0#) m1) >> primitive_ (writeWord64X2Array# a ((i *# 4#) +# 1#) m2) >> primitive_ (writeWord64X2Array# a ((i *# 4#) +# 2#) m3) >> primitive_ (writeWord64X2Array# a ((i *# 4#) +# 3#) m4)
indexWord64X8OffAddr :: Addr -> Int -> Word64X8
indexWord64X8OffAddr (Addr a) (I# i) = Word64X8 (indexWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 0#)) 0#) (indexWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 16#)) 0#) (indexWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 32#)) 0#) (indexWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 48#)) 0#)
readWord64X8OffAddr :: PrimMonad m => Addr -> Int -> m Word64X8
readWord64X8OffAddr (Addr a) (I# i) = primitive (\ s0 -> case (\ addr i' -> readWord64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 64#) +# 0#) s0 of
(# s1, m1 #) -> case (\ addr i' -> readWord64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 64#) +# 16#) s1 of
(# s2, m2 #) -> case (\ addr i' -> readWord64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 64#) +# 32#) s2 of
(# s3, m3 #) -> case (\ addr i' -> readWord64X2OffAddr# (plusAddr# addr i') 0#) a ((i *# 64#) +# 48#) s3 of
(# s4, m4 #) -> (# s4, Word64X8 m1 m2 m3 m4 #))
writeWord64X8OffAddr :: PrimMonad m => Addr -> Int -> Word64X8 -> m ()
writeWord64X8OffAddr (Addr a) (I# i) (Word64X8 m1 m2 m3 m4) = primitive_ (writeWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 0#)) 0# m1) >> primitive_ (writeWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 16#)) 0# m2) >> primitive_ (writeWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 32#)) 0# m3) >> primitive_ (writeWord64X2OffAddr# (plusAddr# a ((i *# 64#) +# 48#)) 0# m4)