Futhark.IR.Kernels.Kernel

Size operations

data SizeOp

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

Eq SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

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

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

Ord SizeOp

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

Defined in Futhark.IR.Kernels.Kernel

Methods

showsPrec :: Int -> SizeOp -> ShowS

show :: SizeOp -> String

showList :: [SizeOp] -> ShowS

Pretty SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

ppr :: SizeOp -> Doc

pprPrec :: Int -> SizeOp -> Doc

pprList :: [SizeOp] -> Doc

TypedOp SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

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

FreeIn SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: SizeOp -> FV

Substitute SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

substituteNames :: Map VName VName -> SizeOp -> SizeOp

Rename SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

rename :: SizeOp -> RenameM SizeOp

IsOp SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

safeOp :: SizeOp -> Bool

cheapOp :: SizeOp -> Bool

AliasedOp SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

opAliases :: SizeOp -> [Names]

consumedInOp :: SizeOp -> Names

OpMetrics SizeOp

Defined in Futhark.IR.Kernels.Kernel

Methods

opMetrics :: SizeOp -> MetricsM ()

Host operations

data HostOp lore op

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

Constructors

SegOp (SegOp SegLevel lore)

A segmented operation.

SizeOp SizeOp

OtherOp op

Instances

(Decorations lore, Eq op) => Eq (HostOp lore op)

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)

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)

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)

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)

Defined in Futhark.IR.Kernels.Kernel

Methods

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

(ASTLore lore, FreeIn op) => FreeIn (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: HostOp lore op -> FV

(ASTLore lore, Substitute op) => Substitute (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

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

(ASTLore lore, Rename op) => Rename (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

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

(ASTLore lore, IsOp op) => IsOp (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

safeOp :: HostOp lore op -> Bool

cheapOp :: HostOp lore op -> Bool

(CanBeAliased (Op lore), CanBeAliased op, ASTLore lore) => CanBeAliased (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Associated Types

type OpWithAliases (HostOp lore op)

Methods

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

addOpAliases :: HostOp lore op -> OpWithAliases (HostOp lore op)

(Aliased lore, AliasedOp op, ASTLore lore) => AliasedOp (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

opAliases :: HostOp lore op -> [Names]

consumedInOp :: HostOp lore op -> Names

(OpMetrics (Op lore), OpMetrics op) => OpMetrics (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

opMetrics :: HostOp lore op -> MetricsM ()

(CanBeWise (Op lore), CanBeWise op, ASTLore lore) => CanBeWise (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Associated Types

type OpWithWisdom (HostOp lore op)

Methods

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

(ASTLore lore, IndexOp op) => IndexOp (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

Methods

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

SizeSubst (HostOp lore op)

Defined in Futhark.Pass.ExplicitAllocations.Kernels

Methods

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

opIsConst :: HostOp lore op -> Bool

(ASTLore lore, Aliased lore, CSEInOp (Op lore), CSEInOp op) => CSEInOp (HostOp lore op)

Defined in Futhark.Optimise.CSE

Methods

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

type OpWithAliases (HostOp lore op)

Defined in Futhark.IR.Kernels.Kernel

type OpWithAliases (HostOp lore op) = HostOp (Aliases lore) (OpWithAliases op)

type OpWithWisdom (HostOp lore op)

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 ()

SegOp refinements

data SegLevel

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

Constructors

SegThread

Fields

• segNumGroups :: Count NumGroups SubExp
• segGroupSize :: Count GroupSize SubExp
• segVirt :: SegVirt

SegGroup

Fields

• segNumGroups :: Count NumGroups SubExp
• segGroupSize :: Count GroupSize SubExp
• segVirt :: SegVirt

Instances

Eq SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

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

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

Ord SegLevel

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

Defined in Futhark.IR.Kernels.Kernel

Methods

showsPrec :: Int -> SegLevel -> ShowS

show :: SegLevel -> String

showList :: [SegLevel] -> ShowS

Pretty SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

ppr :: SegLevel -> Doc

pprPrec :: Int -> SegLevel -> Doc

pprList :: [SegLevel] -> Doc

FreeIn SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

freeIn' :: SegLevel -> FV

Substitute SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

substituteNames :: Map VName VName -> SegLevel -> SegLevel

Rename SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

rename :: SegLevel -> RenameM SegLevel

Simplifiable SegLevel

Defined in Futhark.IR.Kernels.Kernel

Methods

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

Reexports

module Futhark.IR.Kernels.Sizes

module Futhark.IR.SegOp