Page MenuHomePhabricator
Differential D78930 Diff 260625 mlir/include/mlir/Dialect/Affine/IR/AffineOps.h

Changeset View

Standalone View

View Options

mlir/include/mlir/Dialect/Affine/IR/AffineOps.h

Show First 20 Lines • Show All 310 Lines • ▼ Show 20 Linespublic:
static StringRef getTagMapAttrName() { return "tag_map"; } static StringRef getTagMapAttrName() { return "tag_map"; }
static ParseResult parse(OpAsmParser &parser, OperationState &result); static ParseResult parse(OpAsmParser &parser, OperationState &result);
void print(OpAsmPrinter &p); void print(OpAsmPrinter &p);
LogicalResult verify(); LogicalResult verify();
LogicalResult fold(ArrayRef<Attribute> cstOperands, LogicalResult fold(ArrayRef<Attribute> cstOperands,
SmallVectorImpl<OpFoldResult> &results); SmallVectorImpl<OpFoldResult> &results);
}; };
/// The "affine.load" op reads an element from a memref, where the index
/// for each memref dimension is an affine expression of loop induction
/// variables and symbols. The output of 'affine.load' is a new value with the
/// same type as the elements of the memref. An affine expression of loop IVs
/// and symbols must be specified for each dimension of the memref. The keyword
/// 'symbol' can be used to indicate SSA identifiers which are symbolic.
//
// Example 1:
//
// %1 = affine.load %0[%i0 + 3, %i1 + 7] : memref<100x100xf32>
//
// Example 2: Uses 'symbol' keyword for symbols '%n' and '%m'.
//
// %1 = affine.load %0[%i0 + symbol(%n), %i1 + symbol(%m)]
// : memref<100x100xf32>
//
class AffineLoadOp : public Op<AffineLoadOp, OpTrait::OneResult,
OpTrait::AtLeastNOperands<1>::Impl> {
public:
using Op::Op;
/// Builds an affine load op with the specified map and operands.
static void build(OpBuilder &builder, OperationState &result, AffineMap map,
ValueRange operands);
/// Builds an affine load op with an identity map and operands.
static void build(OpBuilder &builder, OperationState &result, Value memref,
ValueRange indices = {});
/// Builds an affine load op with the specified map and its operands.
static void build(OpBuilder &builder, OperationState &result, Value memref,
AffineMap map, ValueRange mapOperands);
/// Returns the operand index of the memref.
unsigned getMemRefOperandIndex() { return 0; }
/// Get memref operand.
Value getMemRef() { return getOperand(getMemRefOperandIndex()); }
void setMemRef(Value value) { setOperand(getMemRefOperandIndex(), value); }
MemRefType getMemRefType() {
return getMemRef().getType().cast<MemRefType>();
}
/// Get affine map operands.
operand_range getMapOperands() { return llvm::drop_begin(getOperands(), 1); }
/// Returns the affine map used to index the memref for this operation.
AffineMap getAffineMap() { return getAffineMapAttr().getValue(); }
AffineMapAttr getAffineMapAttr() {
return getAttr(getMapAttrName()).cast<AffineMapAttr>();
}
/// Returns the AffineMapAttr associated with 'memref'.
NamedAttribute getAffineMapAttrForMemRef(Value memref) {
assert(memref == getMemRef());
return {Identifier::get(getMapAttrName(), getContext()),
getAffineMapAttr()};
}
static StringRef getMapAttrName() { return "map"; }
static StringRef getOperationName() { return "affine.load"; }
// Hooks to customize behavior of this op.
static ParseResult parse(OpAsmParser &parser, OperationState &result);
void print(OpAsmPrinter &p);
LogicalResult verify();
static void getCanonicalizationPatterns(OwningRewritePatternList &results,
MLIRContext *context);
OpFoldResult fold(ArrayRef<Attribute> operands);
};
/// The "affine.store" op writes an element to a memref, where the index
/// for each memref dimension is an affine expression of loop induction
/// variables and symbols. The 'affine.store' op stores a new value which is the
/// same type as the elements of the memref. An affine expression of loop IVs
/// and symbols must be specified for each dimension of the memref. The keyword
/// 'symbol' can be used to indicate SSA identifiers which are symbolic.
//
// Example 1:
//
// affine.store %v0, %0[%i0 + 3, %i1 + 7] : memref<100x100xf32>
//
// Example 2: Uses 'symbol' keyword for symbols '%n' and '%m'.
//
// affine.store %v0, %0[%i0 + symbol(%n), %i1 + symbol(%m)]
// : memref<100x100xf32>
//
class AffineStoreOp : public Op<AffineStoreOp, OpTrait::ZeroResult,
OpTrait::AtLeastNOperands<1>::Impl> {
public:
using Op::Op;
/// Builds an affine store operation with the provided indices (identity map).
static void build(OpBuilder &builder, OperationState &result,
Value valueToStore, Value memref, ValueRange indices);
/// Builds an affine store operation with the specified map and its operands.
static void build(OpBuilder &builder, OperationState &result,
Value valueToStore, Value memref, AffineMap map,
ValueRange mapOperands);
/// Get value to be stored by store operation.
Value getValueToStore() { return getOperand(0); }
/// Returns the operand index of the memref.
unsigned getMemRefOperandIndex() { return 1; }
/// Get memref operand.
Value getMemRef() { return getOperand(getMemRefOperandIndex()); }
void setMemRef(Value value) { setOperand(getMemRefOperandIndex(), value); }
MemRefType getMemRefType() {
return getMemRef().getType().cast<MemRefType>();
}
/// Get affine map operands.
operand_range getMapOperands() { return llvm::drop_begin(getOperands(), 2); }
/// Returns the affine map used to index the memref for this operation.
AffineMap getAffineMap() { return getAffineMapAttr().getValue(); }
AffineMapAttr getAffineMapAttr() {
return getAttr(getMapAttrName()).cast<AffineMapAttr>();
}
/// Returns the AffineMapAttr associated with 'memref'.
NamedAttribute getAffineMapAttrForMemRef(Value memref) {
assert(memref == getMemRef());
return {Identifier::get(getMapAttrName(), getContext()),
getAffineMapAttr()};
}
static StringRef getMapAttrName() { return "map"; }
static StringRef getOperationName() { return "affine.store"; }
// Hooks to customize behavior of this op.
static ParseResult parse(OpAsmParser &parser, OperationState &result);
void print(OpAsmPrinter &p);
LogicalResult verify();
static void getCanonicalizationPatterns(OwningRewritePatternList &results,
MLIRContext *context);
LogicalResult fold(ArrayRef<Attribute> cstOperands,
SmallVectorImpl<OpFoldResult> &results);
};
/// Returns true if the given Value can be used as a dimension id. /// Returns true if the given Value can be used as a dimension id.
bool isValidDim(Value value); bool isValidDim(Value value);
/// Returns true if the given Value can be used as a symbol. /// Returns true if the given Value can be used as a symbol.
bool isValidSymbol(Value value); bool isValidSymbol(Value value);
/// Modifies both `map` and `operands` in-place so as to: /// Modifies both `map` and `operands` in-place so as to:
/// 1. drop duplicate operands /// 1. drop duplicate operands
▲ Show 20 Lines • Show All 157 Lines • Show Last 20 Lines