Futhark.IR.Kernels

A representation with flat parallelism via GPU-oriented kernels.

The Lore definition

data Kernels

The phantom data type for the kernels representation.

Instances

Decorations Kernels

Defined in Futhark.IR.Kernels

Associated Types

type LetDec Kernels

type ExpDec Kernels

type BodyDec Kernels

type FParamInfo Kernels

type LParamInfo Kernels

type RetType Kernels

type BranchType Kernels

type Op Kernels

PrettyLore Kernels

Defined in Futhark.IR.Kernels

Methods

ppExpLore :: ExpDec Kernels -> Exp Kernels -> Maybe Doc

ASTLore Kernels

Defined in Futhark.IR.Kernels

Methods

expTypesFromPattern :: (HasScope Kernels m, Monad m) => Pattern Kernels -> m [BranchType Kernels]

Bindable Kernels

Defined in Futhark.IR.Kernels

Methods

mkExpPat :: [Ident] -> [Ident] -> Exp Kernels -> Pattern Kernels

mkExpDec :: Pattern Kernels -> Exp Kernels -> ExpDec Kernels

mkBody :: Stms Kernels -> Result -> Body Kernels

mkLetNames :: (MonadFreshNames m, HasScope Kernels m) => [VName] -> Exp Kernels -> m (Stm Kernels)

BinderOps Kernels

Defined in Futhark.IR.Kernels

Methods

mkExpDecB :: (MonadBinder m, Lore m ~ Kernels) => Pattern Kernels -> Exp Kernels -> m (ExpDec Kernels)

mkBodyB :: (MonadBinder m, Lore m ~ Kernels) => Stms Kernels -> Result -> m (Body Kernels)

mkLetNamesB :: (MonadBinder m, Lore m ~ Kernels) => [VName] -> Exp Kernels -> m (Stm Kernels)

Checkable Kernels

Defined in Futhark.IR.Kernels

Methods

checkExpLore :: ExpDec Kernels -> TypeM Kernels ()

checkBodyLore :: BodyDec Kernels -> TypeM Kernels ()

checkFParamLore :: VName -> FParamInfo Kernels -> TypeM Kernels ()

checkLParamLore :: VName -> LParamInfo Kernels -> TypeM Kernels ()

checkLetBoundLore :: VName -> LetDec Kernels -> TypeM Kernels ()

checkRetType :: [RetType Kernels] -> TypeM Kernels ()

matchPattern :: Pattern (Aliases Kernels) -> Exp (Aliases Kernels) -> TypeM Kernels ()

primFParam :: VName -> PrimType -> TypeM Kernels (FParam (Aliases Kernels))

matchReturnType :: [RetType Kernels] -> Result -> TypeM Kernels ()

matchBranchType :: [BranchType Kernels] -> Body (Aliases Kernels) -> TypeM Kernels ()

matchLoopResult :: [FParam (Aliases Kernels)] -> [FParam (Aliases Kernels)] -> [SubExp] -> TypeM Kernels ()

CheckableOp Kernels

Defined in Futhark.IR.Kernels

Methods

checkOp :: OpWithAliases (Op Kernels) -> TypeM Kernels ()

HasSegOp Kernels

Defined in Futhark.IR.Kernels

Associated Types

type SegOpLevel Kernels

Methods

asSegOp :: Op Kernels -> Maybe (SegOp (SegOpLevel Kernels) Kernels)

segOp :: SegOp (SegOpLevel Kernels) Kernels -> Op Kernels

BinderOps (Wise Kernels)

Defined in Futhark.IR.Kernels.Simplify

Methods

mkExpDecB :: (MonadBinder m, Lore m ~ Wise Kernels) => Pattern (Wise Kernels) -> Exp (Wise Kernels) -> m (ExpDec (Wise Kernels))

mkBodyB :: (MonadBinder m, Lore m ~ Wise Kernels) => Stms (Wise Kernels) -> Result -> m (Body (Wise Kernels))

mkLetNamesB :: (MonadBinder m, Lore m ~ Wise Kernels) => [VName] -> Exp (Wise Kernels) -> m (Stm (Wise Kernels))

HasSOAC (Wise Kernels)

Defined in Futhark.IR.Kernels.Simplify

Methods

asSOAC :: Op (Wise Kernels) -> Maybe (SOAC (Wise Kernels))

soacOp :: SOAC (Wise Kernels) -> Op (Wise Kernels)

HasSegOp (Wise Kernels)

Defined in Futhark.IR.Kernels.Simplify

Associated Types

type SegOpLevel (Wise Kernels)

Methods

asSegOp :: Op (Wise Kernels) -> Maybe (SegOp (SegOpLevel (Wise Kernels)) (Wise Kernels))

segOp :: SegOp (SegOpLevel (Wise Kernels)) (Wise Kernels) -> Op (Wise Kernels)

type LetDec Kernels

Defined in Futhark.IR.Kernels

type LetDec Kernels = Type

type ExpDec Kernels

Defined in Futhark.IR.Kernels

type ExpDec Kernels = ()

type BodyDec Kernels

Defined in Futhark.IR.Kernels

type BodyDec Kernels = ()

type FParamInfo Kernels

Defined in Futhark.IR.Kernels

type FParamInfo Kernels = DeclType

type LParamInfo Kernels

Defined in Futhark.IR.Kernels

type LParamInfo Kernels = Type

type RetType Kernels

Defined in Futhark.IR.Kernels

type RetType Kernels = DeclExtType

type BranchType Kernels

Defined in Futhark.IR.Kernels

type BranchType Kernels = ExtType

type Op Kernels

Defined in Futhark.IR.Kernels

type Op Kernels = HostOp Kernels (SOAC Kernels)

type SegOpLevel Kernels

Defined in Futhark.IR.Kernels

type SegOpLevel Kernels = SegLevel

type SegOpLevel (Wise Kernels)

Defined in Futhark.IR.Kernels.Simplify

type SegOpLevel (Wise Kernels) = SegLevel

Module re-exports

module Futhark.IR.Prop

module Futhark.IR.Traversals

module Futhark.IR.Pretty

module Futhark.IR.Syntax

module Futhark.IR.Kernels.Kernel

module Futhark.IR.Kernels.Sizes

data SOACMapper flore tlore m

Like Mapper, but just for SOACs.

Constructors

SOACMapper

Fields

• mapOnSOACSubExp :: SubExp -> m SubExp
• mapOnSOACLambda :: Lambda flore -> m (Lambda tlore)
• mapOnSOACVName :: VName -> m VName

data Reduce lore

How to compute a single reduction result.

Constructors

Reduce

Fields

• redComm :: Commutativity
• redLambda :: Lambda lore
• redNeutral :: [SubExp]

Instances

Decorations lore => Eq (Reduce lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: Reduce lore -> Reduce lore -> Bool

(/=) :: Reduce lore -> Reduce lore -> Bool

Decorations lore => Ord (Reduce lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: Reduce lore -> Reduce lore -> Ordering

(<) :: Reduce lore -> Reduce lore -> Bool

(<=) :: Reduce lore -> Reduce lore -> Bool

(>) :: Reduce lore -> Reduce lore -> Bool

(>=) :: Reduce lore -> Reduce lore -> Bool

max :: Reduce lore -> Reduce lore -> Reduce lore

min :: Reduce lore -> Reduce lore -> Reduce lore

Decorations lore => Show (Reduce lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> Reduce lore -> ShowS

show :: Reduce lore -> String

showList :: [Reduce lore] -> ShowS

PrettyLore lore => Pretty (Reduce lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

ppr :: Reduce lore -> Doc

pprPrec :: Int -> Reduce lore -> Doc

pprList :: [Reduce lore] -> Doc

data Scan lore

How to compute a single scan result.

Constructors

Scan

Fields

• scanLambda :: Lambda lore
• scanNeutral :: [SubExp]

Instances

Decorations lore => Eq (Scan lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: Scan lore -> Scan lore -> Bool

(/=) :: Scan lore -> Scan lore -> Bool

Decorations lore => Ord (Scan lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: Scan lore -> Scan lore -> Ordering

(<) :: Scan lore -> Scan lore -> Bool

(<=) :: Scan lore -> Scan lore -> Bool

(>) :: Scan lore -> Scan lore -> Bool

(>=) :: Scan lore -> Scan lore -> Bool

max :: Scan lore -> Scan lore -> Scan lore

min :: Scan lore -> Scan lore -> Scan lore

Decorations lore => Show (Scan lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> Scan lore -> ShowS

show :: Scan lore -> String

showList :: [Scan lore] -> ShowS

PrettyLore lore => Pretty (Scan lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

ppr :: Scan lore -> Doc

pprPrec :: Int -> Scan lore -> Doc

pprList :: [Scan lore] -> Doc

data ScremaForm lore

The essential parts of a Screma factored out (everything except the input arrays).

Constructors

ScremaForm [Scan lore] [Reduce lore] (Lambda lore)

Instances

Decorations lore => Eq (ScremaForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: ScremaForm lore -> ScremaForm lore -> Bool

(/=) :: ScremaForm lore -> ScremaForm lore -> Bool

Decorations lore => Ord (ScremaForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: ScremaForm lore -> ScremaForm lore -> Ordering

(<) :: ScremaForm lore -> ScremaForm lore -> Bool

(<=) :: ScremaForm lore -> ScremaForm lore -> Bool

(>) :: ScremaForm lore -> ScremaForm lore -> Bool

(>=) :: ScremaForm lore -> ScremaForm lore -> Bool

max :: ScremaForm lore -> ScremaForm lore -> ScremaForm lore

min :: ScremaForm lore -> ScremaForm lore -> ScremaForm lore

Decorations lore => Show (ScremaForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> ScremaForm lore -> ShowS

show :: ScremaForm lore -> String

showList :: [ScremaForm lore] -> ShowS

data StreamForm lore

What kind of stream is this?

Constructors

Parallel StreamOrd Commutativity (Lambda lore) [SubExp]

Sequential [SubExp]

Instances

Decorations lore => Eq (StreamForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: StreamForm lore -> StreamForm lore -> Bool

(/=) :: StreamForm lore -> StreamForm lore -> Bool

Decorations lore => Ord (StreamForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: StreamForm lore -> StreamForm lore -> Ordering

(<) :: StreamForm lore -> StreamForm lore -> Bool

(<=) :: StreamForm lore -> StreamForm lore -> Bool

(>) :: StreamForm lore -> StreamForm lore -> Bool

(>=) :: StreamForm lore -> StreamForm lore -> Bool

max :: StreamForm lore -> StreamForm lore -> StreamForm lore

min :: StreamForm lore -> StreamForm lore -> StreamForm lore

Decorations lore => Show (StreamForm lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> StreamForm lore -> ShowS

show :: StreamForm lore -> String

showList :: [StreamForm lore] -> ShowS

data StreamOrd

Is the stream chunk required to correspond to a contiguous subsequence of the original input (InOrder) or not? Disorder streams can be more efficient, but not all algorithms work with this.

Constructors

InOrder

Disorder

Instances

Eq StreamOrd

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: StreamOrd -> StreamOrd -> Bool

(/=) :: StreamOrd -> StreamOrd -> Bool

Ord StreamOrd

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: StreamOrd -> StreamOrd -> Ordering

(<) :: StreamOrd -> StreamOrd -> Bool

(<=) :: StreamOrd -> StreamOrd -> Bool

(>) :: StreamOrd -> StreamOrd -> Bool

(>=) :: StreamOrd -> StreamOrd -> Bool

max :: StreamOrd -> StreamOrd -> StreamOrd

min :: StreamOrd -> StreamOrd -> StreamOrd

Show StreamOrd

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> StreamOrd -> ShowS

show :: StreamOrd -> String

showList :: [StreamOrd] -> ShowS

data SOAC lore

A second-order array combinator (SOAC).

Constructors

Stream SubExp (StreamForm lore) (Lambda lore) [VName]

Scatter SubExp (Lambda lore) [VName] [(SubExp, Int, VName)]

Scatter cs length lambda index and value arrays

<input/output arrays along with their sizes and number of values to write for that array>

length is the length of each index array and value array, since they all must be the same length for any fusion to make sense. If you have a list of index-value array pairs of different sizes, you need to use multiple writes instead.

The lambda body returns the output in this manner:

index_0, index_1, ..., index_n, value_0, value_1, ..., value_n

This must be consistent along all Scatter-related optimisations.

Hist SubExp [HistOp lore] (Lambda lore) [VName]

Hist length dest-arrays-and-ops fun arrays

The first SubExp is the length of the input arrays. The first list describes the operations to perform. The Lambda is the bucket function. Finally comes the input images.

Screma SubExp (ScremaForm lore) [VName]

A combination of scan, reduction, and map. The first SubExp is the size of the input arrays.

Instances

Decorations lore => Eq (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

(==) :: SOAC lore -> SOAC lore -> Bool

(/=) :: SOAC lore -> SOAC lore -> Bool

Decorations lore => Ord (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: SOAC lore -> SOAC lore -> Ordering

(<) :: SOAC lore -> SOAC lore -> Bool

(<=) :: SOAC lore -> SOAC lore -> Bool

(>) :: SOAC lore -> SOAC lore -> Bool

(>=) :: SOAC lore -> SOAC lore -> Bool

max :: SOAC lore -> SOAC lore -> SOAC lore

min :: SOAC lore -> SOAC lore -> SOAC lore

Decorations lore => Show (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> SOAC lore -> ShowS

show :: SOAC lore -> String

showList :: [SOAC lore] -> ShowS

PrettyLore lore => Pretty (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

ppr :: SOAC lore -> Doc

pprPrec :: Int -> SOAC lore -> Doc

pprList :: [SOAC lore] -> Doc

TypedOp (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

opType :: HasScope t m => SOAC lore -> m [ExtType]

ASTLore lore => FreeIn (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

freeIn' :: SOAC lore -> FV

ASTLore lore => Substitute (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

substituteNames :: Map VName VName -> SOAC lore -> SOAC lore

ASTLore lore => Rename (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

rename :: SOAC lore -> RenameM (SOAC lore)

ASTLore lore => IsOp (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

safeOp :: SOAC lore -> Bool

cheapOp :: SOAC lore -> Bool

(ASTLore lore, ASTLore (Aliases lore), CanBeAliased (Op lore)) => CanBeAliased (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Associated Types

type OpWithAliases (SOAC lore)

Methods

removeOpAliases :: OpWithAliases (SOAC lore) -> SOAC lore

addOpAliases :: SOAC lore -> OpWithAliases (SOAC lore)

(ASTLore lore, Aliased lore) => AliasedOp (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

opAliases :: SOAC lore -> [Names]

consumedInOp :: SOAC lore -> Names

OpMetrics (Op lore) => OpMetrics (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

opMetrics :: SOAC lore -> MetricsM ()

(ASTLore lore, CanBeWise (Op lore)) => CanBeWise (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Associated Types

type OpWithWisdom (SOAC lore)

Methods

removeOpWisdom :: OpWithWisdom (SOAC lore) -> SOAC lore

Decorations lore => IndexOp (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

Methods

indexOp :: (ASTLore lore0, IndexOp (Op lore0)) => SymbolTable lore0 -> Int -> SOAC lore -> [PrimExp VName] -> Maybe Indexed

(ASTLore lore, CanBeAliased (Op lore), CSEInOp (OpWithAliases (Op lore))) => CSEInOp (SOAC (Aliases lore))

Defined in Futhark.Optimise.CSE

Methods

cseInOp :: SOAC (Aliases lore) -> CSEM lore0 (SOAC (Aliases lore))

type OpWithAliases (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

type OpWithAliases (SOAC lore) = SOAC (Aliases lore)

type OpWithWisdom (SOAC lore)

Defined in Futhark.IR.SOACS.SOAC

type OpWithWisdom (SOAC lore) = SOAC (Wise lore)

scanResults :: [Scan lore] -> Int

How many reduction results are produced by these Scans?

singleScan :: Bindable lore => [Scan lore] -> Scan lore

Combine multiple scan operators to a single operator.

redResults :: [Reduce lore] -> Int

How many reduction results are produced by these Reduces?

singleReduce :: Bindable lore => [Reduce lore] -> Reduce lore

Combine multiple reduction operators to a single operator.

scremaType :: SubExp -> ScremaForm lore -> [Type]

The types produced by a single Screma, given the size of the input array.

mkIdentityLambda :: (Bindable lore, MonadFreshNames m) => [Type] -> m (Lambda lore)

Construct a lambda that takes parameters of the given types and simply returns them unchanged.

isIdentityLambda :: Lambda lore -> Bool

Is the given lambda an identity lambda?

nilFn :: Bindable lore => Lambda lore

A lambda with no parameters that returns no values.

scanomapSOAC :: [Scan lore] -> Lambda lore -> ScremaForm lore

Construct a Screma with possibly multiple scans, and the given map function.

redomapSOAC :: [Reduce lore] -> Lambda lore -> ScremaForm lore

Construct a Screma with possibly multiple reductions, and the given map function.

scanSOAC :: (Bindable lore, MonadFreshNames m) => [Scan lore] -> m (ScremaForm lore)

Construct a Screma with possibly multiple scans, and identity map function.

reduceSOAC :: (Bindable lore, MonadFreshNames m) => [Reduce lore] -> m (ScremaForm lore)

Construct a Screma with possibly multiple reductions, and identity map function.

mapSOAC :: Lambda lore -> ScremaForm lore

Construct a Screma corresponding to a map.

isScanomapSOAC :: ScremaForm lore -> Maybe ([Scan lore], Lambda lore)

Does this Screma correspond to a scan-map composition?

isScanSOAC :: ScremaForm lore -> Maybe [Scan lore]

Does this Screma correspond to pure scan?

isRedomapSOAC :: ScremaForm lore -> Maybe ([Reduce lore], Lambda lore)

Does this Screma correspond to a reduce-map composition?

isReduceSOAC :: ScremaForm lore -> Maybe [Reduce lore]

Does this Screma correspond to a pure reduce?

isMapSOAC :: ScremaForm lore -> Maybe (Lambda lore)

Does this Screma correspond to a simple map, without any reduction or scan results?

identitySOACMapper :: Monad m => SOACMapper lore lore m

A mapper that simply returns the SOAC verbatim.

mapSOACM :: (Applicative m, Monad m) => SOACMapper flore tlore m -> SOAC flore -> m (SOAC tlore)

Map a monadic action across the immediate children of a SOAC. The mapping does not descend recursively into subexpressions and is done left-to-right.

soacType :: SOAC lore -> [Type]

The type of a SOAC.

typeCheckSOAC :: Checkable lore => SOAC (Aliases lore) -> TypeM lore ()

Type-check a SOAC.

getStreamAccums :: StreamForm lore -> [SubExp]

Get Stream's accumulators as a sub-expression list

ppScrema :: (PrettyLore lore, Pretty inp) => SubExp -> ScremaForm lore -> [inp] -> Doc

Prettyprint the given Screma.

ppHist :: (PrettyLore lore, Pretty inp) => SubExp -> [HistOp lore] -> Lambda lore -> [inp] -> Doc

Prettyprint the given histogram operation.