This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 654 Lines • ▼ Show 20 Lines	let extraClassDeclaration = [{
void setSteps(ValueRange steps) {		void setSteps(ValueRange steps) {
unsigned numLoops = getNumLoops();		unsigned numLoops = getNumLoops();
assert(steps.size() == numLoops &&		assert(steps.size() == numLoops &&
"expected upper bounds for every loop dimension");		"expected upper bounds for every loop dimension");
for (unsigned i = 0, pos = 2 * numLoops; i < numLoops; ++i, ++pos)		for (unsigned i = 0, pos = 2 * numLoops; i < numLoops; ++i, ++pos)
setOperand(pos, steps[i]);		setOperand(pos, steps[i]);
}		}

		/// Operand that corresponds to the `bbArg` block argument.
		gysitUnsubmitted Done Reply Inline Actions nit: Operand gysit: nit: Operand
		OpOperand& getTiedOperand(BlockArgument& bbArg) {
		return getOperation()->getOpOperand(getNumControlOperands() +
		bbArg.getArgNumber() - getNumLoops());
		}

/// Block argument that corresponds to the `input` or `output` operand.		/// Block argument that corresponds to the `input` or `output` operand.
BlockArgument getTiedBlockArgument(OpOperand& operand) {		BlockArgument getTiedBlockArgument(OpOperand& operand) {
auto operandIndex = operand.getOperandNumber();		auto operandIndex = operand.getOperandNumber();
assert(		assert(
operandIndex >= getNumControlOperands() &&		operandIndex >= getNumControlOperands() &&
operandIndex < getNumOperands() &&		operandIndex < getNumOperands() &&
"tied block arg is defined only for `input` and `output` arguments");		"tied block arg is defined only for `input` and `output` arguments");
return getBody()->getArgument(operandIndex - 2 * getNumLoops());		return getBody()->getArgument(operandIndex - 2 * getNumLoops());
▲ Show 20 Lines • Show All 158 Lines • Show Last 20 Lines

This is an archive of the discontinued LLVM Phabricator instance.

[mlir] Add a helper for TiledLoopOp to get an operand tied to the bbArg.ClosedPublic

Details

Diff Detail

Event Timeline

Revision Contents

Diff 390941

mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td

[mlir] Add a helper for TiledLoopOp to get an operand tied to the bbArg.
ClosedPublic