20 #include "llvm/ADT/STLExtras.h"
21 #include "llvm/ADT/SetOperations.h"
22 #include "llvm/ADT/StringExtras.h"
23 #include "llvm/ADT/TypeSwitch.h"
24 #include "llvm/Support/Casting.h"
33 #include "mlir/Dialect/IRDL/IR/IRDL.cpp.inc"
35 #include "mlir/Dialect/IRDL/IR/IRDLDialect.cpp.inc"
37 void IRDLDialect::initialize() {
40 #include "mlir/Dialect/IRDL/IR/IRDLOps.cpp.inc"
43 #define GET_TYPEDEF_LIST
44 #include "mlir/Dialect/IRDL/IR/IRDLTypesGen.cpp.inc"
47 #define GET_ATTRDEF_LIST
48 #include "mlir/Dialect/IRDL/IR/IRDLAttributes.cpp.inc"
60 if (regionParseRes.has_value() &&
failed(regionParseRes.value()))
78 return loc->
emitError(
"name of ") << label <<
" is empty";
80 bool allowUnderscore =
false;
81 for (
auto &elem : in) {
85 << label <<
" should not contain leading or double underscores";
90 <<
" must contain only lowercase letters, digits and "
93 if (llvm::isUpper(elem))
95 << label <<
" should not contain uppercase letters";
98 allowUnderscore = elem !=
'_';
106 return emitOpError(
"invalid dialect name");
114 return isValidName(getSymName(), getOperation(),
"operation");
118 auto symName = getSymName();
119 if (symName.front() ==
'!')
120 symName = symName.substr(1);
121 return isValidName(symName, getOperation(),
"type");
125 auto symName = getSymName();
126 if (symName.front() ==
'#')
127 symName = symName.substr(1);
128 return isValidName(symName, getOperation(),
"attribute");
131 LogicalResult OperationOp::verifyRegions() {
136 auto insertNames = [&](StringRef
kind, ArrayAttr names) {
137 llvm::SmallDenseSet<StringRef> nameSet;
138 nameSet.reserve(names.size());
139 for (
auto name : names)
140 nameSet.insert(llvm::cast<StringAttr>(name).getValue());
141 valueNames.emplace_back(
kind, std::move(nameSet));
144 for (
Operation &op : getBody().getOps()) {
147 [&](OperandsOp op) { insertNames(
"operands", op.getNames()); })
149 [&](ResultsOp op) { insertNames(
"results", op.getNames()); })
151 [&](RegionsOp op) { insertNames(
"regions", op.getNames()); });
157 for (
size_t i : llvm::seq(valueNames.size())) {
158 for (
size_t j : llvm::seq(i + 1, valueNames.size())) {
159 auto [lhs, lhsSet] = valueNames[i];
160 auto &[rhs, rhsSet] = valueNames[
j];
161 llvm::set_intersect(lhsSet, rhsSet);
163 return emitOpError(
"contains a value named '")
164 << *lhsSet.begin() <<
"' for both its " << lhs <<
" and " << rhs;
172 ArrayAttr names,
size_t numOperands) {
173 if (numOperands != names.size())
175 <<
"the number of " << kindName
176 <<
"s and their names must be "
178 << numOperands <<
" and " << names.size() <<
" respectively";
182 StringRef nameRef = llvm::cast<StringAttr>(name).getValue();
187 if (nameMap.contains(nameRef))
188 return op->
emitOpError() <<
"name of " << kindName <<
" #" << i
189 <<
" is a duplicate of the name of " << kindName
190 <<
" #" << nameMap[nameRef];
191 nameMap.insert({nameRef, i});
198 return verifyNames(*
this,
"parameter", getNames(), getNumOperands());
201 template <
typename ValueListOp>
203 StringRef kindName) {
204 size_t numVariadicities = op.getVariadicity().size();
205 size_t numOperands = op.getNumOperands();
207 if (numOperands != numVariadicities)
208 return op.emitOpError()
209 <<
"the number of " << kindName
210 <<
"s and their variadicities must be "
212 << numOperands <<
" and " << numVariadicities <<
" respectively";
214 return verifyNames(op, kindName, op.getNames(), numOperands);
226 size_t namesSize = getAttributeValueNames().size();
227 size_t valuesSize = getAttributeValues().size();
229 if (namesSize != valuesSize)
231 <<
"the number of attribute names and their constraints must be "
233 << namesSize <<
" and " << valuesSize <<
" respectively";
239 std::optional<StringRef> baseName = getBaseName();
240 std::optional<SymbolRefAttr> baseRef = getBaseRef();
241 if (baseName.has_value() == baseRef.has_value())
242 return emitOpError() <<
"the base type or attribute should be specified by "
243 "either a name or a reference";
246 (baseName->empty() || ((*baseName)[0] !=
'!' && (*baseName)[0] !=
'#')))
247 return emitOpError() <<
"the base type or attribute name should start with "
257 Operation *source, SymbolRefAttr symbol) {
262 return source->
emitOpError() <<
"symbol '" << symbol <<
"' not found";
264 if (!isa<TypeOp, AttributeOp>(targetOp))
265 return source->
emitOpError() <<
"symbol '" << symbol
266 <<
"' does not refer to a type or attribute "
267 "definition (refers to '"
268 << targetOp->
getName() <<
"')";
274 std::optional<SymbolRefAttr> baseRef = getBaseRef();
283 std::optional<SymbolRefAttr> baseRef = getBaseType();
297 VariadicityAttr &variadicityAttr) {
319 ArrayAttr &valueNamesAttr, VariadicityArrayAttr *variadicityAttr) {
326 auto parseOne = [&] {
329 VariadicityAttr variadicity;
333 if (variadicityAttr) {
336 variadicities.push_back(variadicity);
343 operands.push_back(operand);
363 ArrayAttr &valueNamesAttr) {
376 ArrayAttr &valueNamesAttr, VariadicityArrayAttr &variadicityAttr) {
382 ArrayAttr valueNamesAttr,
383 VariadicityArrayAttr variadicityAttr) {
385 interleaveComma(llvm::seq<int>(0, operands.size()), p, [&](
int i) {
386 p << llvm::cast<StringAttr>(valueNamesAttr[i]).getValue() <<
": ";
387 if (variadicityAttr) {
388 Variadicity variadicity = variadicityAttr[i].getValue();
389 if (variadicity != Variadicity::single) {
390 p << stringifyVariadicity(variadicity) <<
" ";
405 ArrayAttr valueNamesAttr) {
418 ArrayAttr valueNamesAttr, VariadicityArrayAttr variadicityAttr) {
425 ArrayAttr &attrNamesAttr) {
429 auto parseOperands = [&]() {
444 if (attrNames.empty())
447 interleaveComma(llvm::seq<int>(0, attrNames.size()), p,
448 [&](
int i) { p << attrNames[i] <<
" = " << attrArgs[i]; });
453 if (IntegerAttr numberOfBlocks = getNumberOfBlocksAttr())
454 if (int64_t number = numberOfBlocks.getInt(); number <= 0) {
455 return emitOpError(
"the number of blocks is expected to be >= 1 but got ")
462 return verifyNames(*
this,
"region", getNames(), getNumOperands());
465 #include "mlir/Dialect/IRDL/IR/IRDLInterfaces.cpp.inc"
467 #define GET_TYPEDEF_CLASSES
468 #include "mlir/Dialect/IRDL/IR/IRDLTypesGen.cpp.inc"
470 #include "mlir/Dialect/IRDL/IR/IRDLEnums.cpp.inc"
472 #define GET_ATTRDEF_CLASSES
473 #include "mlir/Dialect/IRDL/IR/IRDLAttributes.cpp.inc"
475 #define GET_OP_CLASSES
476 #include "mlir/Dialect/IRDL/IR/IRDLOps.cpp.inc"
static LogicalResult verifyNames(Operation *op, StringRef kindName, ArrayAttr names, size_t numOperands)
static ParseResult parseValueWithVariadicity(OpAsmParser &p, OpAsmParser::UnresolvedOperand &operand, VariadicityAttr &variadicityAttr)
Parse a value with its variadicity first.
static void printAttributesOp(OpAsmPrinter &p, AttributesOp op, OperandRange attrArgs, ArrayAttr attrNames)
static llvm::LogicalResult isValidName(llvm::StringRef in, mlir::Operation *loc, const Twine &label)
static LogicalResult checkSymbolIsTypeOrAttribute(SymbolTableCollection &symbolTable, Operation *source, SymbolRefAttr symbol)
Finds whether the provided symbol is an IRDL type or attribute definition.
static LogicalResult verifyOperandsResultsCommon(ValueListOp op, StringRef kindName)
static ParseResult parseSingleBlockRegion(OpAsmParser &p, Region ®ion)
Parse a region, and add a single block if the region is empty.
static ParseResult parseAttributesOp(OpAsmParser &p, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &attrOperands, ArrayAttr &attrNamesAttr)
static void printNamedValueList(OpAsmPrinter &p, Operation *op, OperandRange operands, ArrayAttr valueNamesAttr)
Print a list of named values.
static void printNamedValueListWithVariadicity(OpAsmPrinter &p, Operation *op, OperandRange operands, ArrayAttr valueNamesAttr, VariadicityArrayAttr variadicityAttr)
Print a list of named values with their variadicities first.
static ParseResult parseNamedValueList(OpAsmParser &p, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &operands, ArrayAttr &valueNamesAttr)
Parse a list of named values.
static ParseResult parseNamedValueListImpl(OpAsmParser &p, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &operands, ArrayAttr &valueNamesAttr, VariadicityArrayAttr *variadicityAttr)
static void printNamedValueListImpl(OpAsmPrinter &p, Operation *op, OperandRange operands, ArrayAttr valueNamesAttr, VariadicityArrayAttr variadicityAttr)
static ParseResult parseNamedValueListWithVariadicity(OpAsmParser &p, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &operands, ArrayAttr &valueNamesAttr, VariadicityArrayAttr &variadicityAttr)
Parse a list of named values with their variadicities first.
static void printSingleBlockRegion(OpAsmPrinter &p, Operation *op, Region ®ion)
union mlir::linalg::@1244::ArityGroupAndKind::Kind kind
@ Paren
Parens 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 ParseResult parseOptionalKeyword(StringRef keyword)=0
Parse the given keyword if present.
virtual ParseResult parseRBrace()=0
Parse a } token.
virtual ParseResult parseEqual()=0
Parse a = token.
virtual ParseResult parseColon()=0
Parse a : token.
ParseResult parseKeyword(StringRef keyword)
Parse a given keyword.
virtual ParseResult parseAttribute(Attribute &result, Type type={})=0
Parse an arbitrary attribute of a given type and return it in result.
virtual ParseResult parseOptionalLBrace()=0
Parse a { token if present.
Block represents an ordered list of Operations.
This class is a general helper class for creating context-global objects like types,...
MLIRContext * getContext() const
ArrayAttr getArrayAttr(ArrayRef< Attribute > value)
static bool isValidNamespace(StringRef str)
Utility function that returns if the given string is a valid dialect namespace.
MLIRContext is the top-level object for a collection of MLIR operations.
The OpAsmParser has methods for interacting with the asm parser: parsing things from it,...
virtual OptionalParseResult parseOptionalRegion(Region ®ion, ArrayRef< Argument > arguments={}, bool enableNameShadowing=false)=0
Parses a region if present.
virtual ParseResult parseOperand(UnresolvedOperand &result, bool allowResultNumber=true)=0
Parse a single SSA value operand name along with a result number if allowResultNumber is true.
This is a pure-virtual base class that exposes the asmprinter hooks necessary to implement a custom p...
virtual void printRegion(Region &blocks, bool printEntryBlockArgs=true, bool printBlockTerminators=true, bool printEmptyBlock=false)=0
Prints a region.
This class implements the operand iterators for the Operation class.
Operation is the basic unit of execution within MLIR.
InFlightDiagnostic emitError(const Twine &message={})
Emit an error about fatal conditions with this operation, reporting up to any diagnostic handlers tha...
OperationName getName()
The name of an operation is the key identifier for it.
InFlightDiagnostic emitOpError(const Twine &message={})
Emit an error with the op name prefixed, like "'dim' op " which is convenient for verifiers.
This class contains a list of basic blocks and a link to the parent operation it is attached to.
void push_back(Block *block)
BlockListType & getBlocks()
This class represents a collection of SymbolTables.
constexpr void enumerate(std::tuple< Tys... > &tuple, CallbackT &&callback)
Operation * lookupSymbolNearDialect(SymbolTableCollection &symbolTable, Operation *source, SymbolRefAttr symbol)
Looks up a symbol from the symbol table containing the source operation's dialect definition operatio...
Include the generated interface declarations.
auto get(MLIRContext *context, Ts &&...params)
Helper method that injects context only if needed, this helps unify some of the attribute constructio...
LogicalResult verify(Operation *op, bool verifyRecursively=true)
Perform (potentially expensive) checks of invariants, used to detect compiler bugs,...
This is the representation of an operand reference.
Eliminates variable at the specified position using Fourier-Motzkin variable elimination.