Attrs.cpp.inc

Go to the documentation of this file.

/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* AttrDef Definitions                                                        *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/
 
#ifdef GET_ATTRDEF_LIST
#undef GET_ATTRDEF_LIST
 
::llzk::PublicAttr,
::llzk::LoopBoundsAttr
 
#endif  // GET_ATTRDEF_LIST
 
#ifdef GET_ATTRDEF_CLASSES
#undef GET_ATTRDEF_CLASSES
 
static ::mlir::OptionalParseResult generatedAttributeParser(::mlir::AsmParser &parser, ::llvm::StringRef *mnemonic, ::mlir::Type type, ::mlir::Attribute &value) {
  return ::mlir::AsmParser::KeywordSwitch<::mlir::OptionalParseResult>(parser)
    .Case(::llzk::PublicAttr::getMnemonic(), [&](llvm::StringRef, llvm::SMLoc) {
      value = ::llzk::PublicAttr::get(parser.getContext());
      return ::mlir::success(!!value);
    })
    .Case(::llzk::LoopBoundsAttr::getMnemonic(), [&](llvm::StringRef, llvm::SMLoc) {
      value = ::llzk::LoopBoundsAttr::parse(parser, type);
      return ::mlir::success(!!value);
    })
    .Default([&](llvm::StringRef keyword, llvm::SMLoc) {
      *mnemonic = keyword;
      return std::nullopt;
    });
}
 
static ::mlir::LogicalResult generatedAttributePrinter(::mlir::Attribute def, ::mlir::AsmPrinter &printer) {
  return ::llvm::TypeSwitch<::mlir::Attribute, ::mlir::LogicalResult>(def)    .Case<::llzk::PublicAttr>([&](auto t) {
      printer << ::llzk::PublicAttr::getMnemonic();
      return ::mlir::success();
    })
    .Case<::llzk::LoopBoundsAttr>([&](auto t) {
      printer << ::llzk::LoopBoundsAttr::getMnemonic();
t.print(printer);
      return ::mlir::success();
    })
    .Default([](auto) { return ::mlir::failure(); });
}
 
namespace llzk {
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::PublicAttr)
namespace llzk {
namespace detail {
struct LoopBoundsAttrStorage : public ::mlir::AttributeStorage {
  using KeyTy = std::tuple<::llvm::APInt, ::llvm::APInt, ::llvm::APInt>;
  LoopBoundsAttrStorage(::llvm::APInt lower, ::llvm::APInt upper, ::llvm::APInt step) : lower(std::move(lower)), upper(std::move(upper)), step(std::move(step)) {}
 
  KeyTy getAsKey() const {
    return KeyTy(lower, upper, step);
  }
 
  bool operator==(const KeyTy &tblgenKey) const {
    return (::llvm::APInt::isSameValue(lower, std::get<0>(tblgenKey))) && (::llvm::APInt::isSameValue(upper, std::get<1>(tblgenKey))) && (::llvm::APInt::isSameValue(step, std::get<2>(tblgenKey)));
  }
 
  static ::llvm::hash_code hashKey(const KeyTy &tblgenKey) {
    return ::llvm::hash_combine(std::get<0>(tblgenKey), std::get<1>(tblgenKey), std::get<2>(tblgenKey));
  }
 
  static LoopBoundsAttrStorage *construct(::mlir::AttributeStorageAllocator &allocator, KeyTy &&tblgenKey) {
    auto lower = std::move(std::get<0>(tblgenKey));
    auto upper = std::move(std::get<1>(tblgenKey));
    auto step = std::move(std::get<2>(tblgenKey));
    return new (allocator.allocate<LoopBoundsAttrStorage>()) LoopBoundsAttrStorage(std::move(lower), std::move(upper), std::move(step));
  }
 
  ::llvm::APInt lower;
  ::llvm::APInt upper;
  ::llvm::APInt step;
};
} // namespace detail
LoopBoundsAttr LoopBoundsAttr::get(::mlir::MLIRContext *context, ::llvm::APInt lower, ::llvm::APInt upper, ::llvm::APInt step) {
  return Base::get(context, std::move(lower), std::move(upper), std::move(step));
}
 
::mlir::Attribute LoopBoundsAttr::parse(::mlir::AsmParser &odsParser, ::mlir::Type odsType) {
  ::mlir::Builder odsBuilder(odsParser.getContext());
  ::llvm::SMLoc odsLoc = odsParser.getCurrentLocation();
  (void) odsLoc;
  ::mlir::FailureOr<::llvm::APInt> _result_lower;
  ::mlir::FailureOr<::llvm::APInt> _result_upper;
  ::mlir::FailureOr<::llvm::APInt> _result_step;
  // Parse literal '<'
  if (odsParser.parseLess()) return {};
 
  // Parse variable 'lower'
  _result_lower = ::mlir::FieldParser<::llvm::APInt>::parse(odsParser);
  if (::mlir::failed(_result_lower)) {
    odsParser.emitError(odsParser.getCurrentLocation(), "failed to parse LLZK_LoopBoundsAttr parameter 'lower' which is to be a `::llvm::APInt`");
    return {};
  }
  // Parse literal 'to'
  if (odsParser.parseKeyword("to")) return {};
 
  // Parse variable 'upper'
  _result_upper = ::mlir::FieldParser<::llvm::APInt>::parse(odsParser);
  if (::mlir::failed(_result_upper)) {
    odsParser.emitError(odsParser.getCurrentLocation(), "failed to parse LLZK_LoopBoundsAttr parameter 'upper' which is to be a `::llvm::APInt`");
    return {};
  }
  // Parse literal 'step'
  if (odsParser.parseKeyword("step")) return {};
 
  // Parse variable 'step'
  _result_step = ::mlir::FieldParser<::llvm::APInt>::parse(odsParser);
  if (::mlir::failed(_result_step)) {
    odsParser.emitError(odsParser.getCurrentLocation(), "failed to parse LLZK_LoopBoundsAttr parameter 'step' which is to be a `::llvm::APInt`");
    return {};
  }
  // Parse literal '>'
  if (odsParser.parseGreater()) return {};
  assert(::mlir::succeeded(_result_lower));
  assert(::mlir::succeeded(_result_upper));
  assert(::mlir::succeeded(_result_step));
  return LoopBoundsAttr::get(odsParser.getContext(),
      ::llvm::APInt((*_result_lower)),
      ::llvm::APInt((*_result_upper)),
      ::llvm::APInt((*_result_step)));
}
 
void LoopBoundsAttr::print(::mlir::AsmPrinter &odsPrinter) const {
  ::mlir::Builder odsBuilder(getContext());
  odsPrinter << "<";
  odsPrinter.printStrippedAttrOrType(getLower());
  odsPrinter << ' ' << "to";
  odsPrinter << ' ';
  odsPrinter.printStrippedAttrOrType(getUpper());
  odsPrinter << ' ' << "step";
  odsPrinter << ' ';
  odsPrinter.printStrippedAttrOrType(getStep());
  odsPrinter << ">";
}
 
::llvm::APInt LoopBoundsAttr::getLower() const {
  return getImpl()->lower;
}
 
::llvm::APInt LoopBoundsAttr::getUpper() const {
  return getImpl()->upper;
}
 
::llvm::APInt LoopBoundsAttr::getStep() const {
  return getImpl()->step;
}
 
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::LoopBoundsAttr)
namespace llzk {

::mlir::Attribute LLZKDialect::parseAttribute(::mlir::DialectAsmParser &parser,
                                      ::mlir::Type type) const {
  ::llvm::SMLoc typeLoc = parser.getCurrentLocation();
  ::llvm::StringRef attrTag;
  {
    ::mlir::Attribute attr;
    auto parseResult = generatedAttributeParser(parser, &attrTag, type, attr);
    if (parseResult.has_value())
      return attr;
  }
  
  parser.emitError(typeLoc) << "unknown attribute `"
      << attrTag << "` in dialect `" << getNamespace() << "`";
  return {};
}
void LLZKDialect::printAttribute(::mlir::Attribute attr,
                         ::mlir::DialectAsmPrinter &printer) const {
  if (::mlir::succeeded(generatedAttributePrinter(attr, printer)))
    return;
  
}
} // namespace llzk
 
#endif  // GET_ATTRDEF_CLASSES