Differential D78930 Diff 260625 mlir/include/mlir/Dialect/Affine/IR/AffineOps.td

Changeset View

Standalone View

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

Show First 20 Lines • Show All 354 Lines • ▼ Show 20 Lines	OpBuilder getElseBodyBuilder() {
return OpBuilder(&body, std::prev(body.end()));		return OpBuilder(&body, std::prev(body.end()));
}		}
}];		}];

let hasCanonicalizer = 1;		let hasCanonicalizer = 1;
let hasFolder = 1;		let hasFolder = 1;
}		}

		def AffineLoadOp : Affine_Op<"load", []> {
		let summary = "affine load operation";
		let description = [{
		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:

		```mlir
		%1 = affine.load %0[%i0 + 3, %i1 + 7] : memref<100x100xf32>
		```

		Example 2: Uses 'symbol' keyword for symbols '%n' and '%m'.

		```mlir
		%1 = affine.load %0[%i0 + symbol(%n), %i1 + symbol(%m)] : memref<100x100xf32>
		```
		}];

		let arguments = (ins Arg<AnyMemRef, "the reference to load from",
		[MemRead]>:$memref,
		Variadic<Index>:$indices);
		let results = (outs AnyType:$result);

		let builders = [
		/// Builds an affine load op with the specified map and operands.
		OpBuilder<"OpBuilder &builder, OperationState &result, AffineMap map, "
		"ValueRange operands">,
		/// Builds an affine load op with an identity map and operands.
		OpBuilder<"OpBuilder &builder, OperationState &result, Value memref, "
		"ValueRange indices = {}">,
		/// Builds an affine load op with the specified map and its operands.
		OpBuilder<"OpBuilder &builder, OperationState &result, Value memref, "
		"AffineMap map, ValueRange mapOperands">
		];

		let extraClassDeclaration = [{
		/// 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"; }
		}];

		let hasCanonicalizer = 1;
		let hasFolder = 1;
		}

class AffineMinMaxOpBase<string mnemonic, list<OpTrait> traits = []> :		class AffineMinMaxOpBase<string mnemonic, list<OpTrait> traits = []> :
Op<Affine_Dialect, mnemonic, traits> {		Op<Affine_Dialect, mnemonic, traits> {
let arguments = (ins AffineMapAttr:$map, Variadic<Index>:$operands);		let arguments = (ins AffineMapAttr:$map, Variadic<Index>:$operands);
let results = (outs Index);		let results = (outs Index);

let builders = [		let builders = [
OpBuilder<"OpBuilder &builder, OperationState &result, AffineMap affineMap, "		OpBuilder<"OpBuilder &builder, OperationState &result, AffineMap affineMap, "
"ValueRange mapOperands",		"ValueRange mapOperands",
▲ Show 20 Lines • Show All 199 Lines • ▼ Show 20 Lines	let extraClassDeclaration = [{
static StringRef getIsWriteAttrName() { return "isWrite"; }		static StringRef getIsWriteAttrName() { return "isWrite"; }
static StringRef getIsDataCacheAttrName() { return "isDataCache"; }		static StringRef getIsDataCacheAttrName() { return "isDataCache"; }
}];		}];

let hasCanonicalizer = 1;		let hasCanonicalizer = 1;
let hasFolder = 1;		let hasFolder = 1;
}		}

		def AffineStoreOp : Affine_Op<"store", []> {
		let summary = "affine store operation";
		let description = [{
		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:

		```mlir
		affine.store %v0, %0[%i0 + 3, %i1 + 7] : memref<100x100xf32>
		```

		Example 2: Uses 'symbol' keyword for symbols '%n' and '%m'.

		```mlir
		affine.store %v0, %0[%i0 + symbol(%n), %i1 + symbol(%m)] : memref<100x100xf32>
		```
		}];
		let arguments = (ins AnyType:$value,
		Arg<AnyMemRef, "the reference to store to",
		[MemWrite]>:$memref,
		Variadic<Index>:$indices);


		let skipDefaultBuilders = 1;
		let builders = [
		OpBuilder<"OpBuilder &builder, OperationState &result, "
		"Value valueToStore, Value memref, ValueRange indices">,
		OpBuilder<"OpBuilder &builder, OperationState &result, "
		"Value valueToStore, Value memref, AffineMap map, "
		"ValueRange mapOperands">
		];

		let extraClassDeclaration = [{
		/// 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"; }
		}];

		let hasCanonicalizer = 1;
		let hasFolder = 1;
		}

def AffineTerminatorOp :		def AffineTerminatorOp :
Affine_Op<"terminator", [NoSideEffect, Terminator]> {		Affine_Op<"terminator", [NoSideEffect, Terminator]> {
let summary = "affine terminator operation";		let summary = "affine terminator operation";
let description = [{		let description = [{
Syntax:		Syntax:

```		```
operation ::= `"affine.terminator"() : () -> ()`		operation ::= `"affine.terminator"() : () -> ()`
Show All 15 Lines	def AffineTerminatorOp :
// No custom parsing/printing form.		// No custom parsing/printing form.
let parser = ?;		let parser = ?;
let printer = ?;		let printer = ?;

// Fully specified by traits.		// Fully specified by traits.
let verifier = ?;		let verifier = ?;
}		}

#endif // AFFINE_OPS		#endif // AFFINE_OPS
		rriddleUnsubmitted Done Reply Inline Actions Can you keep this file sorted alphabetically? rriddle: Can you keep this file sorted alphabetically?