ConvertToDestinationStyle.cpp File Reference

#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
#include "mlir/Dialect/Bufferization/IR/Bufferization.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/Utils/StaticValueUtils.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/Support/Debug.h"

Go to the source code of this file.

Functions
static Value	createInserts (RewriterBase &rewriter, Location loc, int dim, Value destination, ArrayRef< int64_t > shape, ArrayRef< Value > constants, OperandRange::iterator &elementIt, SmallVectorImpl< Value > &indices)
static void	createMemcpy (OpBuilder &b, Location loc, Value tensorSource, Value memrefDest, const linalg::BufferizeToAllocationOptions &options)
	Create a memcpy from the given source tensor to the given destination memref. More...
static Operation *	movePaddingToFillOrGenericOp (RewriterBase &rewriter, Location loc, PadOp padOp, Value dest)
static SmallVector< Value >	reifyOrComputeDynamicSizes (OpBuilder &b, Value value)
static Value	createAllocationForTensor (RewriterBase &rewriter, Location loc, Value value, const linalg::BufferizeToAllocationOptions &options, Attribute memorySpace={})

Function Documentation

createAllocationForTensor()

static Value createAllocationForTensor ( RewriterBase &  rewriter, Location  loc, Value  value, const linalg::BufferizeToAllocationOptions &  options, Attribute  memorySpace = {}  )

static

Definition at line 192 of file ConvertToDestinationStyle.cpp.

Referenced by mlir::linalg::bufferizeToAllocation().

createInserts()

static Value createInserts ( RewriterBase &  rewriter, Location  loc, int  dim, Value  destination, ArrayRef< int64_t >  shape, ArrayRef< Value >  constants, OperandRange::iterator &  elementIt, SmallVectorImpl< Value > &  indices  )

static

Definition at line 34 of file ConvertToDestinationStyle.cpp.

References mlir::OpBuilder::create().

Referenced by mlir::linalg::rewriteInDestinationPassingStyle().

createMemcpy()

static void createMemcpy ( OpBuilder &  b, Location  loc, Value  tensorSource, Value  memrefDest, const linalg::BufferizeToAllocationOptions &  options  )

static

Create a memcpy from the given source tensor to the given destination memref.

The copy op type can be specified in the options.

Definition at line 58 of file ConvertToDestinationStyle.cpp.

References mlir::OpBuilder::create(), mlir::bufferization::getMemRefTypeWithFullyDynamicLayout(), mlir::Value::getType(), mlir::Type::isa(), mlir::linalg::BufferizeToAllocationOptions::LinalgCopy, mlir::linalg::BufferizeToAllocationOptions::MaterializeInDestination, mlir::linalg::BufferizeToAllocationOptions::MemrefCopy, and options.

movePaddingToFillOrGenericOp()

static Operation* movePaddingToFillOrGenericOp ( RewriterBase &  rewriter, Location  loc, PadOp  padOp, Value  dest  )

static

Definition at line 95 of file ConvertToDestinationStyle.cpp.

References mlir::OpBuilder::create(), mlir::OpBuilder::createBlock(), mlir::Builder::getContext(), mlir::Value::getDefiningOp(), mlir::MLIRContext::getLoadedDialect(), mlir::Value::getLoc(), mlir::Builder::getMultiDimIdentityMap(), mlir::Operation::getParentOp(), mlir::Value::getType(), mlir::m_Constant(), mlir::matchPattern(), mlir::Dialect::materializeConstant(), mlir::RewriterBase::mergeBlocks(), mlir::RewriterBase::replaceOpWithNewOp(), and mlir::OpBuilder::setInsertionPointToStart().

Referenced by mlir::linalg::rewriteInDestinationPassingStyle().

reifyOrComputeDynamicSizes()

static SmallVector<Value> reifyOrComputeDynamicSizes ( OpBuilder &  b, Value  value  )

static

Definition at line 160 of file ConvertToDestinationStyle.cpp.

References mlir::OpBuilder::create(), mlir::Value::getDefiningOp(), mlir::Value::getLoc(), mlir::Value::getType(), mlir::reifyResultShapes(), and mlir::succeeded().