Futhark.CodeGen.ImpCode.GPU

Variation of Futhark.CodeGen.ImpCode that contains the notion of a kernel invocation.

Documentation

type Program = Definitions HostOp

A program that calls kernels.

type HostCode = Code HostOp

Host-level code that can call kernels.

type KernelCode = Code KernelOp

Code inside a kernel.

data KernelConst

A run-time constant related to kernels.

Constructors

SizeConst Name SizeClass

SizeMaxConst SizeClass

Instances

Show KernelConst

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> KernelConst -> ShowS

show :: KernelConst -> String

showList :: [KernelConst] -> ShowS

FreeIn KernelConst

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

freeIn' :: KernelConst -> FV

Eq KernelConst

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

(==) :: KernelConst -> KernelConst -> Bool

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

Ord KernelConst

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

compare :: KernelConst -> KernelConst -> Ordering

(<) :: KernelConst -> KernelConst -> Bool

(<=) :: KernelConst -> KernelConst -> Bool

(>) :: KernelConst -> KernelConst -> Bool

(>=) :: KernelConst -> KernelConst -> Bool

max :: KernelConst -> KernelConst -> KernelConst

min :: KernelConst -> KernelConst -> KernelConst

Pretty KernelConst

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

pretty :: KernelConst -> Doc ann

prettyList :: [KernelConst] -> Doc ann

type KernelConstExp = PrimExp KernelConst

An expression whose variables are kernel constants.

data HostOp

An operation that runs on the host (CPU).

Constructors

CallKernel Kernel

GetSize VName Name SizeClass

CmpSizeLe VName Name SizeClass Exp

GetSizeMax VName SizeClass

Instances

Show HostOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> HostOp -> ShowS

show :: HostOp -> String

showList :: [HostOp] -> ShowS

FreeIn HostOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

freeIn' :: HostOp -> FV

Pretty HostOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

pretty :: HostOp -> Doc ann

prettyList :: [HostOp] -> Doc ann

data KernelOp

An operation that occurs within a kernel body.

Constructors

GetBlockId VName Int

GetLocalId VName Int

GetLocalSize VName Int

GetLockstepWidth VName

Atomic Space AtomicOp

Barrier Fence

MemFence Fence

SharedAlloc VName (Count Bytes (TExp Int64))

ErrorSync Fence

Perform a barrier and also check whether any threads have failed an assertion. Make sure all threads would reach all ErrorSyncs if any of them do. A failing assertion will jump to the next following ErrorSync, so make sure it's not inside control flow or similar.

Instances

Show KernelOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> KernelOp -> ShowS

show :: KernelOp -> String

showList :: [KernelOp] -> ShowS

FreeIn KernelOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

freeIn' :: KernelOp -> FV

Pretty KernelOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

pretty :: KernelOp -> Doc ann

prettyList :: [KernelOp] -> Doc ann

data Fence

When we do a barrier or fence, is it at the local or global level? By the Ord instance, global is greater than local.

Constructors

FenceLocal

FenceGlobal

Instances

Show Fence

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> Fence -> ShowS

show :: Fence -> String

showList :: [Fence] -> ShowS

Eq Fence

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

(==) :: Fence -> Fence -> Bool

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

Ord Fence

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

compare :: Fence -> Fence -> Ordering

(<) :: Fence -> Fence -> Bool

(<=) :: Fence -> Fence -> Bool

(>) :: Fence -> Fence -> Bool

(>=) :: Fence -> Fence -> Bool

max :: Fence -> Fence -> Fence

min :: Fence -> Fence -> Fence

data AtomicOp

Atomic operations return the value stored before the update. This old value is stored in the first VName (except for AtomicWrite). The second VName is the memory block to update. The Exp is the new value.

Constructors

AtomicAdd IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicFAdd FloatType VName VName (Count Elements (TExp Int64)) Exp

AtomicSMax IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicSMin IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicUMax IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicUMin IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicAnd IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicOr IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicXor IntType VName VName (Count Elements (TExp Int64)) Exp

AtomicCmpXchg PrimType VName VName (Count Elements (TExp Int64)) Exp Exp

AtomicXchg PrimType VName VName (Count Elements (TExp Int64)) Exp

AtomicWrite PrimType VName (Count Elements (TExp Int64)) Exp

Corresponds to a write followed by a memory fence. The old value is not read.

Instances

Show AtomicOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> AtomicOp -> ShowS

show :: AtomicOp -> String

showList :: [AtomicOp] -> ShowS

FreeIn AtomicOp

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

freeIn' :: AtomicOp -> FV

type BlockDim = Either Exp KernelConstExp

The size of one dimension of a block.

data Kernel

A generic kernel containing arbitrary kernel code.

Constructors

Kernel

Fields

• kernelBody :: Code KernelOp
• kernelUses :: [KernelUse]

  The host variables referenced by the kernel.

• kernelNumBlocks :: [Exp]
• kernelBlockSize :: [BlockDim]
• kernelName :: Name

  A short descriptive and _unique_ name - should be alphanumeric and without spaces.

• kernelFailureTolerant :: Bool

  If true, this kernel does not need to check whether we are in a failing state, as it can cope. Intuitively, it means that the kernel does not depend on any non-scalar parameters to make control flow decisions. Replication, transpose, and copy kernels are examples of this.

• kernelCheckSharedMemory :: Bool

  If true, multi-versioning branches will consider this kernel when considering the shared memory requirements. Set this to false for kernels that do their own checking.

Instances

Show Kernel

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> Kernel -> ShowS

show :: Kernel -> String

showList :: [Kernel] -> ShowS

FreeIn Kernel

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

freeIn' :: Kernel -> FV

Pretty Kernel

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

pretty :: Kernel -> Doc ann

prettyList :: [Kernel] -> Doc ann

data KernelUse

Information about a host-level variable that is used inside this kernel. When generating the actual kernel code, this is used to deduce which parameters are needed.

Constructors

ScalarUse VName PrimType

MemoryUse VName

ConstUse VName KernelConstExp

Instances

Show KernelUse

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

showsPrec :: Int -> KernelUse -> ShowS

show :: KernelUse -> String

showList :: [KernelUse] -> ShowS

Eq KernelUse

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

(==) :: KernelUse -> KernelUse -> Bool

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

Ord KernelUse

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

compare :: KernelUse -> KernelUse -> Ordering

(<) :: KernelUse -> KernelUse -> Bool

(<=) :: KernelUse -> KernelUse -> Bool

(>) :: KernelUse -> KernelUse -> Bool

(>=) :: KernelUse -> KernelUse -> Bool

max :: KernelUse -> KernelUse -> KernelUse

min :: KernelUse -> KernelUse -> KernelUse

Pretty KernelUse

Defined in Futhark.CodeGen.ImpCode.GPU

Methods

pretty :: KernelUse -> Doc ann

prettyList :: [KernelUse] -> Doc ann

module Futhark.CodeGen.ImpCode

module Futhark.IR.GPU.Sizes