Futhark.IR.Prop

This module provides various simple ways to query and manipulate fundamental Futhark terms, such as types and values. The intent is to keep Futhark.IR.Syntax simple, and put whatever embellishments we need here. This is an internal, desugared representation.

Documentation

module Futhark.IR.Prop.Reshape

module Futhark.IR.Prop.Rearrange

module Futhark.IR.Prop.Types

module Futhark.IR.Prop.Constants

module Futhark.IR.Prop.TypeOf

module Futhark.IR.Prop.Patterns

module Futhark.IR.Prop.Names

module Futhark.IR.RetType

module Futhark.IR.Rephrase

Built-in functions

isBuiltInFunction :: Name -> Bool

isBuiltInFunction k is True if k is an element of builtInFunctions.

builtInFunctions :: Map Name (PrimType, [PrimType])

A map of all built-in functions and their types.

Extra tools

asBasicOp :: Exp rep -> Maybe BasicOp

If the expression is a BasicOp, return it, otherwise Nothing.

safeExp :: IsOp (Op rep) => Exp rep -> Bool

An expression is safe if it is always well-defined (assuming that any required certificates have been checked) in any context. For example, array indexing is not safe, as the index may be out of bounds. On the other hand, adding two numbers cannot fail.

subExpVars :: [SubExp] -> [VName]

Return the variable names used in Var subexpressions. May contain duplicates.

subExpVar :: SubExp -> Maybe VName

If the SubExp is a Var return the variable name.

commutativeLambda :: Lambda rep -> Bool

Does the given lambda represent a known commutative function? Based on pattern matching and checking whether the lambda represents a known arithmetic operator; don't expect anything clever here.

defAux :: dec -> StmAux dec

A StmAux with empty Certs.

stmCerts :: Stm rep -> Certs

The certificates associated with a statement.

certify :: Certs -> Stm rep -> Stm rep

Add certificates to a statement.

expExtTypesFromPat :: Typed dec => Pat dec -> [ExtType]

Construct the type of an expression that would match the pattern.

attrsForAssert :: Attrs -> Attrs

Keep only those attributes that are relevant for Assert expressions.

lamIsBinOp :: ASTRep rep => Lambda rep -> Maybe [(BinOp, PrimType, VName, VName)]

Horizontally fission a lambda that models a binary operator.

type ASTConstraints a = (Eq a, Ord a, Show a, Rename a, Substitute a, FreeIn a, Pretty a)

A handy shorthand for properties that we usually want for things we stuff into ASTs.

class (ASTConstraints op, TypedOp op) => IsOp op where

A type class for operations.

Methods

safeOp :: op -> Bool

Like safeExp, but for arbitrary ops.

cheapOp :: op -> Bool

Should we try to hoist this out of branches?

Instances

IsOp SizeOp

Defined in Futhark.IR.GPU.Op

Methods

safeOp :: SizeOp -> Bool

cheapOp :: SizeOp -> Bool

ASTRep rep => IsOp (SOAC rep)

Defined in Futhark.IR.SOACS.SOAC

Methods

safeOp :: SOAC rep -> Bool

cheapOp :: SOAC rep -> Bool

(ASTRep rep, IsOp (op rep)) => IsOp (HostOp op rep)

Defined in Futhark.IR.GPU.Op

Methods

safeOp :: HostOp op rep -> Bool

cheapOp :: HostOp op rep -> Bool

(ASTRep rep, IsOp (op rep)) => IsOp (MCOp op rep)

Defined in Futhark.IR.MC.Op

Methods

safeOp :: MCOp op rep -> Bool

cheapOp :: MCOp op rep -> Bool

IsOp (inner rep) => IsOp (MemOp inner rep)

Defined in Futhark.IR.Mem

Methods

safeOp :: MemOp inner rep -> Bool

cheapOp :: MemOp inner rep -> Bool

IsOp (NoOp rep)

Defined in Futhark.IR.Prop

Methods

safeOp :: NoOp rep -> Bool

cheapOp :: NoOp rep -> Bool

(ASTRep rep, ASTConstraints lvl) => IsOp (SegOp lvl rep)

Defined in Futhark.IR.SegOp

Methods

safeOp :: SegOp lvl rep -> Bool

cheapOp :: SegOp lvl rep -> Bool

class (RepTypes rep, PrettyRep rep, Renameable rep, Substitutable rep, FreeDec (ExpDec rep), FreeIn (LetDec rep), FreeDec (BodyDec rep), FreeIn (FParamInfo rep), FreeIn (LParamInfo rep), FreeIn (RetType rep), FreeIn (BranchType rep), IsOp (Op rep), RephraseOp (OpC rep)) => ASTRep rep where

Representation-specific attributes; also means the rep supports some basic facilities.

Methods

expTypesFromPat :: (HasScope rep m, Monad m) => Pat (LetDec rep) -> m [BranchType rep]

Given a pattern, construct the type of a body that would match it. An implementation for many representations would be expExtTypesFromPat.

Instances

ASTRep GPU

Defined in Futhark.IR.GPU

Methods

expTypesFromPat :: (HasScope GPU m, Monad m) => Pat (LetDec GPU) -> m [BranchType GPU]

ASTRep GPUMem

Defined in Futhark.IR.GPUMem

Methods

expTypesFromPat :: (HasScope GPUMem m, Monad m) => Pat (LetDec GPUMem) -> m [BranchType GPUMem]

ASTRep MC

Defined in Futhark.IR.MC

Methods

expTypesFromPat :: (HasScope MC m, Monad m) => Pat (LetDec MC) -> m [BranchType MC]

ASTRep MCMem

Defined in Futhark.IR.MCMem

Methods

expTypesFromPat :: (HasScope MCMem m, Monad m) => Pat (LetDec MCMem) -> m [BranchType MCMem]

ASTRep SOACS

Defined in Futhark.IR.SOACS

Methods

expTypesFromPat :: (HasScope SOACS m, Monad m) => Pat (LetDec SOACS) -> m [BranchType SOACS]

ASTRep Seq

Defined in Futhark.IR.Seq

Methods

expTypesFromPat :: (HasScope Seq m, Monad m) => Pat (LetDec Seq) -> m [BranchType Seq]

ASTRep SeqMem

Defined in Futhark.IR.SeqMem

Methods

expTypesFromPat :: (HasScope SeqMem m, Monad m) => Pat (LetDec SeqMem) -> m [BranchType SeqMem]

(ASTRep rep, AliasedOp (OpC rep (Aliases rep)), IsOp (OpC rep (Aliases rep))) => ASTRep (Aliases rep)

Defined in Futhark.IR.Aliases

Methods

expTypesFromPat :: (HasScope (Aliases rep) m, Monad m) => Pat (LetDec (Aliases rep)) -> m [BranchType (Aliases rep)]

(Informing rep, IsOp (OpC rep (Wise rep))) => ASTRep (Wise rep)

Defined in Futhark.Optimise.Simplify.Rep

Methods

expTypesFromPat :: (HasScope (Wise rep) m, Monad m) => Pat (LetDec (Wise rep)) -> m [BranchType (Wise rep)]