src/Data/Array/Mixed/Internal/Arith.hs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

{-# LANGUAGE ImportQualifiedPost #-}
module Data.Array.Mixed.Internal.Arith (
  module Data.Array.Mixed.Internal.Arith,
  module Data.Array.Strided.Arith,
) where

import Data.Array.Internal qualified as OI
import Data.Array.Internal.RankedG qualified as RG
import Data.Array.Internal.RankedS qualified as RS

import Data.Array.Strided qualified as AS
import Data.Array.Strided.Arith


fromO :: RS.Array n a -> AS.Array n a
fromO (RS.A (RG.A sh (OI.T strides offset vec))) = AS.Array sh strides offset vec

toO :: AS.Array n a -> RS.Array n a
toO (AS.Array sh strides offset vec) = RS.A (RG.A sh (OI.T strides offset vec))

liftO1 :: (AS.Array n a -> AS.Array n' b)
       -> RS.Array n a -> RS.Array n' b
liftO1 f = toO . f . fromO

liftO2 :: (AS.Array n a -> AS.Array n1 b -> AS.Array n2 c)
       -> RS.Array n a -> RS.Array n1 b -> RS.Array n2 c
liftO2 f x y = toO (f (fromO x) (fromO y))