MLIR  19.0.0git
ViewLikeInterface.cpp
Go to the documentation of this file.
1 //===- ViewLikeInterface.cpp - View-like operations in MLIR ---------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
10 
11 using namespace mlir;
12 
13 //===----------------------------------------------------------------------===//
14 // ViewLike Interfaces
15 //===----------------------------------------------------------------------===//
16 
17 /// Include the definitions of the loop-like interfaces.
18 #include "mlir/Interfaces/ViewLikeInterface.cpp.inc"
19 
21  StringRef name,
22  unsigned numElements,
23  ArrayRef<int64_t> staticVals,
24  ValueRange values) {
25  // Check static and dynamic offsets/sizes/strides does not overflow type.
26  if (staticVals.size() != numElements)
27  return op->emitError("expected ") << numElements << " " << name
28  << " values, got " << staticVals.size();
29  unsigned expectedNumDynamicEntries =
30  llvm::count_if(staticVals, [](int64_t staticVal) {
31  return ShapedType::isDynamic(staticVal);
32  });
33  if (values.size() != expectedNumDynamicEntries)
34  return op->emitError("expected ")
35  << expectedNumDynamicEntries << " dynamic " << name << " values";
36  return success();
37 }
38 
40 mlir::detail::verifyOffsetSizeAndStrideOp(OffsetSizeAndStrideOpInterface op) {
41  std::array<unsigned, 3> maxRanks = op.getArrayAttrMaxRanks();
42  // Offsets can come in 2 flavors:
43  // 1. Either single entry (when maxRanks == 1).
44  // 2. Or as an array whose rank must match that of the mixed sizes.
45  // So that the result type is well-formed.
46  if (!(op.getMixedOffsets().size() == 1 && maxRanks[0] == 1) && // NOLINT
47  op.getMixedOffsets().size() != op.getMixedSizes().size())
48  return op->emitError(
49  "expected mixed offsets rank to match mixed sizes rank (")
50  << op.getMixedOffsets().size() << " vs " << op.getMixedSizes().size()
51  << ") so the rank of the result type is well-formed.";
52  // Ranks of mixed sizes and strides must always match so the result type is
53  // well-formed.
54  if (op.getMixedSizes().size() != op.getMixedStrides().size())
55  return op->emitError(
56  "expected mixed sizes rank to match mixed strides rank (")
57  << op.getMixedSizes().size() << " vs " << op.getMixedStrides().size()
58  << ") so the rank of the result type is well-formed.";
59 
61  op, "offset", maxRanks[0], op.getStaticOffsets(), op.getOffsets())))
62  return failure();
64  op, "size", maxRanks[1], op.getStaticSizes(), op.getSizes())))
65  return failure();
67  op, "stride", maxRanks[2], op.getStaticStrides(), op.getStrides())))
68  return failure();
69 
70  for (int64_t offset : op.getStaticOffsets()) {
71  if (offset < 0 && !ShapedType::isDynamic(offset))
72  return op->emitError("expected offsets to be non-negative, but got ")
73  << offset;
74  }
75  for (int64_t size : op.getStaticSizes()) {
76  if (size < 0 && !ShapedType::isDynamic(size))
77  return op->emitError("expected sizes to be non-negative, but got ")
78  << size;
79  }
80  return success();
81 }
82 
83 static char getLeftDelimiter(AsmParser::Delimiter delimiter) {
84  switch (delimiter) {
86  return '(';
88  return '<';
90  return '[';
92  return '{';
93  default:
94  llvm_unreachable("unsupported delimiter");
95  }
96 }
97 
98 static char getRightDelimiter(AsmParser::Delimiter delimiter) {
99  switch (delimiter) {
101  return ')';
103  return '>';
105  return ']';
107  return '}';
108  default:
109  llvm_unreachable("unsupported delimiter");
110  }
111 }
112 
114  OperandRange values,
115  ArrayRef<int64_t> integers,
116  TypeRange valueTypes, ArrayRef<bool> scalables,
117  AsmParser::Delimiter delimiter) {
118  char leftDelimiter = getLeftDelimiter(delimiter);
119  char rightDelimiter = getRightDelimiter(delimiter);
120  printer << leftDelimiter;
121  if (integers.empty()) {
122  printer << rightDelimiter;
123  return;
124  }
125 
126  unsigned dynamicValIdx = 0;
127  unsigned scalableIndexIdx = 0;
128  llvm::interleaveComma(integers, printer, [&](int64_t integer) {
129  if (!scalables.empty() && scalables[scalableIndexIdx])
130  printer << "[";
131  if (ShapedType::isDynamic(integer)) {
132  printer << values[dynamicValIdx];
133  if (!valueTypes.empty())
134  printer << " : " << valueTypes[dynamicValIdx];
135  ++dynamicValIdx;
136  } else {
137  printer << integer;
138  }
139  if (!scalables.empty() && scalables[scalableIndexIdx])
140  printer << "]";
141 
142  scalableIndexIdx++;
143  });
144 
145  printer << rightDelimiter;
146 }
147 
149  OpAsmParser &parser,
151  DenseI64ArrayAttr &integers, DenseBoolArrayAttr &scalables,
152  SmallVectorImpl<Type> *valueTypes, AsmParser::Delimiter delimiter) {
153 
154  SmallVector<int64_t, 4> integerVals;
155  SmallVector<bool, 4> scalableVals;
156  auto parseIntegerOrValue = [&]() {
158  auto res = parser.parseOptionalOperand(operand);
159 
160  // When encountering `[`, assume that this is a scalable index.
161  scalableVals.push_back(parser.parseOptionalLSquare().succeeded());
162 
163  if (res.has_value() && succeeded(res.value())) {
164  values.push_back(operand);
165  integerVals.push_back(ShapedType::kDynamic);
166  if (valueTypes && parser.parseColonType(valueTypes->emplace_back()))
167  return failure();
168  } else {
169  int64_t integer;
170  if (failed(parser.parseInteger(integer)))
171  return failure();
172  integerVals.push_back(integer);
173  }
174 
175  // If this is assumed to be a scalable index, verify that there's a closing
176  // `]`.
177  if (scalableVals.back() && parser.parseOptionalRSquare().failed())
178  return failure();
179  return success();
180  };
181  if (parser.parseCommaSeparatedList(delimiter, parseIntegerOrValue,
182  " in dynamic index list"))
183  return parser.emitError(parser.getNameLoc())
184  << "expected SSA value or integer";
185  integers = parser.getBuilder().getDenseI64ArrayAttr(integerVals);
186  scalables = parser.getBuilder().getDenseBoolArrayAttr(scalableVals);
187  return success();
188 }
189 
191  OffsetSizeAndStrideOpInterface a, OffsetSizeAndStrideOpInterface b,
193  if (a.getStaticOffsets().size() != b.getStaticOffsets().size())
194  return false;
195  if (a.getStaticSizes().size() != b.getStaticSizes().size())
196  return false;
197  if (a.getStaticStrides().size() != b.getStaticStrides().size())
198  return false;
199  for (auto it : llvm::zip(a.getMixedOffsets(), b.getMixedOffsets()))
200  if (!cmp(std::get<0>(it), std::get<1>(it)))
201  return false;
202  for (auto it : llvm::zip(a.getMixedSizes(), b.getMixedSizes()))
203  if (!cmp(std::get<0>(it), std::get<1>(it)))
204  return false;
205  for (auto it : llvm::zip(a.getMixedStrides(), b.getMixedStrides()))
206  if (!cmp(std::get<0>(it), std::get<1>(it)))
207  return false;
208  return true;
209 }
210 
212  unsigned idx) {
213  return std::count_if(staticVals.begin(), staticVals.begin() + idx,
214  [&](int64_t val) { return ShapedType::isDynamic(val); });
215 }
static char getLeftDelimiter(AsmParser::Delimiter delimiter)
static char getRightDelimiter(AsmParser::Delimiter delimiter)
Delimiter
These are the supported delimiters around operand lists and region argument lists,...
@ Paren
Parens surrounding zero or more operands.
@ Braces
{} brackets surrounding zero or more operands.
@ Square
Square brackets surrounding zero or more operands.
@ LessGreater
<> brackets surrounding zero or more operands.
virtual ParseResult parseCommaSeparatedList(Delimiter delimiter, function_ref< ParseResult()> parseElementFn, StringRef contextMessage=StringRef())=0
Parse a list of comma-separated items with an optional delimiter.
virtual Builder & getBuilder() const =0
Return a builder which provides useful access to MLIRContext, global objects like types and attribute...
virtual InFlightDiagnostic emitError(SMLoc loc, const Twine &message={})=0
Emit a diagnostic at the specified location and return failure.
ParseResult parseInteger(IntT &result)
Parse an integer value from the stream.
virtual ParseResult parseColonType(Type &result)=0
Parse a colon followed by a type.
virtual SMLoc getNameLoc() const =0
Return the location of the original name token.
virtual ParseResult parseOptionalRSquare()=0
Parse a ] token if present.
virtual ParseResult parseOptionalLSquare()=0
Parse a [ token if present.
DenseI64ArrayAttr getDenseI64ArrayAttr(ArrayRef< int64_t > values)
Definition: Builders.cpp:183
DenseBoolArrayAttr getDenseBoolArrayAttr(ArrayRef< bool > values)
Tensor-typed DenseArrayAttr getters.
Definition: Builders.cpp:167
The OpAsmParser has methods for interacting with the asm parser: parsing things from it,...
virtual OptionalParseResult parseOptionalOperand(UnresolvedOperand &result, bool allowResultNumber=true)=0
Parse a single operand if present.
This is a pure-virtual base class that exposes the asmprinter hooks necessary to implement a custom p...
This class represents a single result from folding an operation.
Definition: OpDefinition.h:266
This class implements the operand iterators for the Operation class.
Definition: ValueRange.h:42
Operation is the basic unit of execution within MLIR.
Definition: Operation.h:88
InFlightDiagnostic emitError(const Twine &message={})
Emit an error about fatal conditions with this operation, reporting up to any diagnostic handlers tha...
Definition: Operation.cpp:268
This class represents success/failure for parsing-like operations that find it important to chain tog...
This class provides an abstraction over the various different ranges of value types.
Definition: TypeRange.h:36
This class provides an abstraction over the different types of ranges over Values.
Definition: ValueRange.h:381
Base class for DenseArrayAttr that is instantiated and specialized for each supported element type be...
bool sameOffsetsSizesAndStrides(OffsetSizeAndStrideOpInterface a, OffsetSizeAndStrideOpInterface b, llvm::function_ref< bool(OpFoldResult, OpFoldResult)> cmp)
unsigned getNumDynamicEntriesUpToIdx(ArrayRef< int64_t > staticVals, unsigned idx)
Helper method to compute the number of dynamic entries of staticVals, up to idx.
LogicalResult verifyOffsetSizeAndStrideOp(OffsetSizeAndStrideOpInterface op)
Include the generated interface declarations.
LogicalResult failure(bool isFailure=true)
Utility function to generate a LogicalResult.
Definition: LogicalResult.h:62
bool succeeded(LogicalResult result)
Utility function that returns true if the provided LogicalResult corresponds to a success value.
Definition: LogicalResult.h:68
LogicalResult success(bool isSuccess=true)
Utility function to generate a LogicalResult.
Definition: LogicalResult.h:56
void printDynamicIndexList(OpAsmPrinter &printer, Operation *op, OperandRange values, ArrayRef< int64_t > integers, TypeRange valueTypes=TypeRange(), ArrayRef< bool > scalables={}, AsmParser::Delimiter delimiter=AsmParser::Delimiter::Square)
Printer hook for custom directive in assemblyFormat.
LogicalResult verifyListOfOperandsOrIntegers(Operation *op, StringRef name, unsigned expectedNumElements, ArrayRef< int64_t > attr, ValueRange values)
Verify that a the values has as many elements as the number of entries in attr for which isDynamic ev...
ParseResult parseDynamicIndexList(OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &values, DenseI64ArrayAttr &integers, DenseBoolArrayAttr &scalableVals, SmallVectorImpl< Type > *valueTypes=nullptr, AsmParser::Delimiter delimiter=AsmParser::Delimiter::Square)
Parser hook for custom directive in assemblyFormat.
bool failed(LogicalResult result)
Utility function that returns true if the provided LogicalResult corresponds to a failure value.
Definition: LogicalResult.h:72
This class represents an efficient way to signal success or failure.
Definition: LogicalResult.h:26
bool succeeded() const
Returns true if the provided LogicalResult corresponds to a success value.
Definition: LogicalResult.h:41
bool failed() const
Returns true if the provided LogicalResult corresponds to a failure value.
Definition: LogicalResult.h:44
This is the representation of an operand reference.