type LetDec GPU
Instance details Defined in Futhark.IR.GPU type LetDec GPU = Type
type ExpDec GPU
Instance details Defined in Futhark.IR.GPU type ExpDec GPU = ()
type BodyDec GPU
Instance details Defined in Futhark.IR.GPU type BodyDec GPU = ()
type FParamInfo GPU
Instance details Defined in Futhark.IR.GPU type FParamInfo GPU = DeclType
type LParamInfo GPU
Instance details Defined in Futhark.IR.GPU type LParamInfo GPU = Type
type RetType GPU
Instance details Defined in Futhark.IR.GPU type RetType GPU = DeclExtType
type BranchType GPU
Instance details Defined in Futhark.IR.GPU type BranchType GPU = ExtType
type OpC GPU
Instance details Defined in Futhark.IR.GPU type OpC GPU = HostOp SOAC

HasSegOp GPU Source #

Instance details

Defined in Futhark.IR.GPU

Associated Types

type SegOpLevel GPU
Instance details Defined in Futhark.IR.GPU type SegOpLevel GPU = SegLevel

Methods

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

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

Checkable GPU Source #

Instance details

Defined in Futhark.IR.GPU

Methods

checkExpDec :: ExpDec GPU -> TypeM GPU () Source #

checkBodyDec :: BodyDec GPU -> TypeM GPU () Source #

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

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

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

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

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

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

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

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

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

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

Transform GPU Source #

Instance details

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 Source #

Instance details

Defined in Futhark.Optimise.ArrayLayout.Layout

Methods

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

BuilderOps (Wise GPU) Source #

Instance details

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)) Source #

mkBodyB :: (MonadBuilder m, Rep m ~ Wise GPU, IsResult res) => Stms (Wise GPU) -> [res] -> m (GBody (Wise GPU) res) Source #

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

HasSOAC (Wise GPU) Source #

Instance details

Defined in Futhark.IR.GPU.Simplify

Methods

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

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

HasSegOp (Wise GPU) Source #

Instance details

Defined in Futhark.IR.GPU.Simplify

Associated Types

type SegOpLevel (Wise GPU)
Instance details Defined in Futhark.IR.GPU.Simplify type SegOpLevel (Wise GPU) = SegLevel

Methods

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

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

TraverseOpStms (Wise GPU) Source #

Instance details

Defined in Futhark.IR.GPU.Simplify

Methods

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

type BodyDec GPU Source #

Instance details

Defined in Futhark.IR.GPU

type BodyDec GPU = ()

type BranchType GPU Source #

Instance details

Defined in Futhark.IR.GPU

type BranchType GPU = ExtType

type ExpDec GPU Source #

Instance details

Defined in Futhark.IR.GPU

type ExpDec GPU = ()

type FParamInfo GPU Source #

Instance details

Defined in Futhark.IR.GPU

type FParamInfo GPU = DeclType

type LParamInfo GPU Source #

Instance details

Defined in Futhark.IR.GPU

type LParamInfo GPU = Type

type LetDec GPU Source #

Instance details

Defined in Futhark.IR.GPU

type LetDec GPU = Type

type OpC GPU Source #

Instance details

Defined in Futhark.IR.GPU

type OpC GPU = HostOp SOAC

type RetType GPU Source #

Instance details

Defined in Futhark.IR.GPU

type RetType GPU = DeclExtType

type SegOpLevel GPU Source #

Instance details

Defined in Futhark.IR.GPU

type SegOpLevel GPU = SegLevel

type SegOpLevel (Wise GPU) Source #

Instance details

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 SOAC rep Source #

A second-order array combinator (SOAC).

Constructors

Stream SubExp [VName] [SubExp] (Lambda rep)
Hist SubExp [VName] [HistOp rep] (Lambda rep)	Hist length arrays dest-arrays-and-ops fun The final lambda produces indexes and values for the `HistOp`s.
JVP [SubExp] [SubExp] (Lambda rep)
VJP [SubExp] [SubExp] (Lambda rep)
Screma SubExp [VName] (ScremaForm rep)	A combination of scan, reduction, and map. The first `SubExp` is the size of the input arrays.

Instances

Instances details

CanBeAliased SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods addOpAliases :: AliasableRep rep => AliasTable -> SOAC rep -> SOAC (Aliases rep) Source #
IsOp SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods safeOp :: ASTRep rep => SOAC rep -> Bool Source # cheapOp :: ASTRep rep => SOAC rep -> Bool Source # opDependencies :: ASTRep rep => SOAC rep -> [Names] Source #
AliasedOp SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods opAliases :: Aliased rep => SOAC rep -> [Names] Source # consumedInOp :: Aliased rep => SOAC rep -> Names Source #
TypedOp SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods opType :: HasScope rep m => SOAC rep -> m [ExtType] Source #
RephraseOp SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods rephraseInOp :: Monad m => Rephraser m from to -> SOAC from -> m (SOAC to) Source #
CanBeWise SOAC Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods addOpWisdom :: Informing rep => SOAC rep -> SOAC (Wise rep) Source #
RepTypes rep => Show (SOAC rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods opMetrics :: SOAC rep -> MetricsM () Source #
RepTypes rep => IndexOp (SOAC rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods indexOp :: (ASTRep rep0, IndexOp (Op rep0)) => SymbolTable rep0 -> Int -> SOAC rep -> [TPrimExp Int64 VName] -> Maybe Indexed Source #
ASTRep rep => FreeIn (SOAC rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods freeIn' :: SOAC rep -> FV Source #
(AliasableRep rep, CSEInOp (Op (Aliases rep))) => CSEInOp (SOAC (Aliases rep)) Source #
Instance details Defined in Futhark.Optimise.CSE Methods cseInOp :: SOAC (Aliases rep) -> CSEM rep0 (SOAC (Aliases rep))
ASTRep rep => Rename (SOAC rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods rename :: SOAC rep -> RenameM (SOAC rep) Source #
ASTRep rep => Substitute (SOAC rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods substituteNames :: Map VName VName -> SOAC rep -> SOAC rep Source #
RepTypes rep => Eq (SOAC rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods (==) :: SOAC rep -> SOAC rep -> Bool # (/=) :: SOAC rep -> SOAC rep -> Bool #
RepTypes rep => Ord (SOAC rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods pretty :: SOAC rep -> Doc ann # prettyList :: [SOAC rep] -> Doc ann #

data Scan rep Source #

How to compute a single scan result.

Constructors

Scan
Fields scanLambda :: Lambda rep scanNeutral :: [SubExp]

Instances

Instances details

RepTypes rep => Show (Scan rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods freeIn' :: Scan rep -> FV Source #
RepTypes rep => Eq (Scan rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods (==) :: Scan rep -> Scan rep -> Bool # (/=) :: Scan rep -> Scan rep -> Bool #
RepTypes rep => Ord (Scan rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods pretty :: Scan rep -> Doc ann # prettyList :: [Scan rep] -> Doc ann #

data Reduce rep Source #

How to compute a single reduction result.

Constructors

Reduce
Fields redComm :: Commutativity redLambda :: Lambda rep redNeutral :: [SubExp]

Instances

Instances details

RepTypes rep => Show (Reduce rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods freeIn' :: Reduce rep -> FV Source #
RepTypes rep => Eq (Reduce rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods (==) :: Reduce rep -> Reduce rep -> Bool # (/=) :: Reduce rep -> Reduce rep -> Bool #
RepTypes rep => Ord (Reduce rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods pretty :: Reduce rep -> Doc ann # prettyList :: [Reduce rep] -> Doc ann #

data ScremaForm rep Source #

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

Constructors

ScremaForm
Fields 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. scremaScans :: [Scan rep] scremaReduces :: [Reduce rep]

Instances

Instances details

RepTypes rep => Show (ScremaForm rep) Source #
Instance details 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) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods freeIn' :: ScremaForm rep -> FV Source #
RepTypes rep => Eq (ScremaForm rep) Source #
Instance details Defined in Futhark.IR.SOACS.SOAC Methods (==) :: ScremaForm rep -> ScremaForm rep -> Bool # (/=) :: ScremaForm rep -> ScremaForm rep -> Bool #
RepTypes rep => Ord (ScremaForm rep) Source #
Instance details 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 #

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

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

scanResults :: [Scan rep] -> Int Source #

How many reduction results are produced by these Scans?

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

Combine multiple scan operators to a single operator.

redResults :: [Reduce rep] -> Int Source #

How many reduction results are produced by these Reduces?

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

Combine multiple reduction operators to a single operator.

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

The type of a SOAC.

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

Type-check a SOAC.

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

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

isIdentityLambda :: Lambda rep -> Bool Source #

Is the given lambda an identity lambda?

nilFn :: Buildable rep => Lambda rep Source #

A lambda with no parameters that returns no values.

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

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

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

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

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

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

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

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

mapSOAC :: Lambda rep -> ScremaForm rep Source #

Construct a Screma corresponding to a map.

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

Does this Screma correspond to a scan-map composition?

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

Does this Screma correspond to a reduce-map composition?

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

Does this Screma correspond to pure scan?

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

Does this Screma correspond to a pure reduce?

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

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 Source #

Prettyprint the given Screma.

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

Prettyprint the given histogram operation.

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

Prettyprint the given Stream.

data SOACMapper frep trep (m :: Type -> Type) Source #

Like Mapper, but just for SOACs.

Constructors

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

identitySOACMapper :: forall rep (m :: Type -> Type). Monad m => SOACMapper rep rep m Source #

A mapper that simply returns the SOAC verbatim.

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

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 Source #

A helper for defining TraverseOpStms.