futhark-0.20.0: An optimising compiler for a functional, array-oriented language.
Safe HaskellNone
LanguageHaskell2010

Futhark.IR.Kernels.Kernel

Synopsis

Size operations

data SizeOp Source #

A simple size-level query or computation.

Constructors

SplitSpace SplitOrdering SubExp SubExp SubExp

SplitSpace o w i elems_per_thread.

Computes how to divide array elements to threads in a kernel. Returns the number of elements in the chunk that the current thread should take.

w is the length of the outer dimension in the array. i is the current thread index. Each thread takes at most elems_per_thread elements.

If the order o is SplitContiguous, thread with index i should receive elements i*elems_per_tread, i*elems_per_thread + 1, ..., i*elems_per_thread + (elems_per_thread-1).

If the order o is SplitStrided stride, the thread will receive elements i, i+stride, i+2*stride, ..., i+(elems_per_thread-1)*stride.

GetSize Name SizeClass

Produce some runtime-configurable size.

GetSizeMax SizeClass

The maximum size of some class.

CmpSizeLe Name SizeClass SubExp

Compare size (likely a threshold) with some integer value.

CalcNumGroups SubExp Name SubExp

CalcNumGroups w max_num_groups group_size calculates the number of GPU workgroups to use for an input of the given size. The Name is a size name. Note that w is an i64 to avoid overflow issues.

Instances

Instances details
Eq SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

(==) :: SizeOp -> SizeOp -> Bool #

(/=) :: SizeOp -> SizeOp -> Bool #

Ord SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

compare :: SizeOp -> SizeOp -> Ordering #

(<) :: SizeOp -> SizeOp -> Bool #

(<=) :: SizeOp -> SizeOp -> Bool #

(>) :: SizeOp -> SizeOp -> Bool #

(>=) :: SizeOp -> SizeOp -> Bool #

max :: SizeOp -> SizeOp -> SizeOp #

min :: SizeOp -> SizeOp -> SizeOp #

Show SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

showsPrec :: Int -> SizeOp -> ShowS #

show :: SizeOp -> String #

showList :: [SizeOp] -> ShowS #

Pretty SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

ppr :: SizeOp -> Doc #

pprPrec :: Int -> SizeOp -> Doc #

pprList :: [SizeOp] -> Doc #

TypedOp SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opType :: HasScope t m => SizeOp -> m [ExtType] Source #

FreeIn SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: SizeOp -> FV Source #

Substitute SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

substituteNames :: Map VName VName -> SizeOp -> SizeOp Source #

Rename SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

rename :: SizeOp -> RenameM SizeOp Source #

IsOp SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

safeOp :: SizeOp -> Bool Source #

cheapOp :: SizeOp -> Bool Source #

AliasedOp SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opAliases :: SizeOp -> [Names] Source #

consumedInOp :: SizeOp -> Names Source #

OpMetrics SizeOp Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opMetrics :: SizeOp -> MetricsM () Source #

Host operations

data HostOp lore op Source #

A host-level operation; parameterised by what else it can do.

Constructors

SegOp (SegOp SegLevel lore)

A segmented operation.

SizeOp SizeOp 
OtherOp op 

Instances

Instances details
(Decorations lore, Eq op) => Eq (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

(==) :: HostOp lore op -> HostOp lore op -> Bool #

(/=) :: HostOp lore op -> HostOp lore op -> Bool #

(Decorations lore, Ord op) => Ord (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

compare :: HostOp lore op -> HostOp lore op -> Ordering #

(<) :: HostOp lore op -> HostOp lore op -> Bool #

(<=) :: HostOp lore op -> HostOp lore op -> Bool #

(>) :: HostOp lore op -> HostOp lore op -> Bool #

(>=) :: HostOp lore op -> HostOp lore op -> Bool #

max :: HostOp lore op -> HostOp lore op -> HostOp lore op #

min :: HostOp lore op -> HostOp lore op -> HostOp lore op #

(Decorations lore, Show op) => Show (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

showsPrec :: Int -> HostOp lore op -> ShowS #

show :: HostOp lore op -> String #

showList :: [HostOp lore op] -> ShowS #

(PrettyLore lore, Pretty op) => Pretty (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

ppr :: HostOp lore op -> Doc #

pprPrec :: Int -> HostOp lore op -> Doc #

pprList :: [HostOp lore op] -> Doc #

TypedOp op => TypedOp (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opType :: HasScope t m => HostOp lore op -> m [ExtType] Source #

(ASTLore lore, FreeIn op) => FreeIn (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: HostOp lore op -> FV Source #

(ASTLore lore, Substitute op) => Substitute (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

substituteNames :: Map VName VName -> HostOp lore op -> HostOp lore op Source #

(ASTLore lore, Rename op) => Rename (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

rename :: HostOp lore op -> RenameM (HostOp lore op) Source #

(ASTLore lore, IsOp op) => IsOp (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

safeOp :: HostOp lore op -> Bool Source #

cheapOp :: HostOp lore op -> Bool Source #

(CanBeAliased (Op lore), CanBeAliased op, ASTLore lore) => CanBeAliased (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Associated Types

type OpWithAliases (HostOp lore op) Source #

Methods

removeOpAliases :: OpWithAliases (HostOp lore op) -> HostOp lore op Source #

addOpAliases :: AliasTable -> HostOp lore op -> OpWithAliases (HostOp lore op) Source #

(Aliased lore, AliasedOp op, ASTLore lore) => AliasedOp (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opAliases :: HostOp lore op -> [Names] Source #

consumedInOp :: HostOp lore op -> Names Source #

(OpMetrics (Op lore), OpMetrics op) => OpMetrics (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

opMetrics :: HostOp lore op -> MetricsM () Source #

(CanBeWise (Op lore), CanBeWise op, ASTLore lore) => CanBeWise (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Associated Types

type OpWithWisdom (HostOp lore op) Source #

Methods

removeOpWisdom :: OpWithWisdom (HostOp lore op) -> HostOp lore op Source #

(ASTLore lore, IndexOp op) => IndexOp (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

indexOp :: (ASTLore lore0, IndexOp (Op lore0)) => SymbolTable lore0 -> Int -> HostOp lore op -> [TPrimExp Int64 VName] -> Maybe Indexed Source #

SizeSubst (HostOp lore op) Source # 
Instance details

Defined in Futhark.Pass.ExplicitAllocations.Kernels

Methods

opSizeSubst :: PatternT dec -> HostOp lore op -> ChunkMap Source #

opIsConst :: HostOp lore op -> Bool Source #

(ASTLore lore, Aliased lore, CSEInOp (Op lore), CSEInOp op) => CSEInOp (HostOp lore op) Source # 
Instance details

Defined in Futhark.Optimise.CSE

Methods

cseInOp :: HostOp lore op -> CSEM lore0 (HostOp lore op)

type OpWithAliases (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

type OpWithAliases (HostOp lore op) = HostOp (Aliases lore) (OpWithAliases op)
type OpWithWisdom (HostOp lore op) Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

type OpWithWisdom (HostOp lore op) = HostOp (Wise lore) (OpWithWisdom op)

typeCheckHostOp :: Checkable lore => (SegLevel -> OpWithAliases (Op lore) -> TypeM lore ()) -> Maybe SegLevel -> (op -> TypeM lore ()) -> HostOp (Aliases lore) op -> TypeM lore () Source #

SegOp refinements

data SegLevel Source #

At which level the *body* of a SegOp executes.

Constructors

SegThread 

Fields

SegGroup 

Fields

Instances

Instances details
Eq SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

(==) :: SegLevel -> SegLevel -> Bool #

(/=) :: SegLevel -> SegLevel -> Bool #

Ord SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

compare :: SegLevel -> SegLevel -> Ordering #

(<) :: SegLevel -> SegLevel -> Bool #

(<=) :: SegLevel -> SegLevel -> Bool #

(>) :: SegLevel -> SegLevel -> Bool #

(>=) :: SegLevel -> SegLevel -> Bool #

max :: SegLevel -> SegLevel -> SegLevel #

min :: SegLevel -> SegLevel -> SegLevel #

Show SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

showsPrec :: Int -> SegLevel -> ShowS #

show :: SegLevel -> String #

showList :: [SegLevel] -> ShowS #

Pretty SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

ppr :: SegLevel -> Doc #

pprPrec :: Int -> SegLevel -> Doc #

pprList :: [SegLevel] -> Doc #

FreeIn SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: SegLevel -> FV Source #

Substitute SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

substituteNames :: Map VName VName -> SegLevel -> SegLevel Source #

Rename SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

rename :: SegLevel -> RenameM SegLevel Source #

Simplifiable SegLevel Source # 
Instance details

Defined in Futhark.IR.Kernels.Kernel

Methods

simplify :: SimplifiableLore lore => SegLevel -> SimpleM lore SegLevel Source #

Reexports

module Futhark.IR.Kernels.Sizes

module Futhark.IR.SegOp