Futhark.Representation.Kernels

A representation with flat parallelism via GPU-oriented kernels.

The Lore definition

data Kernels

Instances

Annotations Kernels

Defined in Futhark.Representation.Kernels

Associated Types

type LetAttr Kernels

type ExpAttr Kernels

type BodyAttr Kernels

type FParamAttr Kernels

type LParamAttr Kernels

type RetType Kernels

type BranchType Kernels

type Op Kernels

PrettyLore Kernels

Defined in Futhark.Representation.Kernels

Methods

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

Attributes Kernels

Defined in Futhark.Representation.Kernels

Methods

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

Bindable Kernels

Defined in Futhark.Representation.Kernels

Methods

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

mkExpAttr :: Pattern Kernels -> Exp Kernels -> ExpAttr Kernels

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

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

BinderOps Kernels

Defined in Futhark.Representation.Kernels

Methods

mkExpAttrB :: (MonadBinder m, Lore m ~ Kernels) => Pattern Kernels -> Exp Kernels -> m (ExpAttr 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.Representation.Kernels

Methods

checkExpLore :: ExpAttr Kernels -> TypeM Kernels ()

checkBodyLore :: BodyAttr Kernels -> TypeM Kernels ()

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

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

checkLetBoundLore :: VName -> LetAttr 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 ()

CheckableOp Kernels

Defined in Futhark.Representation.Kernels

Methods

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

Scoped Kernels LoopNesting

Defined in Futhark.Pass.ExtractKernels.Distribution

Methods

scopeOf :: LoopNesting -> Scope Kernels

Monad m => LocalScope Kernels (DistNestT m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

localScope :: Scope Kernels -> DistNestT m a -> DistNestT m a

Monad m => HasScope Kernels (DistNestT m)

Defined in Futhark.Pass.ExtractKernels.DistributeNests

Methods

lookupType :: VName -> DistNestT m Type

lookupInfo :: VName -> DistNestT m (NameInfo Kernels)

askScope :: DistNestT m (Scope Kernels)

asksScope :: (Scope Kernels -> a) -> DistNestT m a

BinderOps (Wise Kernels)

Defined in Futhark.Representation.Kernels.Simplify

Methods

mkExpAttrB :: (MonadBinder m, Lore m ~ Wise Kernels) => Pattern (Wise Kernels) -> Exp (Wise Kernels) -> m (ExpAttr (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))

type LetAttr Kernels

Defined in Futhark.Representation.Kernels

type LetAttr Kernels = Type

type ExpAttr Kernels

Defined in Futhark.Representation.Kernels

type ExpAttr Kernels = ()

type BodyAttr Kernels

Defined in Futhark.Representation.Kernels

type BodyAttr Kernels = ()

type FParamAttr Kernels

Defined in Futhark.Representation.Kernels

type FParamAttr Kernels = DeclType

type LParamAttr Kernels

Defined in Futhark.Representation.Kernels

type LParamAttr Kernels = Type

type RetType Kernels

Defined in Futhark.Representation.Kernels

type RetType Kernels = DeclExtType

type BranchType Kernels

Defined in Futhark.Representation.Kernels

type BranchType Kernels = ExtType

type Op Kernels

Defined in Futhark.Representation.Kernels

type Op Kernels = HostOp Kernels (SOAC Kernels)

Module re-exports

module Futhark.Representation.AST.Attributes

module Futhark.Representation.AST.Traversals

module Futhark.Representation.AST.Pretty

module Futhark.Representation.AST.Syntax

module Futhark.Representation.Kernels.Kernel

module Futhark.Representation.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

Constructors

Reduce

Fields

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

Instances

Annotations lore => Eq (Reduce lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

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

Annotations lore => Ord (Reduce lore)

Defined in Futhark.Representation.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

Annotations lore => Show (Reduce lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

showsPrec :: Int -> Reduce lore -> ShowS

show :: Reduce lore -> String

showList :: [Reduce lore] -> ShowS

PrettyLore lore => Pretty (Reduce lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

ppr :: Reduce lore -> Doc

pprPrec :: Int -> Reduce lore -> Doc

pprList :: [Reduce lore] -> Doc

type Scan lore = (Lambda lore, [SubExp])

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

Annotations lore => Eq (ScremaForm lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

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

Annotations lore => Ord (ScremaForm lore)

Defined in Futhark.Representation.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

Annotations lore => Show (ScremaForm lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

showsPrec :: Int -> ScremaForm lore -> ShowS

show :: ScremaForm lore -> String

showList :: [ScremaForm lore] -> ShowS

data StreamForm lore

Constructors

Parallel StreamOrd Commutativity (Lambda lore) [SubExp]

Sequential [SubExp]

Instances

Annotations lore => Eq (StreamForm lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

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

Annotations lore => Ord (StreamForm lore)

Defined in Futhark.Representation.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

Annotations lore => Show (StreamForm lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

showsPrec :: Int -> StreamForm lore -> ShowS

show :: StreamForm lore -> String

showList :: [StreamForm lore] -> ShowS

data SOAC lore

Constructors

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

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

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

Screma SubExp (ScremaForm lore) [VName]

A combination of scan, reduction, and map. The first SubExp is the size of the input arrays. The first Lambda/SubExp pair is for scan and its neutral elements. The second is for the reduction. The final lambda is for the map part, and finally comes the input arrays.

CmpThreshold SubExp String

Instances

Annotations lore => Eq (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

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

Annotations lore => Ord (SOAC lore)

Defined in Futhark.Representation.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

Annotations lore => Show (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

showsPrec :: Int -> SOAC lore -> ShowS

show :: SOAC lore -> String

showList :: [SOAC lore] -> ShowS

PrettyLore lore => Pretty (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

ppr :: SOAC lore -> Doc

pprPrec :: Int -> SOAC lore -> Doc

pprList :: [SOAC lore] -> Doc

TypedOp (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

Attributes lore => FreeIn (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

freeIn' :: SOAC lore -> FV

Attributes lore => Substitute (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

Attributes lore => Rename (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

Attributes lore => IsOp (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

safeOp :: SOAC lore -> Bool

cheapOp :: SOAC lore -> Bool

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

Defined in Futhark.Representation.SOACS.SOAC

Associated Types

type OpWithAliases (SOAC lore)

Methods

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

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

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

Defined in Futhark.Representation.SOACS.SOAC

Methods

opAliases :: SOAC lore -> [Names]

consumedInOp :: SOAC lore -> Names

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

Defined in Futhark.Representation.SOACS.SOAC

Methods

opMetrics :: SOAC lore -> MetricsM ()

(Attributes lore, CanBeRanged (Op lore)) => CanBeRanged (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Associated Types

type OpWithRanges (SOAC lore)

Methods

removeOpRanges :: OpWithRanges (SOAC lore) -> SOAC lore

addOpRanges :: SOAC lore -> OpWithRanges (SOAC lore)

Ranged inner => RangedOp (SOAC inner)

Defined in Futhark.Representation.SOACS.SOAC

Methods

opRanges :: SOAC inner -> [Range]

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

Defined in Futhark.Representation.SOACS.SOAC

Associated Types

type OpWithWisdom (SOAC lore)

Methods

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

Annotations lore => IndexOp (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

Methods

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

(Attributes 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.Representation.SOACS.SOAC

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

type OpWithRanges (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

type OpWithRanges (SOAC lore) = SOAC (Ranges lore)

type OpWithWisdom (SOAC lore)

Defined in Futhark.Representation.SOACS.SOAC

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

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]

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 :: Lambda lore -> [SubExp] -> Lambda lore -> ScremaForm lore

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

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

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

mapSOAC :: Bindable lore => Lambda lore -> ScremaForm lore

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

isScanSOAC :: ScremaForm lore -> Maybe (Lambda lore, [SubExp])

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

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

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

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]

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

getStreamAccums :: StreamForm lore -> [SubExp]

Get Stream's accumulators as a sub-expression list

getStreamOrder :: StreamForm lore -> StreamOrd

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

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