Types.cpp.inc

Go to the documentation of this file.

/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* TypeDef Definitions                                                        *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/
 
#ifdef GET_TYPEDEF_LIST
#undef GET_TYPEDEF_LIST
 
::llzk::array::ArrayType
 
#endif // GET_TYPEDEF_LIST
 
#ifdef GET_TYPEDEF_CLASSES
#undef GET_TYPEDEF_CLASSES
 
static ::mlir::OptionalParseResult generatedTypeParser(::mlir::AsmParser &parser, ::llvm::StringRef *mnemonic, ::mlir::Type &value) {
  return ::mlir::AsmParser::KeywordSwitch<::mlir::OptionalParseResult>(parser)
    .Case(::llzk::array::ArrayType::getMnemonic(), [&](llvm::StringRef, llvm::SMLoc) {
      value = ::llzk::array::ArrayType::parse(parser);
      return ::mlir::success(!!value);
    })
    .Default([&](llvm::StringRef keyword, llvm::SMLoc) {
      *mnemonic = keyword;
      return std::nullopt;
    });
}
 
static ::llvm::LogicalResult generatedTypePrinter(::mlir::Type def, ::mlir::AsmPrinter &printer) {
  return ::llvm::TypeSwitch<::mlir::Type, ::llvm::LogicalResult>(def)    .Case<::llzk::array::ArrayType>([&](auto t) {
      printer << ::llzk::array::ArrayType::getMnemonic();
t.print(printer);
      return ::mlir::success();
    })
    .Default([](auto) { return ::mlir::failure(); });
}
 
namespace llzk::array {
 
namespace detail {
 
struct ArrayTypeStorage : public ::mlir::TypeStorage {
  using KeyTy = std::tuple<::mlir::Type, ::llvm::ArrayRef<::mlir::Attribute>, ::llvm::ArrayRef<int64_t>>;
  ArrayTypeStorage(::mlir::Type elementType, ::llvm::ArrayRef<::mlir::Attribute> dimensionSizes, ::llvm::ArrayRef<int64_t> shape) : elementType(std::move(elementType)), dimensionSizes(std::move(dimensionSizes)), shape(std::move(shape)) {}
 
  KeyTy getAsKey() const {
    return KeyTy(elementType, dimensionSizes, shape);
  }
 
  bool operator==(const KeyTy &tblgenKey) const {
    return (elementType == std::get<0>(tblgenKey)) && (dimensionSizes == std::get<1>(tblgenKey)) && (shape == 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 ArrayTypeStorage *construct(::mlir::TypeStorageAllocator &allocator, KeyTy &&tblgenKey) {
    auto elementType = std::move(std::get<0>(tblgenKey));
    auto dimensionSizes = std::move(std::get<1>(tblgenKey));
    auto shape = std::move(std::get<2>(tblgenKey));
    dimensionSizes = allocator.copyInto(dimensionSizes);
    shape = allocator.copyInto(shape);
    return new (allocator.allocate<ArrayTypeStorage>()) ArrayTypeStorage(std::move(elementType), std::move(dimensionSizes), std::move(shape));
  }
 
  ::mlir::Type elementType;
  ::llvm::ArrayRef<::mlir::Attribute> dimensionSizes;
  ::llvm::ArrayRef<int64_t> shape;
};
 
} // namespace detail
namespace {
  const ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError = nullptr;
}
ArrayType ArrayType::get(::mlir::Type elementType, ::llvm::ArrayRef<::mlir::Attribute> dimensionSizes) {
  assert(elementType && "element type cannot be null");
  ::mlir::MLIRContext *ctx = elementType.getContext();
  OwningEmitErrorFn emitErrorFn = ::llzk::wrapNullableInFlightDiagnostic(emitError, ctx);
  auto dimSizesRes = forceIntAttrTypes(dimensionSizes, emitErrorFn);
  if(::mlir::failed(dimSizesRes)) { return ArrayType(); }
  ::llvm::SmallVector<::mlir::Attribute> dimSizes = *dimSizesRes;
  ::llvm::SmallVector<int64_t> shape;
  ::mlir::LogicalResult res = computeShapeFromDims(emitErrorFn, dimSizes, shape);
  if(::mlir::failed(res)) { return ArrayType(); }
  return Base::get(ctx, elementType, dimSizes, shape);
}
 
ArrayType ArrayType::getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::Type elementType, ::llvm::ArrayRef<::mlir::Attribute> dimensionSizes) {
  assert(elementType && "element type cannot be null");
  ::mlir::MLIRContext *ctx = elementType.getContext();
  OwningEmitErrorFn emitErrorFn = ::llzk::wrapNullableInFlightDiagnostic(emitError, ctx);
  auto dimSizesRes = forceIntAttrTypes(dimensionSizes, emitErrorFn);
  if(::mlir::failed(dimSizesRes)) { return ArrayType(); }
  ::llvm::SmallVector<::mlir::Attribute> dimSizes = *dimSizesRes;
  ::llvm::SmallVector<int64_t> shape;
  ::mlir::LogicalResult res = computeShapeFromDims(emitErrorFn, dimSizes, shape);
  if(::mlir::failed(res)) { return ArrayType(); }
  return Base::getChecked(emitError, ctx, elementType, dimSizes, shape);
}
 
ArrayType ArrayType::get(::mlir::Type elementType, ::llvm::ArrayRef<int64_t> shape) {
  assert(elementType && "element type cannot be null");
  ::mlir::MLIRContext *ctx = elementType.getContext();
  ::llvm::SmallVector<::mlir::Attribute> dimSizes;
  ::mlir::LogicalResult res = computeDimsFromShape(ctx, shape, dimSizes);
  if(::mlir::failed(res)) { return ArrayType(); }
  return Base::get(ctx, elementType, dimSizes, shape);
}
 
ArrayType ArrayType::getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::Type elementType, ::llvm::ArrayRef<int64_t> shape) {
  assert(elementType && "element type cannot be null");
  ::mlir::MLIRContext *ctx = elementType.getContext();
  ::llvm::SmallVector<::mlir::Attribute> dimSizes;
  ::mlir::LogicalResult res = computeDimsFromShape(ctx, shape, dimSizes);
  if(::mlir::failed(res)) { return ArrayType(); }
  return Base::getChecked(emitError, ctx, elementType, dimSizes, shape);
}
 
::llvm::LogicalResult ArrayType::verifyInvariants(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::Type elementType, ::llvm::ArrayRef<::mlir::Attribute> dimensionSizes, ::llvm::ArrayRef<int64_t> shape) {
  if (::mlir::failed(verify(emitError, elementType, dimensionSizes, shape)))
    return ::mlir::failure();
  return ::mlir::success();
}
 
::mlir::Type ArrayType::parse(::mlir::AsmParser &odsParser) {
  ::mlir::Builder odsBuilder(odsParser.getContext());
  ::llvm::SMLoc odsLoc = odsParser.getCurrentLocation();
  (void) odsLoc;
  ::mlir::FailureOr<::mlir::Type> _result_elementType;
  ::mlir::FailureOr<::llvm::SmallVector<::mlir::Attribute>> _result_dimensionSizes;
  ::mlir::FailureOr<::llvm::SmallVector<int64_t>> _result_shape;
  // Parse literal '<'
  if (odsParser.parseLess()) return {};
  {
    auto odsCustomLoc = odsParser.getCurrentLocation();
    (void)odsCustomLoc;
    auto odsCustomResult = parseAttrVec(odsParser,
      ::mlir::detail::unwrapForCustomParse(_result_dimensionSizes));
    if (::mlir::failed(odsCustomResult)) return {};
    if (::mlir::failed(_result_dimensionSizes)) {
      odsParser.emitError(odsCustomLoc, "custom parser failed to parse parameter 'dimensionSizes'");
      return {};
    }
  }
  {
    auto odsCustomLoc = odsParser.getCurrentLocation();
    (void)odsCustomLoc;
    auto odsCustomResult = parseDerivedShape(odsParser,
      ::mlir::detail::unwrapForCustomParse(_result_shape),
      *_result_dimensionSizes);
    if (::mlir::failed(odsCustomResult)) return {};
    if (::mlir::failed(_result_shape)) {
      odsParser.emitError(odsCustomLoc, "custom parser failed to parse parameter 'shape'");
      return {};
    }
  }
  // Parse literal 'x'
  if (odsParser.parseKeyword("x")) return {};
 
  // Parse variable 'elementType'
  _result_elementType = ::mlir::FieldParser<::mlir::Type>::parse(odsParser);
  if (::mlir::failed(_result_elementType)) {
    odsParser.emitError(odsParser.getCurrentLocation(), "failed to parse LLZK_ArrayType parameter 'elementType' which is to be a `::mlir::Type`");
    return {};
  }
  // Parse literal '>'
  if (odsParser.parseGreater()) return {};
  assert(::mlir::succeeded(_result_elementType));
  assert(::mlir::succeeded(_result_dimensionSizes));
  assert(::mlir::succeeded(_result_shape));
  return odsParser.getChecked<ArrayType>(odsLoc, odsParser.getContext(),
      ::mlir::Type((*_result_elementType)),
      ::llvm::ArrayRef<::mlir::Attribute>((*_result_dimensionSizes)),
      ::llvm::ArrayRef<int64_t>((*_result_shape)));
}
 
void ArrayType::print(::mlir::AsmPrinter &odsPrinter) const {
  ::mlir::Builder odsBuilder(getContext());
  odsPrinter << "<";
  printAttrVec(odsPrinter,
    getDimensionSizes());
  printDerivedShape(odsPrinter,
    getShape(),
    getDimensionSizes());
  odsPrinter << ' ' << "x";
  odsPrinter << ' ';
  odsPrinter.printStrippedAttrOrType(getElementType());
  odsPrinter << ">";
}
 
::mlir::Type ArrayType::getElementType() const {
  return getImpl()->elementType;
}
 
::llvm::ArrayRef<::mlir::Attribute> ArrayType::getDimensionSizes() const {
  return getImpl()->dimensionSizes;
}
 
::llvm::ArrayRef<int64_t> ArrayType::getShape() const {
  return getImpl()->shape;
}
 
 
} // namespace llzk::array
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::ArrayType)
namespace llzk::array {
 

::mlir::Type ArrayDialect::parseType(::mlir::DialectAsmParser &parser) const {
  ::llvm::SMLoc typeLoc = parser.getCurrentLocation();
  ::llvm::StringRef mnemonic;
  ::mlir::Type genType;
  auto parseResult = generatedTypeParser(parser, &mnemonic, genType);
  if (parseResult.has_value())
    return genType;
  
  parser.emitError(typeLoc) << "unknown  type `"
      << mnemonic << "` in dialect `" << getNamespace() << "`";
  return {};
}

void ArrayDialect::printType(::mlir::Type type,
                    ::mlir::DialectAsmPrinter &printer) const {
  if (::mlir::succeeded(generatedTypePrinter(type, printer)))
    return;
  
}
 
} // namespace llzk::array
 
#endif // GET_TYPEDEF_CLASSES