Futhark.IR.GPU

A representation with flat parallelism via GPU-oriented kernels.

Documentation

data GPU

The phantom data type for the kernels representation.

Instances

Analyse GPU

Defined in Futhark.Analysis.AccessPattern

Methods

analyseOp :: Op GPU -> Context GPU -> [VName] -> (Context GPU, IndexTable GPU)

PrimExpAnalysis GPU

Defined in Futhark.Analysis.PrimExp.Table

Methods

opPrimExp :: Scope GPU -> Op GPU -> State PrimExpTable ()

BuilderOps GPU

Defined in Futhark.IR.GPU

Methods

mkExpDecB :: (MonadBuilder m, Rep m ~ GPU) => Pat (LetDec GPU) -> Exp GPU -> m (ExpDec GPU)

mkBodyB :: (MonadBuilder m, Rep m ~ GPU) => Stms GPU -> Result -> m (Body GPU)

mkLetNamesB :: (MonadBuilder m, Rep m ~ GPU) => [VName] -> Exp GPU -> m (Stm GPU)

Buildable GPU

Defined in Futhark.IR.GPU

Methods

mkExpPat :: [Ident] -> Exp GPU -> Pat (LetDec GPU)

mkExpDec :: Pat (LetDec GPU) -> Exp GPU -> ExpDec GPU

mkBody :: Stms GPU -> Result -> Body GPU

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

PrettyRep GPU

Defined in Futhark.IR.GPU

Methods

ppExpDec :: ExpDec GPU -> Exp GPU -> Maybe (Doc a)

ASTRep GPU

Defined in Futhark.IR.GPU

Methods

expTypesFromPat :: (HasScope GPU m, Monad m) => Pat (LetDec GPU) -> m [BranchType GPU]

RepTypes GPU

Defined in Futhark.IR.GPU

Associated Types

type LetDec GPU

type ExpDec GPU

type BodyDec GPU

type FParamInfo GPU

type LParamInfo GPU

type RetType GPU

type BranchType GPU

type OpC GPU :: Type -> Type

HasSegOp GPU

Defined in Futhark.IR.GPU

Associated Types

type SegOpLevel GPU

Methods

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

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

Checkable GPU

Defined in Futhark.IR.GPU

Methods

checkExpDec :: ExpDec GPU -> TypeM GPU ()

checkBodyDec :: BodyDec GPU -> TypeM GPU ()

checkFParamDec :: VName -> FParamInfo GPU -> TypeM GPU ()

checkLParamDec :: VName -> LParamInfo GPU -> TypeM GPU ()

checkLetBoundDec :: VName -> LetDec GPU -> TypeM GPU ()

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

matchPat :: Pat (LetDec (Aliases GPU)) -> Exp (Aliases GPU) -> TypeM GPU ()

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

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

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

matchLoopResult :: [FParam (Aliases GPU)] -> Result -> TypeM GPU ()

checkOp :: Op (Aliases GPU) -> TypeM GPU ()

Transform GPU

Defined in Futhark.Optimise.ArrayLayout.Transform

Methods

onOp :: Monad m => SOACMapper GPU GPU m -> Op GPU -> m (Op GPU)

transformOp :: LayoutTable -> ExpMap GPU -> Stm GPU -> Op GPU -> TransformM GPU (LayoutTable, ExpMap GPU)

Layout GPU

Defined in Futhark.Optimise.ArrayLayout.Layout

Methods

permutationFromDimAccess :: PrimExpTable -> SegOpName -> ArrayName -> IndexExprName -> [DimAccess GPU] -> Maybe Permutation

BuilderOps (Wise GPU)

Defined in Futhark.IR.GPU.Simplify

Methods

mkExpDecB :: (MonadBuilder m, Rep m ~ Wise GPU) => Pat (LetDec (Wise GPU)) -> Exp (Wise GPU) -> m (ExpDec (Wise GPU))

mkBodyB :: (MonadBuilder m, Rep m ~ Wise GPU) => Stms (Wise GPU) -> Result -> m (Body (Wise GPU))

mkLetNamesB :: (MonadBuilder m, Rep m ~ Wise GPU) => [VName] -> Exp (Wise GPU) -> m (Stm (Wise GPU))

HasSOAC (Wise GPU)

Defined in Futhark.IR.GPU.Simplify

Methods

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

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

HasSegOp (Wise GPU)

Defined in Futhark.IR.GPU.Simplify

Associated Types

type SegOpLevel (Wise GPU)

Methods

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

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

TraverseOpStms (Wise GPU)

Defined in Futhark.IR.GPU.Simplify

Methods

traverseOpStms :: Monad m => OpStmsTraverser m (Op (Wise GPU)) (Wise GPU)

type BodyDec GPU

Defined in Futhark.IR.GPU

type BodyDec GPU = ()

type BranchType GPU

Defined in Futhark.IR.GPU

type BranchType GPU = ExtType

type ExpDec GPU

Defined in Futhark.IR.GPU

type ExpDec GPU = ()

type FParamInfo GPU

Defined in Futhark.IR.GPU

type FParamInfo GPU = DeclType

type LParamInfo GPU

Defined in Futhark.IR.GPU

type LParamInfo GPU = Type

type LetDec GPU

Defined in Futhark.IR.GPU

type LetDec GPU = Type

type OpC GPU

Defined in Futhark.IR.GPU

type OpC GPU = HostOp SOAC

type RetType GPU

Defined in Futhark.IR.GPU

type RetType GPU = DeclExtType

type SegOpLevel GPU

Defined in Futhark.IR.GPU

type SegOpLevel GPU = SegLevel

type SegOpLevel (Wise GPU)

Defined in Futhark.IR.GPU.Simplify

type SegOpLevel (Wise GPU) = SegLevel

Module re-exports

module Futhark.IR.Prop

module Futhark.IR.Traversals

module Futhark.IR.Pretty

module Futhark.IR.Syntax

module Futhark.IR.GPU.Op

module Futhark.IR.GPU.Sizes

data Scan rep

How to compute a single scan result.

Constructors

Scan

Fields

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

Instances

RepTypes rep => Show (Scan rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> Scan rep -> ShowS

show :: Scan rep -> String

showList :: [Scan rep] -> ShowS

ASTRep rep => FreeIn (Scan rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

freeIn' :: Scan rep -> FV

RepTypes rep => Eq (Scan rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

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

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

RepTypes rep => Ord (Scan rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: Scan rep -> Scan rep -> Ordering

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

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

(>) :: Scan rep -> Scan rep -> Bool

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

max :: Scan rep -> Scan rep -> Scan rep

min :: Scan rep -> Scan rep -> Scan rep

PrettyRep rep => Pretty (Scan rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

pretty :: Scan rep -> Doc ann

prettyList :: [Scan rep] -> Doc ann

data SOAC rep

A second-order array combinator (SOAC).

Constructors

Stream SubExp [VName] [SubExp] (Lambda rep)

Scatter SubExp [VName] (ScatterSpec VName) (Lambda rep)

Scatter length inputs lambda spec

Scatter maps values from a set of input arrays to indices and values of a set of output arrays. It is able to write multiple values to multiple outputs each of which may have multiple dimensions.

inputs is a list of input arrays, all having size length, elements of which are applied to the lambda function. For instance, if there are two arrays, lambda will get two values as input, one from each array.

spec specifies the result of the lambda and which arrays to write to.

lambda is a function that takes inputs from inputs and returns values according to spec. It returns values in the following manner:

index_0, index_1, ..., index_n, value_0, value_1, ..., value_m

For each output in spec, lambda returns i * j index values and j output values, where i is the number of dimensions (rank) of the given output, and j is the number of output values written to the given output.

For example, given the following scatter specification:

([x1, y1, z1

, 2, arr1), ([x2, y2], 1, arr2)]

lambda will produce 6 (3 * 2) index values and 2 output values for arr1, and 2 (2 * 1) index values and 1 output value for arr2. Additionally, the results are grouped, so the first 6 index values will correspond to the first two output values, and so on. For this example, lambda should return a total of 11 values, 8 index values and 3 output values. See also splitScatterResults.

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

Hist length arrays dest-arrays-and-ops fun

The final lambda produces indexes and values for the HistOps.

JVP (Lambda rep) [SubExp] [SubExp]

VJP (Lambda rep) [SubExp] [SubExp]

Screma SubExp [VName] (ScremaForm rep)

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

Instances

CanBeAliased SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

addOpAliases :: AliasableRep rep => AliasTable -> SOAC rep -> SOAC (Aliases rep)

IsOp SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

safeOp :: ASTRep rep => SOAC rep -> Bool

cheapOp :: ASTRep rep => SOAC rep -> Bool

opDependencies :: ASTRep rep => SOAC rep -> [Names]

AliasedOp SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

opAliases :: Aliased rep => SOAC rep -> [Names]

consumedInOp :: Aliased rep => SOAC rep -> Names

TypedOp SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

opType :: HasScope rep m => SOAC rep -> m [ExtType]

RephraseOp SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

rephraseInOp :: Monad m => Rephraser m from to -> SOAC from -> m (SOAC to)

CanBeWise SOAC

Defined in Futhark.IR.SOACS.SOAC

Methods

addOpWisdom :: Informing rep => SOAC rep -> SOAC (Wise rep)

RepTypes rep => Show (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> SOAC rep -> ShowS

show :: SOAC rep -> String

showList :: [SOAC rep] -> ShowS

OpMetrics (Op rep) => OpMetrics (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

opMetrics :: SOAC rep -> MetricsM ()

RepTypes rep => IndexOp (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

indexOp :: (ASTRep rep0, IndexOp (Op rep0)) => SymbolTable rep0 -> Int -> SOAC rep -> [TPrimExp Int64 VName] -> Maybe Indexed

ASTRep rep => FreeIn (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

freeIn' :: SOAC rep -> FV

(AliasableRep rep, CSEInOp (Op (Aliases rep))) => CSEInOp (SOAC (Aliases rep))

Defined in Futhark.Optimise.CSE

Methods

cseInOp :: SOAC (Aliases rep) -> CSEM rep0 (SOAC (Aliases rep))

ASTRep rep => Rename (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

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

ASTRep rep => Substitute (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

substituteNames :: Map VName VName -> SOAC rep -> SOAC rep

RepTypes rep => Eq (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

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

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

RepTypes rep => Ord (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: SOAC rep -> SOAC rep -> Ordering

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

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

(>) :: SOAC rep -> SOAC rep -> Bool

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

max :: SOAC rep -> SOAC rep -> SOAC rep

min :: SOAC rep -> SOAC rep -> SOAC rep

PrettyRep rep => Pretty (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

pretty :: SOAC rep -> Doc ann

prettyList :: [SOAC rep] -> Doc ann

data Reduce rep

How to compute a single reduction result.

Constructors

Reduce

Fields

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

Instances

RepTypes rep => Show (Reduce rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> Reduce rep -> ShowS

show :: Reduce rep -> String

showList :: [Reduce rep] -> ShowS

ASTRep rep => FreeIn (Reduce rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

freeIn' :: Reduce rep -> FV

RepTypes rep => Eq (Reduce rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

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

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

RepTypes rep => Ord (Reduce rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: Reduce rep -> Reduce rep -> Ordering

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

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

(>) :: Reduce rep -> Reduce rep -> Bool

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

max :: Reduce rep -> Reduce rep -> Reduce rep

min :: Reduce rep -> Reduce rep -> Reduce rep

PrettyRep rep => Pretty (Reduce rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

pretty :: Reduce rep -> Doc ann

prettyList :: [Reduce rep] -> Doc ann

data ScremaForm rep

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

Constructors

ScremaForm

Fields

• scremaScans :: [Scan rep]
• scremaReduces :: [Reduce rep]
• scremaLambda :: Lambda rep

  The "main" lambda of the Screma. For a map, this is equivalent to isMapSOAC. Note that the meaning of the return value of this lambda depends crucially on exactly which Screma this is. The parameters will correspond exactly to elements of the input arrays, however.

Instances

RepTypes rep => Show (ScremaForm rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

showsPrec :: Int -> ScremaForm rep -> ShowS

show :: ScremaForm rep -> String

showList :: [ScremaForm rep] -> ShowS

ASTRep rep => FreeIn (ScremaForm rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

freeIn' :: ScremaForm rep -> FV

RepTypes rep => Eq (ScremaForm rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

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

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

RepTypes rep => Ord (ScremaForm rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

compare :: ScremaForm rep -> ScremaForm rep -> Ordering

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

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

(>) :: ScremaForm rep -> ScremaForm rep -> Bool

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

max :: ScremaForm rep -> ScremaForm rep -> ScremaForm rep

min :: ScremaForm rep -> ScremaForm rep -> ScremaForm rep

type ScatterSpec v = [(Shape, Int, v)]

How the results of a scatter operation should be written. Each element of the list consists of a v (often a VName) specifying which array to scatter to, a ShapeBase describing the shape of that array, and an Int describing how many elements should be written to that array for each invocation of the scatter.

data SOACMapper frep trep m

Like Mapper, but just for SOACs.

Constructors

SOACMapper

Fields

• mapOnSOACSubExp :: SubExp -> m SubExp
• mapOnSOACLambda :: Lambda frep -> m (Lambda trep)
• mapOnSOACVName :: VName -> m VName

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

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

scanResults :: [Scan rep] -> Int

How many reduction results are produced by these Scans?

singleScan :: Buildable rep => [Scan rep] -> Scan rep

Combine multiple scan operators to a single operator.

redResults :: [Reduce rep] -> Int

How many reduction results are produced by these Reduces?

singleReduce :: Buildable rep => [Reduce rep] -> Reduce rep

Combine multiple reduction operators to a single operator.

soacType :: Typed (LParamInfo rep) => SOAC rep -> [Type]

The type of a SOAC.

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

Type-check a SOAC.

mkIdentityLambda :: (Buildable rep, MonadFreshNames m) => [Type] -> m (Lambda rep)

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

isIdentityLambda :: Lambda rep -> Bool

Is the given lambda an identity lambda?

nilFn :: Buildable rep => Lambda rep

A lambda with no parameters that returns no values.

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

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

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

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

scanSOAC :: (Buildable rep, MonadFreshNames m) => [Scan rep] -> m (ScremaForm rep)

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

reduceSOAC :: (Buildable rep, MonadFreshNames m) => [Reduce rep] -> m (ScremaForm rep)

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

mapSOAC :: Lambda rep -> ScremaForm rep

Construct a Screma corresponding to a map.

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

Does this Screma correspond to a scan-map composition?

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

Does this Screma correspond to a reduce-map composition?

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

Does this Screma correspond to pure scan?

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

Does this Screma correspond to a pure reduce?

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

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

ppScrema :: (PrettyRep rep, Pretty inp) => SubExp -> [inp] -> ScremaForm rep -> Doc ann

Prettyprint the given Screma.

ppHist :: (PrettyRep rep, Pretty inp) => SubExp -> [inp] -> [HistOp rep] -> Lambda rep -> Doc ann

Prettyprint the given histogram operation.

ppStream :: (PrettyRep rep, Pretty inp) => SubExp -> [inp] -> [SubExp] -> Lambda rep -> Doc ann

Prettyprint the given Stream.

ppScatter :: (PrettyRep rep, Pretty inp) => SubExp -> [inp] -> [(Shape, Int, VName)] -> Lambda rep -> Doc ann

Prettyprint the given Scatter.

groupScatterResults :: ScatterSpec array -> [a] -> [(Shape, array, [([a], a)])]

groupScatterResults specification results

Blocks the index values and result values of results according to the specification.

This function is used for extracting and grouping the results of a scatter. In the SOACS representation, the lambda inside a Scatter returns all indices and values as one big list. This function groups each value with its corresponding indices (as determined by the Shape of the output array).

The elements of the resulting list correspond to the shape and name of the output parameters, in addition to a list of values written to that output parameter, along with the array indices marking where to write them to.

See Scatter for more information.

groupScatterResults' :: [(Shape, Int, array)] -> [a] -> [([a], a)]

groupScatterResults' specification results

Blocks the index values and result values of results according to the specification. This is the simpler version of groupScatterResults, which doesn't return any information about shapes or output arrays.

See groupScatterResults for more information,

splitScatterResults :: [(Shape, Int, array)] -> [a] -> ([a], [a])

splitScatterResults specification results

Splits the results array into indices and values according to the specification.

See groupScatterResults for more information.

identitySOACMapper :: forall rep m. Monad m => SOACMapper rep rep m

A mapper that simply returns the SOAC verbatim.

mapSOACM :: Monad m => SOACMapper frep trep m -> SOAC frep -> m (SOAC trep)

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.

traverseSOACStms :: Monad m => OpStmsTraverser m (SOAC rep) rep

A helper for defining TraverseOpStms.