Ops.cpp.inc

Go to the documentation of this file.

/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* Op Definitions                                                             *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|* From: Ops.td                                                               *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/
 
#ifdef GET_OP_LIST
#undef GET_OP_LIST
 
::llzk::array::ArrayLengthOp,
::llzk::array::CreateArrayOp,
::llzk::array::ExtractArrayOp,
::llzk::array::InsertArrayOp,
::llzk::array::ReadArrayOp,
::llzk::array::WriteArrayOp
#endif  // GET_OP_LIST
 
#ifdef GET_OP_CLASSES
#undef GET_OP_CLASSES
 
 
//===----------------------------------------------------------------------===//
// Local Utility Method Definitions
//===----------------------------------------------------------------------===//
 
namespace llzk {
namespace array {
 
static ::mlir::LogicalResult __mlir_ods_local_type_constraint_Ops0(
    ::mlir::Operation *op, ::mlir::Type type, ::llvm::StringRef valueKind,
    unsigned valueIndex) {
  if (!((::llvm::isa<::llzk::array::ArrayType>(type)))) {
    return op->emitOpError(valueKind) << " #" << valueIndex
        << " must be n-dimensional array, but got " << type;
  }
  return ::mlir::success();
}
 
static ::mlir::LogicalResult __mlir_ods_local_type_constraint_Ops1(
    ::mlir::Operation *op, ::mlir::Type type, ::llvm::StringRef valueKind,
    unsigned valueIndex) {
  if (!((::llvm::isa<::mlir::IndexType>(type)))) {
    return op->emitOpError(valueKind) << " #" << valueIndex
        << " must be index, but got " << type;
  }
  return ::mlir::success();
}
 
static ::mlir::LogicalResult __mlir_ods_local_type_constraint_Ops2(
    ::mlir::Operation *op, ::mlir::Type type, ::llvm::StringRef valueKind,
    unsigned valueIndex) {
  if (!((::llzk::isValidArrayElemType(type)))) {
    return op->emitOpError(valueKind) << " #" << valueIndex
        << " must be variadic of a valid array element type, but got " << type;
  }
  return ::mlir::success();
}
 
static ::mlir::LogicalResult __mlir_ods_local_type_constraint_Ops3(
    ::mlir::Operation *op, ::mlir::Type type, ::llvm::StringRef valueKind,
    unsigned valueIndex) {
  if (!((::llvm::isa<::mlir::IndexType>(type)))) {
    return op->emitOpError(valueKind) << " #" << valueIndex
        << " must be variadic of index, but got " << type;
  }
  return ::mlir::success();
}
 
static ::mlir::LogicalResult __mlir_ods_local_type_constraint_Ops4(
    ::mlir::Operation *op, ::mlir::Type type, ::llvm::StringRef valueKind,
    unsigned valueIndex) {
  if (!((::llzk::isValidArrayElemType(type)))) {
    return op->emitOpError(valueKind) << " #" << valueIndex
        << " must be a valid array element type, but got " << type;
  }
  return ::mlir::success();
}
 
static ::mlir::LogicalResult __mlir_ods_local_attr_constraint_Ops0(
    ::mlir::Attribute attr, ::llvm::StringRef attrName, llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) {
  if (attr && !((::llvm::isa<::mlir::DenseI32ArrayAttr>(attr))))
    return emitError() << "attribute '" << attrName
        << "' failed to satisfy constraint: i32 dense array attribute";
  return ::mlir::success();
}
static ::mlir::LogicalResult __mlir_ods_local_attr_constraint_Ops0(
    ::mlir::Operation *op, ::mlir::Attribute attr, ::llvm::StringRef attrName) {
  return __mlir_ods_local_attr_constraint_Ops0(attr, attrName, [op]() {
    return op->emitOpError();
  });
}
} // namespace array
} // namespace llzk
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::ArrayLengthOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
ArrayLengthOpGenericAdaptorBase::ArrayLengthOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const ::mlir::EmptyProperties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.len", odsAttrs.getContext());
}
 
ArrayLengthOpGenericAdaptorBase::ArrayLengthOpGenericAdaptorBase(ArrayLengthOp op) : ArrayLengthOpGenericAdaptorBase(op->getAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> ArrayLengthOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  return {index, 1};
}
 
::mlir::DictionaryAttr ArrayLengthOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
} // namespace detail
ArrayLengthOpAdaptor::ArrayLengthOpAdaptor(ArrayLengthOp op) : ArrayLengthOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult ArrayLengthOpAdaptor::verify(::mlir::Location loc) {
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> ArrayLengthOp::getODSOperandIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::operand_range ArrayLengthOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> ArrayLengthOp::getArrRef() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
}
 
::mlir::TypedValue<::mlir::IndexType> ArrayLengthOp::getDim() {
  return ::llvm::cast<::mlir::TypedValue<::mlir::IndexType>>(*getODSOperands(1).begin());
}
 
::mlir::OpOperand &ArrayLengthOp::getArrRefMutable() {
  auto range = getODSOperandIndexAndLength(0);
  return getOperation()->getOpOperand(range.first);
}
 
::mlir::OpOperand &ArrayLengthOp::getDimMutable() {
  auto range = getODSOperandIndexAndLength(1);
  return getOperation()->getOpOperand(range.first);
}
 
std::pair<unsigned, unsigned> ArrayLengthOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range ArrayLengthOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::mlir::IndexType> ArrayLengthOp::getLength() {
  return ::llvm::cast<::mlir::TypedValue<::mlir::IndexType>>(*getODSResults(0).begin());
}
 
void ArrayLengthOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type length, ::mlir::Value arr_ref, ::mlir::Value dim) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(dim);
  odsState.addTypes(length);
}
 
void ArrayLengthOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value arr_ref, ::mlir::Value dim) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(dim);
 
        ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
        if (::mlir::succeeded(ArrayLengthOp::inferReturnTypes(odsBuilder.getContext(),
                      odsState.location, odsState.operands,
                      odsState.attributes.getDictionary(odsState.getContext()),
                      odsState.getRawProperties(),
                      odsState.regions, inferredReturnTypes)))
          odsState.addTypes(inferredReturnTypes);
        else
          ::llvm::report_fatal_error("Failed to infer result type(s).");
}
 
void ArrayLengthOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value arr_ref, ::mlir::Value dim) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(dim);
  assert(resultTypes.size() == 1u && "mismatched number of results");
  odsState.addTypes(resultTypes);
}
 
void ArrayLengthOp::build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() == 2u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
  assert(resultTypes.size() == 1u && "mismatched number of return types");
  odsState.addTypes(resultTypes);
}
 
void ArrayLengthOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() == 2u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
 
  ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
  if (::mlir::succeeded(ArrayLengthOp::inferReturnTypes(odsBuilder.getContext(),
          odsState.location, operands,
          odsState.attributes.getDictionary(odsState.getContext()),
          odsState.getRawProperties(),
          odsState.regions, inferredReturnTypes))) {
    assert(inferredReturnTypes.size() == 1u && "mismatched number of return types");
    odsState.addTypes(inferredReturnTypes);
  } else {
    ::llvm::report_fatal_error("Failed to infer result type(s).");
  }
}
 
::mlir::LogicalResult ArrayLengthOp::verifyInvariantsImpl() {
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops1(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSResults(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops1(*this, v.getType(), "result", index++)))
        return ::mlir::failure();
    }
  }
  return ::mlir::success();
}
 
::mlir::LogicalResult ArrayLengthOp::verifyInvariants() {
  return verifyInvariantsImpl();
}
 
::mlir::LogicalResult ArrayLengthOp::inferReturnTypes(::mlir::MLIRContext *context, ::std::optional<::mlir::Location> location, ::mlir::ValueRange operands, ::mlir::DictionaryAttr attributes, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions, ::llvm::SmallVectorImpl<::mlir::Type>&inferredReturnTypes) {
  inferredReturnTypes.resize(1);
  ::mlir::Builder odsBuilder(context);
  ::mlir::Type odsInferredType0 = odsBuilder.getIndexType();
  inferredReturnTypes[0] = odsInferredType0;
  return ::mlir::success();
}
 
::mlir::ParseResult ArrayLengthOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::mlir::OpAsmParser::UnresolvedOperand arr_refRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> arr_refOperands(arr_refRawOperands);  ::llvm::SMLoc arr_refOperandsLoc;
  (void)arr_refOperandsLoc;
  ::mlir::OpAsmParser::UnresolvedOperand dimRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> dimOperands(dimRawOperands);  ::llvm::SMLoc dimOperandsLoc;
  (void)dimOperandsLoc;
  ::mlir::Type arr_refRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> arr_refTypes(arr_refRawTypes);
 
  arr_refOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(arr_refRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseComma())
    return ::mlir::failure();
 
  dimOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(dimRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    arr_refRawTypes[0] = type;
  }
  {
    auto loc = parser.getCurrentLocation();(void)loc;
    if (parser.parseOptionalAttrDict(result.attributes))
      return ::mlir::failure();
  }
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  result.addTypes(odsBuildableType0);
  if (parser.resolveOperands(arr_refOperands, arr_refTypes, arr_refOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(dimOperands, odsBuildableType0, dimOperandsLoc, result.operands))
    return ::mlir::failure();
  return ::mlir::success();
}
 
void ArrayLengthOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  _odsPrinter << ' ';
  _odsPrinter << getArrRef();
  _odsPrinter << ",";
  _odsPrinter << ' ';
  _odsPrinter << getDim();
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getArrRef().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
  _odsPrinter.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
}
 
void ArrayLengthOp::getEffects(::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect>> &effects) {
}
 
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::ArrayLengthOp)
 
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::CreateArrayOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
CreateArrayOpGenericAdaptorBase::CreateArrayOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const Properties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), properties(properties), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.new", odsAttrs.getContext());
}
 
CreateArrayOpGenericAdaptorBase::CreateArrayOpGenericAdaptorBase(CreateArrayOp op) : CreateArrayOpGenericAdaptorBase(op->getDiscardableAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> CreateArrayOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  ::llvm::ArrayRef<int32_t> sizeAttr = getProperties().operandSegmentSizes;
 
  unsigned start = 0;
  for (unsigned i = 0; i < index; ++i)
    start += sizeAttr[i];
  return {start, sizeAttr[index]};
}
 
::mlir::DictionaryAttr CreateArrayOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
::mlir::DenseI32ArrayAttr CreateArrayOpGenericAdaptorBase::getNumDimsPerMapAttr() {
  auto attr = ::llvm::dyn_cast_or_null<::mlir::DenseI32ArrayAttr>(getProperties().numDimsPerMap);
  return attr;
}
 
::llvm::ArrayRef<int32_t> CreateArrayOpGenericAdaptorBase::getNumDimsPerMap() {
  auto attr = getNumDimsPerMapAttr();
  return attr;
}
 
::mlir::DenseI32ArrayAttr CreateArrayOpGenericAdaptorBase::getMapOpGroupSizesAttr() {
  auto attr = ::llvm::cast<::mlir::DenseI32ArrayAttr>(getProperties().mapOpGroupSizes);
  return attr;
}
 
::llvm::ArrayRef<int32_t> CreateArrayOpGenericAdaptorBase::getMapOpGroupSizes() {
  auto attr = getMapOpGroupSizesAttr();
  return attr;
}
 
} // namespace detail
CreateArrayOpAdaptor::CreateArrayOpAdaptor(CreateArrayOp op) : CreateArrayOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult CreateArrayOpAdaptor::verify(::mlir::Location loc) {
  auto tblgen_mapOpGroupSizes = getProperties().mapOpGroupSizes; (void)tblgen_mapOpGroupSizes;
  if (!tblgen_mapOpGroupSizes) return emitError(loc, "'array.new' op ""requires attribute 'mapOpGroupSizes'");
  auto tblgen_numDimsPerMap = getProperties().numDimsPerMap; (void)tblgen_numDimsPerMap;
 
  if (tblgen_numDimsPerMap && !((::llvm::isa<::mlir::DenseI32ArrayAttr>(tblgen_numDimsPerMap))))
    return emitError(loc, "'array.new' op ""attribute 'numDimsPerMap' failed to satisfy constraint: i32 dense array attribute");
 
  if (tblgen_mapOpGroupSizes && !((::llvm::isa<::mlir::DenseI32ArrayAttr>(tblgen_mapOpGroupSizes))))
    return emitError(loc, "'array.new' op ""attribute 'mapOpGroupSizes' failed to satisfy constraint: i32 dense array attribute");
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> CreateArrayOp::getODSOperandIndexAndLength(unsigned index) {
  ::llvm::ArrayRef<int32_t> sizeAttr = getProperties().operandSegmentSizes;
 
  unsigned start = 0;
  for (unsigned i = 0; i < index; ++i)
    start += sizeAttr[i];
  return {start, sizeAttr[index]};
}
 
::mlir::Operation::operand_range CreateArrayOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::Operation::operand_range CreateArrayOp::getElements() {
  return getODSOperands(0);
}
 
::mlir::OperandRangeRange CreateArrayOp::getMapOperands() {
  return getODSOperands(1).split(getMapOpGroupSizesAttr());
}
 
::mlir::MutableOperandRange CreateArrayOp::getElementsMutable() {
  auto range = getODSOperandIndexAndLength(0);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second, ::mlir::MutableOperandRange::OperandSegment(0u, {getOperandSegmentSizesAttrName(), ::mlir::DenseI32ArrayAttr::get(getContext(), getProperties().operandSegmentSizes)}));
  return mutableRange;
}
 
::mlir::MutableOperandRangeRange CreateArrayOp::getMapOperandsMutable() {
  auto range = getODSOperandIndexAndLength(1);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second, ::mlir::MutableOperandRange::OperandSegment(1u, {getOperandSegmentSizesAttrName(), ::mlir::DenseI32ArrayAttr::get(getContext(), getProperties().operandSegmentSizes)}));
  return mutableRange.split(*(*this)->getAttrDictionary().getNamed(getMapOpGroupSizesAttrName()));
}
 
std::pair<unsigned, unsigned> CreateArrayOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range CreateArrayOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> CreateArrayOp::getResult() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSResults(0).begin());
}
 
::mlir::LogicalResult CreateArrayOp::setPropertiesFromAttr(Properties &prop, ::mlir::Attribute attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) {
  ::mlir::DictionaryAttr dict = ::llvm::dyn_cast<::mlir::DictionaryAttr>(attr);
  if (!dict) {
    emitError() << "expected DictionaryAttr to set properties";
    return ::mlir::failure();
  }
 
  {
    auto &propStorage = prop.mapOpGroupSizes;
       auto attr = dict.get("mapOpGroupSizes");
    if (attr || /*isRequired=*/true) {
      if (!attr) {
        emitError() << "expected key entry for mapOpGroupSizes in DictionaryAttr to set "
                   "Properties.";
        return ::mlir::failure();
      }
      auto convertedAttr = ::llvm::dyn_cast<std::remove_reference_t<decltype(propStorage)>>(attr);
      if (convertedAttr) {
        propStorage = convertedAttr;
      } else {
        emitError() << "Invalid attribute `mapOpGroupSizes` in property conversion: " << attr;
        return ::mlir::failure();
      }
    }
  }
 
  {
    auto &propStorage = prop.numDimsPerMap;
       auto attr = dict.get("numDimsPerMap");
    if (attr || /*isRequired=*/false) {
      if (!attr) {
        emitError() << "expected key entry for numDimsPerMap in DictionaryAttr to set "
                   "Properties.";
        return ::mlir::failure();
      }
      auto convertedAttr = ::llvm::dyn_cast<std::remove_reference_t<decltype(propStorage)>>(attr);
      if (convertedAttr) {
        propStorage = convertedAttr;
      } else {
        emitError() << "Invalid attribute `numDimsPerMap` in property conversion: " << attr;
        return ::mlir::failure();
      }
    }
  }
;
    {
      auto setFromAttr = [] (auto &propStorage, ::mlir::Attribute propAttr,
               ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) {
        return convertFromAttribute(propStorage, propAttr, emitError);;
      };
         auto attr = dict.get("operandSegmentSizes");   if (!attr) attr = dict.get("operand_segment_sizes");;
      if (!attr) {
        emitError() << "expected key entry for operandSegmentSizes in DictionaryAttr to set "
                   "Properties.";
        return ::mlir::failure();
      }
      if (::mlir::failed(setFromAttr(prop.operandSegmentSizes, attr, emitError)))
        return ::mlir::failure();
    }
  return ::mlir::success();
}
 
::mlir::Attribute CreateArrayOp::getPropertiesAsAttr(::mlir::MLIRContext *ctx, const Properties &prop) {
    ::mlir::SmallVector<::mlir::NamedAttribute> attrs;
    ::mlir::Builder odsBuilder{ctx};
 
    {
      const auto &propStorage = prop.mapOpGroupSizes;
      if (propStorage)
        attrs.push_back(odsBuilder.getNamedAttr("mapOpGroupSizes",
                                       propStorage));
    }
 
    {
      const auto &propStorage = prop.numDimsPerMap;
      if (propStorage)
        attrs.push_back(odsBuilder.getNamedAttr("numDimsPerMap",
                                       propStorage));
    }
 
    {
      const auto &propStorage = prop.operandSegmentSizes;
      attrs.push_back(odsBuilder.getNamedAttr("operandSegmentSizes",
                                              ::mlir::DenseI32ArrayAttr::get(ctx, propStorage)));
    }
 
  if (!attrs.empty())
    return odsBuilder.getDictionaryAttr(attrs);
  return {};
}
 
llvm::hash_code CreateArrayOp::computePropertiesHash(const Properties &prop) {
  auto hash_operandSegmentSizes = [] (const auto &propStorage) -> llvm::hash_code {
    return ::llvm::hash_combine_range(std::begin(propStorage), std::end(propStorage));;
  };
  return llvm::hash_combine(
    llvm::hash_value(prop.mapOpGroupSizes.getAsOpaquePointer()), 
    llvm::hash_value(prop.numDimsPerMap.getAsOpaquePointer()), 
    hash_operandSegmentSizes(prop.operandSegmentSizes));
}
 
std::optional<mlir::Attribute> CreateArrayOp::getInherentAttr(::mlir::MLIRContext *ctx, const Properties &prop, llvm::StringRef name) {
    if (name == "mapOpGroupSizes")
      return prop.mapOpGroupSizes;
 
    if (name == "numDimsPerMap")
      return prop.numDimsPerMap;
    if (name == "operand_segment_sizes" || name == "operandSegmentSizes") return ::mlir::DenseI32ArrayAttr::get(ctx, prop.operandSegmentSizes);
  return std::nullopt;
}
 
void CreateArrayOp::setInherentAttr(Properties &prop, llvm::StringRef name, mlir::Attribute value) {
    if (name == "mapOpGroupSizes") {
       prop.mapOpGroupSizes = ::llvm::dyn_cast_or_null<std::remove_reference_t<decltype(prop.mapOpGroupSizes)>>(value);
       return;
    }
 
    if (name == "numDimsPerMap") {
       prop.numDimsPerMap = ::llvm::dyn_cast_or_null<std::remove_reference_t<decltype(prop.numDimsPerMap)>>(value);
       return;
    }
        if (name == "operand_segment_sizes" || name == "operandSegmentSizes") {
       auto arrAttr = ::llvm::dyn_cast_or_null<::mlir::DenseI32ArrayAttr>(value);
       if (!arrAttr) return;
       if (arrAttr.size() != sizeof(prop.operandSegmentSizes) / sizeof(int32_t))
         return;
       llvm::copy(arrAttr.asArrayRef(), prop.operandSegmentSizes.begin());
       return;
    }
}
 
void CreateArrayOp::populateInherentAttrs(::mlir::MLIRContext *ctx, const Properties &prop, ::mlir::NamedAttrList &attrs) {
    if (prop.mapOpGroupSizes) attrs.append("mapOpGroupSizes", prop.mapOpGroupSizes);
 
    if (prop.numDimsPerMap) attrs.append("numDimsPerMap", prop.numDimsPerMap);
  attrs.append("operandSegmentSizes", ::mlir::DenseI32ArrayAttr::get(ctx, prop.operandSegmentSizes));
}
 
::mlir::LogicalResult CreateArrayOp::verifyInherentAttrs(::mlir::OperationName opName, ::mlir::NamedAttrList &attrs, llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) {
    {
      ::mlir::Attribute attr = attrs.get(getMapOpGroupSizesAttrName(opName));
      if (attr && ::mlir::failed(__mlir_ods_local_attr_constraint_Ops0(attr, "mapOpGroupSizes", emitError)))
        return ::mlir::failure();
    }
 
    {
      ::mlir::Attribute attr = attrs.get(getNumDimsPerMapAttrName(opName));
      if (attr && ::mlir::failed(__mlir_ods_local_attr_constraint_Ops0(attr, "numDimsPerMap", emitError)))
        return ::mlir::failure();
    }
    return ::mlir::success();
}
 
::mlir::LogicalResult CreateArrayOp::readProperties(::mlir::DialectBytecodeReader &reader, ::mlir::OperationState &state) {
  auto &prop = state.getOrAddProperties<Properties>(); (void)prop;
  if (::mlir::failed(reader.readAttribute(prop.mapOpGroupSizes)))
    return ::mlir::failure();
 
  if (::mlir::failed(reader.readOptionalAttribute(prop.numDimsPerMap)))
    return ::mlir::failure();
 
  if (reader.getBytecodeVersion() < /*kNativePropertiesODSSegmentSize=*/6) {
    auto &propStorage = prop.operandSegmentSizes;
    ::mlir::DenseI32ArrayAttr attr;
    if (::mlir::failed(reader.readAttribute(attr))) return ::mlir::failure();
    if (attr.size() > static_cast<int64_t>(sizeof(propStorage) / sizeof(int32_t))) {
      reader.emitError("size mismatch for operand/result_segment_size");
      return ::mlir::failure();
    }
    ::llvm::copy(::llvm::ArrayRef<int32_t>(attr), propStorage.begin());
  }
 
  {
    auto &propStorage = prop.operandSegmentSizes;
    auto readProp = [&]() {
 
  if (reader.getBytecodeVersion() >= /*kNativePropertiesODSSegmentSize=*/6)
    return reader.readSparseArray(::llvm::MutableArrayRef(propStorage));
;
      return ::mlir::success();
    };
    if (::mlir::failed(readProp()))
      return ::mlir::failure();
  }
  return ::mlir::success();
}
 
void CreateArrayOp::writeProperties(::mlir::DialectBytecodeWriter &writer) {
  auto &prop = getProperties(); (void)prop;
  writer.writeAttribute(prop.mapOpGroupSizes);
 
  writer.writeOptionalAttribute(prop.numDimsPerMap);
 
if (writer.getBytecodeVersion() < /*kNativePropertiesODSSegmentSize=*/6) {
  auto &propStorage = prop.operandSegmentSizes;
  writer.writeAttribute(::mlir::DenseI32ArrayAttr::get(this->getContext(), propStorage));
}
 
  {
    auto &propStorage = prop.operandSegmentSizes;
 
  if (writer.getBytecodeVersion() >= /*kNativePropertiesODSSegmentSize=*/6)
    writer.writeSparseArray(::llvm::ArrayRef(propStorage));
;
  }
}
 
::mlir::DenseI32ArrayAttr CreateArrayOp::getNumDimsPerMapAttr() {
  return ::llvm::dyn_cast_or_null<::mlir::DenseI32ArrayAttr>(getProperties().numDimsPerMap);
}
 
::llvm::ArrayRef<int32_t> CreateArrayOp::getNumDimsPerMap() {
  auto attr = getNumDimsPerMapAttr();
  return attr;
}
 
::mlir::DenseI32ArrayAttr CreateArrayOp::getMapOpGroupSizesAttr() {
  return ::llvm::cast<::mlir::DenseI32ArrayAttr>(getProperties().mapOpGroupSizes);
}
 
::llvm::ArrayRef<int32_t> CreateArrayOp::getMapOpGroupSizes() {
  auto attr = getMapOpGroupSizesAttr();
  return attr;
}
 
void CreateArrayOp::setNumDimsPerMapAttr(::mlir::DenseI32ArrayAttr attr) {
  (*this)->setAttr(getNumDimsPerMapAttrName(), attr);
}
 
void CreateArrayOp::setNumDimsPerMap(::llvm::ArrayRef<int32_t> attrValue) {
  (*this)->setAttr(getNumDimsPerMapAttrName(), ::mlir::Builder((*this)->getContext()).getDenseI32ArrayAttr(attrValue));
}
 
void CreateArrayOp::setMapOpGroupSizesAttr(::mlir::DenseI32ArrayAttr attr) {
  (*this)->setAttr(getMapOpGroupSizesAttrName(), attr);
}
 
void CreateArrayOp::setMapOpGroupSizes(::llvm::ArrayRef<int32_t> attrValue) {
  (*this)->setAttr(getMapOpGroupSizesAttrName(), ::mlir::Builder((*this)->getContext()).getDenseI32ArrayAttr(attrValue));
}
 
void CreateArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::llzk::array::ArrayType result, ::llvm::ArrayRef<::mlir::ValueRange> mapOperands, ::llvm::ArrayRef<int32_t> numDimsPerMap) {
                        build(odsBuilder, odsState, result, mapOperands, odsBuilder.getDenseI32ArrayAttr(numDimsPerMap));
                      
}
 
void CreateArrayOp::populateDefaultProperties(::mlir::OperationName opName, Properties &properties) {
  ::mlir::Builder odsBuilder(opName.getContext());
  if (!properties.numDimsPerMap)
    properties.numDimsPerMap = odsBuilder.getDenseI32ArrayAttr({});
}
 
::mlir::LogicalResult CreateArrayOp::verifyInvariantsImpl() {
  auto tblgen_mapOpGroupSizes = getProperties().mapOpGroupSizes; (void)tblgen_mapOpGroupSizes;
  if (!tblgen_mapOpGroupSizes) return emitOpError("requires attribute 'mapOpGroupSizes'");
  auto tblgen_numDimsPerMap = getProperties().numDimsPerMap; (void)tblgen_numDimsPerMap;
 
  if (::mlir::failed(__mlir_ods_local_attr_constraint_Ops0(*this, tblgen_numDimsPerMap, "numDimsPerMap")))
    return ::mlir::failure();
 
  if (::mlir::failed(__mlir_ods_local_attr_constraint_Ops0(*this, tblgen_mapOpGroupSizes, "mapOpGroupSizes")))
    return ::mlir::failure();
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops2(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
    if (::mlir::failed(::mlir::OpTrait::impl::verifyValueSizeAttr(*this, "mapOpGroupSizes", "mapOperands", valueGroup1.size())))
      return ::mlir::failure();
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops3(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSResults(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "result", index++)))
        return ::mlir::failure();
    }
  }
  if (!((getElements().empty() || std::equal_to<>()(resultTypeToElementsTypes((*this->getODSResults(0).begin()).getType()), this->getODSOperands(0).getType()))))
    return emitOpError("failed to verify that operand types match result type");
  return ::mlir::success();
}
 
::mlir::LogicalResult CreateArrayOp::verifyInvariants() {
  if(::mlir::succeeded(verifyInvariantsImpl()) && ::mlir::succeeded(verify()))
    return ::mlir::success();
  return ::mlir::failure();
}
 
::mlir::ParseResult CreateArrayOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> mapOperandsOperands;
    llvm::SmallVector<int32_t> mapOperandsOperandGroupSizes;
  ::llvm::SMLoc mapOperandsOperandsLoc;
  (void)mapOperandsOperandsLoc;
  ::mlir::DenseI32ArrayAttr numDimsPerMapAttr;
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> elementsOperands;
  ::llvm::SMLoc elementsOperandsLoc;
  (void)elementsOperandsLoc;
  ::mlir::Type resultRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> resultTypes(resultRawTypes);
  ::llvm::SmallVector<::mlir::Type, 1> elementsTypes;
  if (::mlir::succeeded(parser.parseOptionalLBrace())) {
  {
    mapOperandsOperandsLoc = parser.getCurrentLocation();
    ::llvm::SmallVector<::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand>> mapOperandsOperandGroups;
    auto odsResult = parseMultiDimAndSymbolList(parser, mapOperandsOperandGroups, numDimsPerMapAttr);
    if (odsResult) return ::mlir::failure();
    for (const auto &subRange : mapOperandsOperandGroups) {
      mapOperandsOperands.append(subRange.begin(), subRange.end());
      mapOperandsOperandGroupSizes.push_back(subRange.size());
    }
    if (numDimsPerMapAttr)
      result.getOrAddProperties<CreateArrayOp::Properties>().numDimsPerMap = numDimsPerMapAttr;
  }
  if (parser.parseRBrace())
    return ::mlir::failure();
  } else {
 
  elementsOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperandList(elementsOperands))
    return ::mlir::failure();
  }
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    resultRawTypes[0] = type;
  }
  {
    auto odsResult = parseInferredArrayType(parser, elementsTypes, elementsOperands, resultRawTypes[0]);
    if (odsResult) return ::mlir::failure();
  }
  {
    auto odsResult = parseAttrDictWithWarnings(parser, result.attributes, result);
    if (odsResult) return ::mlir::failure();
  }
::llvm::copy(::llvm::ArrayRef<int32_t>({static_cast<int32_t>(elementsOperands.size()), static_cast<int32_t>(mapOperandsOperands.size())}), result.getOrAddProperties<CreateArrayOp::Properties>().operandSegmentSizes.begin());
  result.getOrAddProperties<CreateArrayOp::Properties>().mapOpGroupSizes = parser.getBuilder().getDenseI32ArrayAttr(mapOperandsOperandGroupSizes);
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  result.addTypes(resultTypes);
  if (parser.resolveOperands(elementsOperands, elementsTypes, elementsOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(mapOperandsOperands, odsBuildableType0, mapOperandsOperandsLoc, result.operands))
    return ::mlir::failure();
  return ::mlir::success();
}
 
void CreateArrayOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  if (((!getMapOperands().empty()) || (getNumDimsPerMapAttr() && getNumDimsPerMapAttr() != ::mlir::OpBuilder((*this)->getContext()).getDenseI32ArrayAttr({})))) {
    _odsPrinter << "{";
    printMultiDimAndSymbolList(_odsPrinter, *this, getMapOperands(), getNumDimsPerMapAttr());
    _odsPrinter << "}";
  } else {
    _odsPrinter << ' ';
    _odsPrinter << getElements();
  }
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getResult().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  printInferredArrayType(_odsPrinter, *this, getElements().getTypes(), getElements(), getResult().getType());
  _odsPrinter << ' ';
  printAttrDictWithWarnings(_odsPrinter, *this, getOperation()->getAttrDictionary(), getProperties());
}
 
void CreateArrayOp::getEffects(::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect>> &effects) {
}
 
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::CreateArrayOp)
 
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::ExtractArrayOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
ExtractArrayOpGenericAdaptorBase::ExtractArrayOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const ::mlir::EmptyProperties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.extract", odsAttrs.getContext());
}
 
ExtractArrayOpGenericAdaptorBase::ExtractArrayOpGenericAdaptorBase(ExtractArrayOp op) : ExtractArrayOpGenericAdaptorBase(op->getAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> ExtractArrayOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  bool isVariadic[] = {false, true};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (odsOperandsSize - 1) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::DictionaryAttr ExtractArrayOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
} // namespace detail
ExtractArrayOpAdaptor::ExtractArrayOpAdaptor(ExtractArrayOp op) : ExtractArrayOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult ExtractArrayOpAdaptor::verify(::mlir::Location loc) {
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> ExtractArrayOp::getODSOperandIndexAndLength(unsigned index) {
  bool isVariadic[] = {false, true};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (getOperation()->getNumOperands() - 1) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::Operation::operand_range ExtractArrayOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> ExtractArrayOp::getArrRef() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
}
 
::mlir::Operation::operand_range ExtractArrayOp::getIndices() {
  return getODSOperands(1);
}
 
::mlir::OpOperand &ExtractArrayOp::getArrRefMutable() {
  auto range = getODSOperandIndexAndLength(0);
  return getOperation()->getOpOperand(range.first);
}
 
::mlir::MutableOperandRange ExtractArrayOp::getIndicesMutable() {
  auto range = getODSOperandIndexAndLength(1);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second);
  return mutableRange;
}
 
std::pair<unsigned, unsigned> ExtractArrayOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range ExtractArrayOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> ExtractArrayOp::getResult() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSResults(0).begin());
}
 
void ExtractArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addTypes(result);
}
 
void ExtractArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
 
        ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
        if (::mlir::succeeded(ExtractArrayOp::inferReturnTypes(odsBuilder.getContext(),
                      odsState.location, odsState.operands,
                      odsState.attributes.getDictionary(odsState.getContext()),
                      odsState.getRawProperties(),
                      odsState.regions, inferredReturnTypes)))
          odsState.addTypes(inferredReturnTypes);
        else
          ::llvm::report_fatal_error("Failed to infer result type(s).");
}
 
void ExtractArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  assert(resultTypes.size() == 1u && "mismatched number of results");
  odsState.addTypes(resultTypes);
}
 
void ExtractArrayOp::build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 1u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
  assert(resultTypes.size() == 1u && "mismatched number of return types");
  odsState.addTypes(resultTypes);
}
 
void ExtractArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 1u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
 
  ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
  if (::mlir::succeeded(ExtractArrayOp::inferReturnTypes(odsBuilder.getContext(),
          odsState.location, operands,
          odsState.attributes.getDictionary(odsState.getContext()),
          odsState.getRawProperties(),
          odsState.regions, inferredReturnTypes))) {
    assert(inferredReturnTypes.size() == 1u && "mismatched number of return types");
    odsState.addTypes(inferredReturnTypes);
  } else {
    ::llvm::report_fatal_error("Failed to infer result type(s).");
  }
}
 
::mlir::LogicalResult ExtractArrayOp::verifyInvariantsImpl() {
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops3(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSResults(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "result", index++)))
        return ::mlir::failure();
    }
  }
  return ::mlir::success();
}
 
::mlir::LogicalResult ExtractArrayOp::verifyInvariants() {
  return verifyInvariantsImpl();
}
 
::mlir::ParseResult ExtractArrayOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::mlir::OpAsmParser::UnresolvedOperand arr_refRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> arr_refOperands(arr_refRawOperands);  ::llvm::SMLoc arr_refOperandsLoc;
  (void)arr_refOperandsLoc;
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> indicesOperands;
  ::llvm::SMLoc indicesOperandsLoc;
  (void)indicesOperandsLoc;
  ::mlir::Type arr_refRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> arr_refTypes(arr_refRawTypes);
 
  arr_refOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(arr_refRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseLSquare())
    return ::mlir::failure();
 
  indicesOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperandList(indicesOperands))
    return ::mlir::failure();
  if (parser.parseRSquare())
    return ::mlir::failure();
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    arr_refRawTypes[0] = type;
  }
  {
    auto loc = parser.getCurrentLocation();(void)loc;
    if (parser.parseOptionalAttrDict(result.attributes))
      return ::mlir::failure();
  }
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  if (parser.resolveOperands(arr_refOperands, arr_refTypes, arr_refOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(indicesOperands, odsBuildableType0, indicesOperandsLoc, result.operands))
    return ::mlir::failure();
 
  ::llvm::SmallVector<::mlir::Type> inferredReturnTypes;
  if (::mlir::failed(ExtractArrayOp::inferReturnTypes(parser.getContext(),
      result.location, result.operands,
      result.attributes.getDictionary(parser.getContext()),
      result.getRawProperties(),
      result.regions, inferredReturnTypes)))
    return ::mlir::failure();
  result.addTypes(inferredReturnTypes);
  return ::mlir::success();
}
 
void ExtractArrayOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  _odsPrinter << ' ';
  _odsPrinter << getArrRef();
  _odsPrinter << "[";
  _odsPrinter << getIndices();
  _odsPrinter << "]";
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getArrRef().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
  _odsPrinter.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
}
 
    ::mlir::LogicalResult
    ExtractArrayOp::inferReturnTypes(::mlir::MLIRContext *context,
                      std::optional<::mlir::Location> location,
                      ::mlir::ValueRange operands, ::mlir::DictionaryAttr attributes,
                      ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions,
                      ::llvm::SmallVectorImpl<::mlir::Type> &inferredReturnTypes) {
      ExtractArrayOp::Adaptor adaptor(operands, attributes, properties, regions);
      return ExtractArrayOp::inferReturnTypes(context,
        location, adaptor, inferredReturnTypes);
    }
 

bool ExtractArrayOp::isRead() {
  return true;
}
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::ExtractArrayOp)
 
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::InsertArrayOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
InsertArrayOpGenericAdaptorBase::InsertArrayOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const ::mlir::EmptyProperties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.insert", odsAttrs.getContext());
}
 
InsertArrayOpGenericAdaptorBase::InsertArrayOpGenericAdaptorBase(InsertArrayOp op) : InsertArrayOpGenericAdaptorBase(op->getAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> InsertArrayOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  bool isVariadic[] = {false, true, false};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (odsOperandsSize - 2) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::DictionaryAttr InsertArrayOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
} // namespace detail
InsertArrayOpAdaptor::InsertArrayOpAdaptor(InsertArrayOp op) : InsertArrayOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult InsertArrayOpAdaptor::verify(::mlir::Location loc) {
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> InsertArrayOp::getODSOperandIndexAndLength(unsigned index) {
  bool isVariadic[] = {false, true, false};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (getOperation()->getNumOperands() - 2) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::Operation::operand_range InsertArrayOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> InsertArrayOp::getArrRef() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
}
 
::mlir::Operation::operand_range InsertArrayOp::getIndices() {
  return getODSOperands(1);
}
 
::mlir::TypedValue<::llzk::array::ArrayType> InsertArrayOp::getRvalue() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(2).begin());
}
 
::mlir::OpOperand &InsertArrayOp::getArrRefMutable() {
  auto range = getODSOperandIndexAndLength(0);
  return getOperation()->getOpOperand(range.first);
}
 
::mlir::MutableOperandRange InsertArrayOp::getIndicesMutable() {
  auto range = getODSOperandIndexAndLength(1);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second);
  return mutableRange;
}
 
::mlir::OpOperand &InsertArrayOp::getRvalueMutable() {
  auto range = getODSOperandIndexAndLength(2);
  return getOperation()->getOpOperand(range.first);
}
 
std::pair<unsigned, unsigned> InsertArrayOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range InsertArrayOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
void InsertArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value arr_ref, ::mlir::ValueRange indices, ::mlir::Value rvalue) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addOperands(rvalue);
}
 
void InsertArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value arr_ref, ::mlir::ValueRange indices, ::mlir::Value rvalue) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addOperands(rvalue);
  assert(resultTypes.size() == 0u && "mismatched number of results");
  odsState.addTypes(resultTypes);
}
 
void InsertArrayOp::build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 2u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
  assert(resultTypes.size() == 0u && "mismatched number of return types");
  odsState.addTypes(resultTypes);
}
 
::mlir::LogicalResult InsertArrayOp::verifyInvariantsImpl() {
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops3(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup2 = getODSOperands(2);
 
    for (auto v : valueGroup2) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  return ::mlir::success();
}
 
::mlir::LogicalResult InsertArrayOp::verifyInvariants() {
  if(::mlir::succeeded(verifyInvariantsImpl()) && ::mlir::succeeded(verify()))
    return ::mlir::success();
  return ::mlir::failure();
}
 
::mlir::ParseResult InsertArrayOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::mlir::OpAsmParser::UnresolvedOperand arr_refRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> arr_refOperands(arr_refRawOperands);  ::llvm::SMLoc arr_refOperandsLoc;
  (void)arr_refOperandsLoc;
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> indicesOperands;
  ::llvm::SMLoc indicesOperandsLoc;
  (void)indicesOperandsLoc;
  ::mlir::OpAsmParser::UnresolvedOperand rvalueRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> rvalueOperands(rvalueRawOperands);  ::llvm::SMLoc rvalueOperandsLoc;
  (void)rvalueOperandsLoc;
  ::mlir::Type arr_refRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> arr_refTypes(arr_refRawTypes);
  ::mlir::Type rvalueRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> rvalueTypes(rvalueRawTypes);
 
  arr_refOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(arr_refRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseLSquare())
    return ::mlir::failure();
 
  indicesOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperandList(indicesOperands))
    return ::mlir::failure();
  if (parser.parseRSquare())
    return ::mlir::failure();
  if (parser.parseEqual())
    return ::mlir::failure();
 
  rvalueOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(rvalueRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    arr_refRawTypes[0] = type;
  }
  if (parser.parseComma())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    rvalueRawTypes[0] = type;
  }
  {
    auto loc = parser.getCurrentLocation();(void)loc;
    if (parser.parseOptionalAttrDict(result.attributes))
      return ::mlir::failure();
  }
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  if (parser.resolveOperands(arr_refOperands, arr_refTypes, arr_refOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(indicesOperands, odsBuildableType0, indicesOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(rvalueOperands, rvalueTypes, rvalueOperandsLoc, result.operands))
    return ::mlir::failure();
  return ::mlir::success();
}
 
void InsertArrayOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  _odsPrinter << ' ';
  _odsPrinter << getArrRef();
  _odsPrinter << "[";
  _odsPrinter << getIndices();
  _odsPrinter << "]";
  _odsPrinter << ' ' << "=";
  _odsPrinter << ' ';
  _odsPrinter << getRvalue();
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getArrRef().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  _odsPrinter << ",";
  _odsPrinter << ' ';
  {
    auto type = getRvalue().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
  _odsPrinter.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
}

bool InsertArrayOp::isRead() {
  return false;
}
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::InsertArrayOp)
 
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::ReadArrayOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
ReadArrayOpGenericAdaptorBase::ReadArrayOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const ::mlir::EmptyProperties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.read", odsAttrs.getContext());
}
 
ReadArrayOpGenericAdaptorBase::ReadArrayOpGenericAdaptorBase(ReadArrayOp op) : ReadArrayOpGenericAdaptorBase(op->getAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> ReadArrayOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  bool isVariadic[] = {false, true};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (odsOperandsSize - 1) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::DictionaryAttr ReadArrayOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
} // namespace detail
ReadArrayOpAdaptor::ReadArrayOpAdaptor(ReadArrayOp op) : ReadArrayOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult ReadArrayOpAdaptor::verify(::mlir::Location loc) {
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> ReadArrayOp::getODSOperandIndexAndLength(unsigned index) {
  bool isVariadic[] = {false, true};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (getOperation()->getNumOperands() - 1) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::Operation::operand_range ReadArrayOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> ReadArrayOp::getArrRef() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
}
 
::mlir::Operation::operand_range ReadArrayOp::getIndices() {
  return getODSOperands(1);
}
 
::mlir::OpOperand &ReadArrayOp::getArrRefMutable() {
  auto range = getODSOperandIndexAndLength(0);
  return getOperation()->getOpOperand(range.first);
}
 
::mlir::MutableOperandRange ReadArrayOp::getIndicesMutable() {
  auto range = getODSOperandIndexAndLength(1);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second);
  return mutableRange;
}
 
std::pair<unsigned, unsigned> ReadArrayOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range ReadArrayOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::Value ReadArrayOp::getResult() {
  return ::llvm::cast<::mlir::Value>(*getODSResults(0).begin());
}
 
void ReadArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Type result, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addTypes(result);
}
 
void ReadArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
 
        ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
        if (::mlir::succeeded(ReadArrayOp::inferReturnTypes(odsBuilder.getContext(),
                      odsState.location, odsState.operands,
                      odsState.attributes.getDictionary(odsState.getContext()),
                      odsState.getRawProperties(),
                      odsState.regions, inferredReturnTypes)))
          odsState.addTypes(inferredReturnTypes);
        else
          ::llvm::report_fatal_error("Failed to infer result type(s).");
}
 
void ReadArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value arr_ref, ::mlir::ValueRange indices) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  assert(resultTypes.size() == 1u && "mismatched number of results");
  odsState.addTypes(resultTypes);
}
 
void ReadArrayOp::build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 1u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
  assert(resultTypes.size() == 1u && "mismatched number of return types");
  odsState.addTypes(resultTypes);
}
 
void ReadArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 1u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
 
  ::llvm::SmallVector<::mlir::Type, 2> inferredReturnTypes;
  if (::mlir::succeeded(ReadArrayOp::inferReturnTypes(odsBuilder.getContext(),
          odsState.location, operands,
          odsState.attributes.getDictionary(odsState.getContext()),
          odsState.getRawProperties(),
          odsState.regions, inferredReturnTypes))) {
    assert(inferredReturnTypes.size() == 1u && "mismatched number of return types");
    odsState.addTypes(inferredReturnTypes);
  } else {
    ::llvm::report_fatal_error("Failed to infer result type(s).");
  }
}
 
::mlir::LogicalResult ReadArrayOp::verifyInvariantsImpl() {
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops3(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSResults(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops4(*this, v.getType(), "result", index++)))
        return ::mlir::failure();
    }
  }
  if (!((::llzk::typesUnify(::llvm::cast<::llzk::array::ArrayType>((*this->getODSOperands(0).begin()).getType()).getElementType(), (*this->getODSResults(0).begin()).getType()))))
    return emitOpError("failed to verify that result type matches with arr_ref element type");
  return ::mlir::success();
}
 
::mlir::LogicalResult ReadArrayOp::verifyInvariants() {
  if(::mlir::succeeded(verifyInvariantsImpl()) && ::mlir::succeeded(verify()))
    return ::mlir::success();
  return ::mlir::failure();
}
 
::mlir::ParseResult ReadArrayOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::mlir::OpAsmParser::UnresolvedOperand arr_refRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> arr_refOperands(arr_refRawOperands);  ::llvm::SMLoc arr_refOperandsLoc;
  (void)arr_refOperandsLoc;
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> indicesOperands;
  ::llvm::SMLoc indicesOperandsLoc;
  (void)indicesOperandsLoc;
  ::mlir::Type arr_refRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> arr_refTypes(arr_refRawTypes);
  ::mlir::Type resultRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> resultTypes(resultRawTypes);
 
  arr_refOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(arr_refRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseLSquare())
    return ::mlir::failure();
 
  indicesOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperandList(indicesOperands))
    return ::mlir::failure();
  if (parser.parseRSquare())
    return ::mlir::failure();
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    arr_refRawTypes[0] = type;
  }
  if (parser.parseComma())
    return ::mlir::failure();
 
  {
    ::mlir::Type type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    resultRawTypes[0] = type;
  }
  {
    auto loc = parser.getCurrentLocation();(void)loc;
    if (parser.parseOptionalAttrDict(result.attributes))
      return ::mlir::failure();
  }
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  result.addTypes(resultTypes);
  if (parser.resolveOperands(arr_refOperands, arr_refTypes, arr_refOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(indicesOperands, odsBuildableType0, indicesOperandsLoc, result.operands))
    return ::mlir::failure();
  return ::mlir::success();
}
 
void ReadArrayOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  _odsPrinter << ' ';
  _odsPrinter << getArrRef();
  _odsPrinter << "[";
  _odsPrinter << getIndices();
  _odsPrinter << "]";
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getArrRef().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  _odsPrinter << ",";
  _odsPrinter << ' ';
  {
    auto type = getResult().getType();
    if (auto validType = ::llvm::dyn_cast<::mlir::Type>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
  _odsPrinter.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
}
 
    ::mlir::LogicalResult
    ReadArrayOp::inferReturnTypes(::mlir::MLIRContext *context,
                      std::optional<::mlir::Location> location,
                      ::mlir::ValueRange operands, ::mlir::DictionaryAttr attributes,
                      ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions,
                      ::llvm::SmallVectorImpl<::mlir::Type> &inferredReturnTypes) {
      ReadArrayOp::Adaptor adaptor(operands, attributes, properties, regions);
      return ReadArrayOp::inferReturnTypes(context,
        location, adaptor, inferredReturnTypes);
    }
 

bool ReadArrayOp::canRewire(const ::mlir::DestructurableMemorySlot &slot,
            ::llvm::SmallPtrSetImpl<::mlir::Attribute> &usedIndices,
            ::mlir::SmallVectorImpl<::mlir::MemorySlot>
            &mustBeSafelyUsed) {
  return ::llvm::cast<ArrayAccessOpInterface>(getOperation())
            .canRewire(slot, usedIndices, mustBeSafelyUsed);
}

::mlir::DeletionKind ReadArrayOp::rewire(const
::mlir::DestructurableMemorySlot &slot,
            ::llvm::DenseMap<::mlir::Attribute, ::mlir::MemorySlot> &subslots,
            ::mlir::RewriterBase &rewriter) {
  return ::llvm::cast<ArrayAccessOpInterface>(getOperation())
            .rewire(slot, subslots, rewriter);
}

bool ReadArrayOp::loadsFrom(const ::mlir::MemorySlot &slot) {
  return getArrRef() == slot.ptr;
}

bool ReadArrayOp::storesTo(const ::mlir::MemorySlot &slot) {
  return false;
}

::mlir::Value ReadArrayOp::getStored(const ::mlir::MemorySlot &, ::mlir::RewriterBase &) {
  llvm_unreachable("getStored() should not be called on ReadArrayOp");
}

bool ReadArrayOp::isRead() {
  return true;
}
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::ReadArrayOp)
 
namespace llzk {
namespace array {
 
//===----------------------------------------------------------------------===//
// ::llzk::array::WriteArrayOp definitions
//===----------------------------------------------------------------------===//
 
namespace detail {
WriteArrayOpGenericAdaptorBase::WriteArrayOpGenericAdaptorBase(::mlir::DictionaryAttr attrs, const ::mlir::EmptyProperties &properties, ::mlir::RegionRange regions) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
    odsOpName.emplace("array.write", odsAttrs.getContext());
}
 
WriteArrayOpGenericAdaptorBase::WriteArrayOpGenericAdaptorBase(WriteArrayOp op) : WriteArrayOpGenericAdaptorBase(op->getAttrDictionary(), op.getProperties(), op->getRegions()) {}
 
std::pair<unsigned, unsigned> WriteArrayOpGenericAdaptorBase::getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
  bool isVariadic[] = {false, true, false};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (odsOperandsSize - 2) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::DictionaryAttr WriteArrayOpGenericAdaptorBase::getAttributes() {
  return odsAttrs;
}
 
} // namespace detail
WriteArrayOpAdaptor::WriteArrayOpAdaptor(WriteArrayOp op) : WriteArrayOpGenericAdaptor(op->getOperands(), op) {}
 
::mlir::LogicalResult WriteArrayOpAdaptor::verify(::mlir::Location loc) {
  return ::mlir::success();
}
 
std::pair<unsigned, unsigned> WriteArrayOp::getODSOperandIndexAndLength(unsigned index) {
  bool isVariadic[] = {false, true, false};
  int prevVariadicCount = 0;
  for (unsigned i = 0; i < index; ++i)
    if (isVariadic[i]) ++prevVariadicCount;
 
  // Calculate how many dynamic values a static variadic operand corresponds to.
  // This assumes all static variadic operands have the same dynamic value count.
  int variadicSize = (getOperation()->getNumOperands() - 2) / 1;
  // `index` passed in as the parameter is the static index which counts each
  // operand (variadic or not) as size 1. So here for each previous static variadic
  // operand, we need to offset by (variadicSize - 1) to get where the dynamic
  // value pack for this static operand starts.
  int start = index + (variadicSize - 1) * prevVariadicCount;
  int size = isVariadic[index] ? variadicSize : 1;
  return {start, size};
}
 
::mlir::Operation::operand_range WriteArrayOp::getODSOperands(unsigned index) {
  auto valueRange = getODSOperandIndexAndLength(index);
  return {std::next(getOperation()->operand_begin(), valueRange.first),
           std::next(getOperation()->operand_begin(), valueRange.first + valueRange.second)};
}
 
::mlir::TypedValue<::llzk::array::ArrayType> WriteArrayOp::getArrRef() {
  return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
}
 
::mlir::Operation::operand_range WriteArrayOp::getIndices() {
  return getODSOperands(1);
}
 
::mlir::Value WriteArrayOp::getRvalue() {
  return ::llvm::cast<::mlir::Value>(*getODSOperands(2).begin());
}
 
::mlir::OpOperand &WriteArrayOp::getArrRefMutable() {
  auto range = getODSOperandIndexAndLength(0);
  return getOperation()->getOpOperand(range.first);
}
 
::mlir::MutableOperandRange WriteArrayOp::getIndicesMutable() {
  auto range = getODSOperandIndexAndLength(1);
  auto mutableRange = ::mlir::MutableOperandRange(getOperation(), range.first, range.second);
  return mutableRange;
}
 
::mlir::OpOperand &WriteArrayOp::getRvalueMutable() {
  auto range = getODSOperandIndexAndLength(2);
  return getOperation()->getOpOperand(range.first);
}
 
std::pair<unsigned, unsigned> WriteArrayOp::getODSResultIndexAndLength(unsigned index) {
  return {index, 1};
}
 
::mlir::Operation::result_range WriteArrayOp::getODSResults(unsigned index) {
  auto valueRange = getODSResultIndexAndLength(index);
  return {std::next(getOperation()->result_begin(), valueRange.first),
           std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
}
 
void WriteArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value arr_ref, ::mlir::ValueRange indices, ::mlir::Value rvalue) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addOperands(rvalue);
}
 
void WriteArrayOp::build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value arr_ref, ::mlir::ValueRange indices, ::mlir::Value rvalue) {
  odsState.addOperands(arr_ref);
  odsState.addOperands(indices);
  odsState.addOperands(rvalue);
  assert(resultTypes.size() == 0u && "mismatched number of results");
  odsState.addTypes(resultTypes);
}
 
void WriteArrayOp::build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes) {
  assert(operands.size() >= 2u && "mismatched number of parameters");
  odsState.addOperands(operands);
  odsState.addAttributes(attributes);
  assert(resultTypes.size() == 0u && "mismatched number of return types");
  odsState.addTypes(resultTypes);
}
 
::mlir::LogicalResult WriteArrayOp::verifyInvariantsImpl() {
  {
    unsigned index = 0; (void)index;
    auto valueGroup0 = getODSOperands(0);
 
    for (auto v : valueGroup0) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops0(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup1 = getODSOperands(1);
 
    for (auto v : valueGroup1) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops3(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
    auto valueGroup2 = getODSOperands(2);
 
    for (auto v : valueGroup2) {
      if (::mlir::failed(__mlir_ods_local_type_constraint_Ops4(*this, v.getType(), "operand", index++)))
        return ::mlir::failure();
    }
  }
  if (!((::llzk::typesUnify(::llvm::cast<::llzk::array::ArrayType>((*this->getODSOperands(0).begin()).getType()).getElementType(), (*this->getODSOperands(2).begin()).getType()))))
    return emitOpError("failed to verify that rvalue type matches with arr_ref element type");
  return ::mlir::success();
}
 
::mlir::LogicalResult WriteArrayOp::verifyInvariants() {
  if(::mlir::succeeded(verifyInvariantsImpl()) && ::mlir::succeeded(verify()))
    return ::mlir::success();
  return ::mlir::failure();
}
 
::mlir::ParseResult WriteArrayOp::parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result) {
  ::mlir::OpAsmParser::UnresolvedOperand arr_refRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> arr_refOperands(arr_refRawOperands);  ::llvm::SMLoc arr_refOperandsLoc;
  (void)arr_refOperandsLoc;
  ::llvm::SmallVector<::mlir::OpAsmParser::UnresolvedOperand, 4> indicesOperands;
  ::llvm::SMLoc indicesOperandsLoc;
  (void)indicesOperandsLoc;
  ::mlir::OpAsmParser::UnresolvedOperand rvalueRawOperands[1];
  ::llvm::ArrayRef<::mlir::OpAsmParser::UnresolvedOperand> rvalueOperands(rvalueRawOperands);  ::llvm::SMLoc rvalueOperandsLoc;
  (void)rvalueOperandsLoc;
  ::mlir::Type arr_refRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> arr_refTypes(arr_refRawTypes);
  ::mlir::Type rvalueRawTypes[1];
  ::llvm::ArrayRef<::mlir::Type> rvalueTypes(rvalueRawTypes);
 
  arr_refOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(arr_refRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseLSquare())
    return ::mlir::failure();
 
  indicesOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperandList(indicesOperands))
    return ::mlir::failure();
  if (parser.parseRSquare())
    return ::mlir::failure();
  if (parser.parseEqual())
    return ::mlir::failure();
 
  rvalueOperandsLoc = parser.getCurrentLocation();
  if (parser.parseOperand(rvalueRawOperands[0]))
    return ::mlir::failure();
  if (parser.parseColon())
    return ::mlir::failure();
 
  {
    ::llzk::array::ArrayType type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    arr_refRawTypes[0] = type;
  }
  if (parser.parseComma())
    return ::mlir::failure();
 
  {
    ::mlir::Type type;
    if (parser.parseCustomTypeWithFallback(type))
      return ::mlir::failure();
    rvalueRawTypes[0] = type;
  }
  {
    auto loc = parser.getCurrentLocation();(void)loc;
    if (parser.parseOptionalAttrDict(result.attributes))
      return ::mlir::failure();
  }
  ::mlir::Type odsBuildableType0 = parser.getBuilder().getIndexType();
  if (parser.resolveOperands(arr_refOperands, arr_refTypes, arr_refOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(indicesOperands, odsBuildableType0, indicesOperandsLoc, result.operands))
    return ::mlir::failure();
  if (parser.resolveOperands(rvalueOperands, rvalueTypes, rvalueOperandsLoc, result.operands))
    return ::mlir::failure();
  return ::mlir::success();
}
 
void WriteArrayOp::print(::mlir::OpAsmPrinter &_odsPrinter) {
  _odsPrinter << ' ';
  _odsPrinter << getArrRef();
  _odsPrinter << "[";
  _odsPrinter << getIndices();
  _odsPrinter << "]";
  _odsPrinter << ' ' << "=";
  _odsPrinter << ' ';
  _odsPrinter << getRvalue();
  _odsPrinter << ' ' << ":";
  _odsPrinter << ' ';
  {
    auto type = getArrRef().getType();
    if (auto validType = ::llvm::dyn_cast<::llzk::array::ArrayType>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  _odsPrinter << ",";
  _odsPrinter << ' ';
  {
    auto type = getRvalue().getType();
    if (auto validType = ::llvm::dyn_cast<::mlir::Type>(type))
      _odsPrinter.printStrippedAttrOrType(validType);
   else
     _odsPrinter << type;
  }
  ::llvm::SmallVector<::llvm::StringRef, 2> elidedAttrs;
  _odsPrinter.printOptionalAttrDict((*this)->getAttrs(), elidedAttrs);
}

bool WriteArrayOp::canRewire(const ::mlir::DestructurableMemorySlot &slot,
            ::llvm::SmallPtrSetImpl<::mlir::Attribute> &usedIndices,
            ::mlir::SmallVectorImpl<::mlir::MemorySlot>
            &mustBeSafelyUsed) {
  return ::llvm::cast<ArrayAccessOpInterface>(getOperation())
            .canRewire(slot, usedIndices, mustBeSafelyUsed);
}

::mlir::DeletionKind WriteArrayOp::rewire(const
::mlir::DestructurableMemorySlot &slot,
            ::llvm::DenseMap<::mlir::Attribute, ::mlir::MemorySlot> &subslots,
            ::mlir::RewriterBase &rewriter) {
  return ::llvm::cast<ArrayAccessOpInterface>(getOperation())
            .rewire(slot, subslots, rewriter);
}

bool WriteArrayOp::loadsFrom(const ::mlir::MemorySlot &slot) {
  return false;
}

bool WriteArrayOp::storesTo(const ::mlir::MemorySlot &slot) {
  return getArrRef() == slot.ptr;
}

::mlir::Value WriteArrayOp::getStored(const ::mlir::MemorySlot &, ::mlir::RewriterBase &) {
  return getRvalue();
}

bool WriteArrayOp::isRead() {
  return false;
}
} // namespace array
} // namespace llzk
MLIR_DEFINE_EXPLICIT_TYPE_ID(::llzk::array::WriteArrayOp)
 
 
#endif  // GET_OP_CLASSES