Futhark.CodeGen.Backends.GenericPython

A generic Python code generator which is polymorphic in the type of the operations. Concretely, we use this to handle both sequential and PyOpenCL Python code.

Documentation

compileProg :: MonadFreshNames m => CompilerMode -> String -> Constructor -> [PyStmt] -> [PyStmt] -> Operations op s -> s -> [PyStmt] -> [Option] -> Definitions op -> m Text

data CompilerMode

Are we compiling a library or an executable?

Instances

Show CompilerMode

Defined in Futhark.Compiler.Config

Methods

showsPrec :: Int -> CompilerMode -> ShowS

show :: CompilerMode -> String

showList :: [CompilerMode] -> ShowS

Eq CompilerMode

Defined in Futhark.Compiler.Config

Methods

(==) :: CompilerMode -> CompilerMode -> Bool

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

Ord CompilerMode

Defined in Futhark.Compiler.Config

Methods

compare :: CompilerMode -> CompilerMode -> Ordering

(<) :: CompilerMode -> CompilerMode -> Bool

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

(>) :: CompilerMode -> CompilerMode -> Bool

(>=) :: CompilerMode -> CompilerMode -> Bool

max :: CompilerMode -> CompilerMode -> CompilerMode

min :: CompilerMode -> CompilerMode -> CompilerMode

data Constructor

The class generated by the code generator must have a constructor, although it can be vacuous.

Constructors

Constructor [String] [PyStmt]

emptyConstructor :: Constructor

A constructor that takes no arguments and does nothing.

compileName :: VName -> String

compileVar :: VName -> CompilerM op s PyExp

compileDim :: DimSize -> CompilerM op s PyExp

compileExp :: Exp -> CompilerM op s PyExp

compilePrimExp :: Monad m => (v -> m PyExp) -> PrimExp v -> m PyExp

Tell me how to compile a v, and I'll Compile any PrimExp v for you.

compileCode :: Code op -> CompilerM op s ()

compilePrimValue :: PrimValue -> PyExp

compilePrimType :: PrimType -> String

The ctypes type corresponding to a PrimType.

compilePrimTypeExt :: PrimType -> Signedness -> String

The ctypes type corresponding to a PrimType, taking sign into account.

compilePrimToNp :: PrimType -> String

The Numpy type corresponding to a PrimType.

compilePrimToExtNp :: PrimType -> Signedness -> String

The Numpy type corresponding to a PrimType, taking sign into account.

fromStorage :: PrimType -> PyExp -> PyExp

Convert from storage to scalar representation for the given type.

toStorage :: PrimType -> PyExp -> PyExp

Convert from scalar to storage representation for the given type.

data Operations op s

Constructors

Operations

Fields

• opsWriteScalar :: WriteScalar op s
• opsReadScalar :: ReadScalar op s
• opsAllocate :: Allocate op s
• opsCopies :: Map (Space, Space) (DoLMADCopy op s)

  (dst,src)-space mapping to copy functions.

• opsCompiler :: OpCompiler op s
• opsEntryOutput :: EntryOutput op s
• opsEntryInput :: EntryInput op s

type DoLMADCopy op s = PrimType -> [Count Elements PyExp] -> PyExp -> (Count Elements PyExp, [Count Elements PyExp]) -> PyExp -> (Count Elements PyExp, [Count Elements PyExp]) -> CompilerM op s ()

Perform an LMADCopy. It is expected that these functions are each specialised on which spaces they operate on, so that is not part of their arguments.

defaultOperations :: Operations op s

A set of operations that fail for every operation involving non-default memory spaces. Uses plain pointers and malloc for memory management.

unpackDim :: PyExp -> DimSize -> Int32 -> CompilerM op s ()

newtype CompilerM op s a

Constructors

CompilerM (RWS (CompilerEnv op s) [PyStmt] (CompilerState s) a)

Instances

MonadState (CompilerState s) (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

get :: CompilerM op s (CompilerState s)

put :: CompilerState s -> CompilerM op s ()

state :: (CompilerState s -> (a, CompilerState s)) -> CompilerM op s a

MonadWriter [PyStmt] (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

writer :: (a, [PyStmt]) -> CompilerM op s a

tell :: [PyStmt] -> CompilerM op s ()

listen :: CompilerM op s a -> CompilerM op s (a, [PyStmt])

pass :: CompilerM op s (a, [PyStmt] -> [PyStmt]) -> CompilerM op s a

Applicative (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

pure :: a -> CompilerM op s a

(<*>) :: CompilerM op s (a -> b) -> CompilerM op s a -> CompilerM op s b

liftA2 :: (a -> b -> c) -> CompilerM op s a -> CompilerM op s b -> CompilerM op s c

(*>) :: CompilerM op s a -> CompilerM op s b -> CompilerM op s b

(<*) :: CompilerM op s a -> CompilerM op s b -> CompilerM op s a

Functor (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

fmap :: (a -> b) -> CompilerM op s a -> CompilerM op s b

(<$) :: a -> CompilerM op s b -> CompilerM op s a

Monad (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

(>>=) :: CompilerM op s a -> (a -> CompilerM op s b) -> CompilerM op s b

(>>) :: CompilerM op s a -> CompilerM op s b -> CompilerM op s b

return :: a -> CompilerM op s a

MonadFreshNames (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

getNameSource :: CompilerM op s VNameSource

putNameSource :: VNameSource -> CompilerM op s ()

MonadReader (CompilerEnv op s) (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

ask :: CompilerM op s (CompilerEnv op s)

local :: (CompilerEnv op s -> CompilerEnv op s) -> CompilerM op s a -> CompilerM op s a

reader :: (CompilerEnv op s -> a) -> CompilerM op s a

type OpCompiler op s = op -> CompilerM op s ()

A substitute expression compiler, tried before the main compilation function.

type WriteScalar op s = PyExp -> PyExp -> PrimType -> SpaceId -> PyExp -> CompilerM op s ()

Write a scalar to the given memory block with the given index and in the given memory space.

type ReadScalar op s = PyExp -> PyExp -> PrimType -> SpaceId -> CompilerM op s PyExp

Read a scalar from the given memory block with the given index and in the given memory space.

type Allocate op s = PyExp -> PyExp -> SpaceId -> CompilerM op s ()

Allocate a memory block of the given size in the given memory space, saving a reference in the given variable name.

type Copy op s = PyExp -> PyExp -> Space -> PyExp -> PyExp -> Space -> PyExp -> PrimType -> CompilerM op s ()

Copy from one memory block to another.

type EntryOutput op s = VName -> SpaceId -> PrimType -> Signedness -> [DimSize] -> CompilerM op s PyExp

Construct the Python array being returned from an entry point.

type EntryInput op s = PyExp -> SpaceId -> PrimType -> Signedness -> [DimSize] -> PyExp -> CompilerM op s ()

Unpack the array being passed to an entry point.

data CompilerEnv op s

Constructors

CompilerEnv

Fields

• envOperations :: Operations op s
• envVarExp :: Map String PyExp

Instances

MonadReader (CompilerEnv op s) (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

ask :: CompilerM op s (CompilerEnv op s)

local :: (CompilerEnv op s -> CompilerEnv op s) -> CompilerM op s a -> CompilerM op s a

reader :: (CompilerEnv op s -> a) -> CompilerM op s a

data CompilerState s

Constructors

CompilerState

Fields

• compNameSrc :: VNameSource
• compInit :: [PyStmt]
• compUserState :: s

Instances

MonadState (CompilerState s) (CompilerM op s)

Defined in Futhark.CodeGen.Backends.GenericPython

Methods

get :: CompilerM op s (CompilerState s)

put :: CompilerState s -> CompilerM op s ()

state :: (CompilerState s -> (a, CompilerState s)) -> CompilerM op s a

stm :: PyStmt -> CompilerM op s ()

atInit :: PyStmt -> CompilerM op s ()

collect' :: CompilerM op s a -> CompilerM op s (a, [PyStmt])

collect :: CompilerM op s () -> CompilerM op s [PyStmt]

simpleCall :: String -> [PyExp] -> PyExp

A Call where the function is a variable and every argument is a simple Arg.