| Safe Haskell | None |
|---|---|
| Language | GHC2021 |
Futhark.Internalise.AccurateSizes
Synopsis
- argShapes :: [VName] -> [FParam SOACS] -> [Type] -> InternaliseM [SubExp]
- ensureResultShape :: ErrorMsg SubExp -> [Type] -> Result -> InternaliseM Result
- ensureResultExtShape :: ErrorMsg SubExp -> [ExtType] -> Result -> InternaliseM Result
- ensureExtShape :: ErrorMsg SubExp -> ExtType -> Name -> SubExp -> InternaliseM SubExp
- ensureShape :: ErrorMsg SubExp -> Type -> Name -> SubExp -> InternaliseM SubExp
- ensureArgShapes :: Typed (TypeBase Shape o) => ErrorMsg SubExp -> [VName] -> [TypeBase Shape o] -> [SubExp] -> InternaliseM [SubExp]
Documentation
ensureResultShape :: ErrorMsg SubExp -> [Type] -> Result -> InternaliseM Result Source #
ensureResultExtShape :: ErrorMsg SubExp -> [ExtType] -> Result -> InternaliseM Result Source #
ensureExtShape :: ErrorMsg SubExp -> ExtType -> Name -> SubExp -> InternaliseM SubExp Source #
ensureShape :: ErrorMsg SubExp -> Type -> Name -> SubExp -> InternaliseM SubExp Source #
ensureArgShapes :: Typed (TypeBase Shape o) => ErrorMsg SubExp -> [VName] -> [TypeBase Shape o] -> [SubExp] -> InternaliseM [SubExp] Source #
Reshape the arguments to a function so that they fit the expected shape declarations. Not used to change rank of arguments. Assumes everything is otherwise type-correct.