Data.Array.Knead.Shape.Nested

Documentation

class (C sh, C (Index sh)) => C sh where

Associated Types

type Index sh :: *

Methods

intersectCode :: T sh -> T sh -> CodeGenFunction r (T sh)

sizeCode :: T sh -> CodeGenFunction r (Value Size)

size :: sh -> Int

flattenIndexRec :: T sh -> T (Index sh) -> CodeGenFunction r (Value Size, Value Size)

Result is (size, flattenedIndex). size must equal the result of sizeCode. We use this for sharing intermediate results.

iterator :: Index sh ~ ix => T sh -> T r (T ix)

loop :: (Index sh ~ ix, C ix, Phi state) => (T ix -> state -> CodeGenFunction r state) -> T sh -> state -> CodeGenFunction r state

Instances

C Int8

C Int16

C Int32

C Int64

C Word32

C Word64

C ()

(Integral n, ToSize n, Comparison n) => C (Shifted n)

(Integral n, ToSize n, Comparison n) => C (Range n)

(C n, C m) => C (n, m)

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

(C n, C m, C l) => C (n, m, l)

type Size = Word64

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

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

load :: C sh => f sh -> Value (Ptr (Struct sh)) -> CodeGenFunction r (T sh)

intersect :: C sh => Exp sh -> Exp sh -> Exp sh

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

data Range n

Range denotes an inclusive range like those of the Haskell 98 standard Array type from the array package. E.g. the shape type (Range Int32, Range Int64) is equivalent to the ix type (Int32, Int64) for Arrays.

Constructors

Range n n

Instances

Storable n => Storable (Range n)

C n => C (Range n)

C n => C (Range n)

Compose n => Compose (Range n)

Decompose pn => Decompose (Range pn)

(Integral n, ToSize n, Comparison n) => C (Range n)

data Shifted n

Shifted denotes a range defined by the start index and the length.

Constructors

Shifted

Fields

shiftedOffset :: n

shiftedSize :: n

Instances

Storable n => Storable (Shifted n)

C n => C (Shifted n)

C n => C (Shifted n)

Compose n => Compose (Shifted n)

Decompose pn => Decompose (Shifted pn)

(Integral n, ToSize n, Comparison n) => C (Shifted n)

class C sh => Scalar sh where

Methods

scalar :: Value val => val sh

zeroIndex :: Value val => f sh -> val (Index sh)

Instances

Scalar ()

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

class (C sh, IntegerConstant (Index sh), Additive (Index sh)) => Sequence sh where

Methods

sequenceShapeFromIndex :: T (Index sh) -> CodeGenFunction r (T sh)

Instances

Sequence Int8

Sequence Int16

Sequence Int32

Sequence Int64

Sequence Word32

Sequence Word64