BufferResultsToOutParams.cpp File Reference

#include "mlir/Dialect/Bufferization/IR/AllocationOpInterface.h"
#include "mlir/Dialect/Bufferization/Transforms/Passes.h"
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/IR/Operation.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Dialect/Bufferization/Transforms/Passes.h.inc"

Go to the source code of this file.

Namespaces
	mlir
	Include the generated interface declarations.
	mlir::bufferization

Macros
#define	GEN_PASS_DEF_BUFFERRESULTSTOOUTPARAMSPASS

Typedefs
using	AllocationFn = bufferization::BufferResultsToOutParamsOpts::AllocationFn
using	MemCpyFn = bufferization::BufferResultsToOutParamsOpts::MemCpyFn

Functions
static bool	hasFullyDynamicLayoutMap (MemRefType type)
	Return true if the given MemRef type has a fully dynamic layout. More...
static bool	hasStaticIdentityLayout (MemRefType type)
	Return true if the given MemRef type has a static identity layout (i.e., no layout). More...
static LogicalResult	updateFuncOp (func::FuncOp func, SmallVectorImpl< BlockArgument > &appendedEntryArgs, bool addResultAttribute)
static LogicalResult	updateReturnOps (func::FuncOp func, ArrayRef< BlockArgument > appendedEntryArgs, const bufferization::BufferResultsToOutParamsOpts &options)
static LogicalResult	updateCalls (ModuleOp module, const bufferization::BufferResultsToOutParamsOpts &options)

Macro Definition Documentation

GEN_PASS_DEF_BUFFERRESULTSTOOUTPARAMSPASS

#define GEN_PASS_DEF_BUFFERRESULTSTOOUTPARAMSPASS

Definition at line 19 of file BufferResultsToOutParams.cpp.

Typedef Documentation

AllocationFn

using AllocationFn = bufferization::BufferResultsToOutParamsOpts::AllocationFn

Definition at line 25 of file BufferResultsToOutParams.cpp.

MemCpyFn

using MemCpyFn = bufferization::BufferResultsToOutParamsOpts::MemCpyFn

Definition at line 26 of file BufferResultsToOutParams.cpp.

Function Documentation

hasFullyDynamicLayoutMap()

static bool hasFullyDynamicLayoutMap ( MemRefType  type )

static

Return true if the given MemRef type has a fully dynamic layout.

Definition at line 29 of file BufferResultsToOutParams.cpp.

Referenced by updateCalls(), and updateFuncOp().

hasStaticIdentityLayout()

static bool hasStaticIdentityLayout ( MemRefType  type )

static

Return true if the given MemRef type has a static identity layout (i.e., no layout).

Definition at line 43 of file BufferResultsToOutParams.cpp.

Referenced by updateCalls(), and updateFuncOp().

updateCalls()

static LogicalResult updateCalls ( ModuleOp  module, const bufferization::BufferResultsToOutParamsOpts &  options  )

static

Definition at line 146 of file BufferResultsToOutParams.cpp.

References mlir::OpBuilder::create(), mlir::get(), mlir::Value::getType(), hasFullyDynamicLayoutMap(), hasStaticIdentityLayout(), mlir::SymbolTable::lookup(), options, and mlir::Operation::replaceAllUsesWith().

Referenced by mlir::bufferization::promoteBufferResultsToOutParams().

updateFuncOp()

static LogicalResult updateFuncOp ( func::FuncOp  func, SmallVectorImpl< BlockArgument > &  appendedEntryArgs, bool  addResultAttribute  )

static

Definition at line 53 of file BufferResultsToOutParams.cpp.

References mlir::detail::enumerate(), mlir::get(), hasFullyDynamicLayoutMap(), and hasStaticIdentityLayout().

Referenced by mlir::bufferization::promoteBufferResultsToOutParams().

updateReturnOps()

static LogicalResult updateReturnOps ( func::FuncOp  func, ArrayRef< BlockArgument >  appendedEntryArgs, const bufferization::BufferResultsToOutParamsOpts &  options  )

static

Definition at line 112 of file BufferResultsToOutParams.cpp.

References mlir::WalkResult::advance(), mlir::OpBuilder::create(), and options.

Referenced by mlir::bufferization::promoteBufferResultsToOutParams().