Data.Array.Knead.Index.Linear

Documentation

class C ix where

Methods

switch :: f Z -> (forall ix0 i. (C ix0, Single i) => f (ix0 :. i)) -> f ix

Instances

C Z

(C ix0, Single i) => C (:. ix0 i)

switchInt :: C ix => f Z -> (forall ix0. C ix0 => f (ix0 :. Int)) -> f ix

intersect :: C sh => Exp (Shape sh) -> Exp (Shape sh) -> Exp (Shape sh)

value :: (C sh, Value val) => T tag sh -> val (T tag sh)

constant :: (C sh, Value val) => val Int -> val (T tag sh)

paramWith :: (C sh, Value val) => T p (T tag sh) -> (forall parameters. (Storable parameters, C parameters) => (p -> parameters) -> (T parameters -> val (T tag sh)) -> a) -> a

tunnel :: C sh => T p (T tag sh) -> Tunnel p (T tag sh)

flattenIndex :: C sh => T (Shape sh) -> T (Index sh) -> CodeGenFunction r (Value Word32)

peek :: C sh => Ptr Word32 -> IO sh

poke :: C sh => Ptr Word32 -> sh -> IO ()

computeSize :: C sh => T (Shape sh) -> CodeGenFunction r (Value Word32)

type family Struct sh

newtype T tag sh

Constructors

Cons

Fields

decons :: sh

Instances

C sh => Storable (T tag sh)

(C sh, StructFields (Struct sh)) => C (T tag sh)

C sh => C (T tag sh)

Decompose tag sh => Decompose (T tag sh)

(~ * tag ShapeTag, ~ * sh Z) => Scalar (T tag sh)

(~ * tag ShapeTag, C sh) => C (T tag sh)

data Z

Constructors

Z

Instances

Eq Z

Ord Z

Read Z

Show Z

C Z

z :: Value val => val (T tag Z)

data tail :. head

Constructors

!tail :. !head

Instances

(Decompose tag sh, Decompose s) => Decompose tag (:. sh s)

(Eq tail, Eq head) => Eq (:. tail head)

(Ord tail, Ord head) => Ord (:. tail head)

(Read tail, Read head) => Read (:. tail head)

(Show tail, Show head) => Show (:. tail head)

(Compose sh, ~ * (Composed sh) (T (Tag (Composed sh)) (Unwrap (Composed sh))), Compose s) => Compose (:. sh s)

(C ix0, Single i) => C (:. ix0 i)

type Shape = T ShapeTag

shape :: sh -> Shape sh

type Index = T IndexTag

index :: ix -> Index ix

cons :: Value val => val (T tag sh) -> val i -> val (T tag (sh :. i))

(#:.) :: Value val => val (T tag sh) -> val i -> val (T tag (sh :. i))

head :: Value val => val (T tag (sh :. i)) -> val i

tail :: Value val => val (T tag (sh :. i)) -> val (T tag sh)

switchR :: Value val => (val (T tag sh) -> val i -> a) -> val (T tag (sh :. i)) -> a

loadMultiValue :: C sh => Value (Ptr (Struct (Struct sh))) -> CodeGenFunction r (T (T tag sh))

storeMultiValue :: C sh => T (T tag sh) -> Value (Ptr (Struct (Struct sh))) -> CodeGenFunction r ()