Ops.h.inc

Go to the documentation of this file.

/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* Op Declarations                                                            *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|* From: Ops.td                                                               *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/
 
namespace llzk {
namespace constrain {
class EmitContainmentOp;
} // namespace constrain
} // namespace llzk
namespace llzk {
namespace constrain {
class EmitEqualityOp;
} // namespace constrain
} // namespace llzk
#ifdef GET_OP_CLASSES
#undef GET_OP_CLASSES
 
namespace llzk {
namespace constrain {
 
//===----------------------------------------------------------------------===//
// ::llzk::constrain::EmitContainmentOp declarations
//===----------------------------------------------------------------------===//
 
namespace detail {
class EmitContainmentOpGenericAdaptorBase {
public:
protected:
  ::mlir::DictionaryAttr odsAttrs;
  ::std::optional<::mlir::OperationName> odsOpName;
  ::mlir::RegionRange odsRegions;
public:
  EmitContainmentOpGenericAdaptorBase(::mlir::DictionaryAttr attrs = {}, const ::mlir::EmptyProperties &properties = {}, ::mlir::RegionRange regions = {}) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
      odsOpName.emplace("constrain.in", odsAttrs.getContext());
  }
 
  EmitContainmentOpGenericAdaptorBase(::mlir::Operation *op) : odsAttrs(op->getRawDictionaryAttrs()), odsOpName(op->getName()), odsRegions(op->getRegions()) {}
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
    return {index, 1};
  }
 
  ::mlir::DictionaryAttr getAttributes() {
    return odsAttrs;
  }
 
};
} // namespace detail
template <typename RangeT>
class EmitContainmentOpGenericAdaptor : public detail::EmitContainmentOpGenericAdaptorBase {
  using ValueT = ::llvm::detail::ValueOfRange<RangeT>;
  using Base = detail::EmitContainmentOpGenericAdaptorBase;
public:
  EmitContainmentOpGenericAdaptor(RangeT values, ::mlir::DictionaryAttr attrs = {}, const ::mlir::EmptyProperties &properties = {}, ::mlir::RegionRange regions = {}) : Base(attrs, properties, regions), odsOperands(values) {}
 
  EmitContainmentOpGenericAdaptor(RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions = {}) : EmitContainmentOpGenericAdaptor(values, attrs, (properties ? *properties.as<::mlir::EmptyProperties *>() : ::mlir::EmptyProperties{}), regions) {}
 
  EmitContainmentOpGenericAdaptor(RangeT values, const EmitContainmentOpGenericAdaptorBase &base) : Base(base), odsOperands(values) {}
 
  template <typename LateInst = EmitContainmentOp, typename = std::enable_if_t<std::is_same_v<LateInst, EmitContainmentOp>>>
  EmitContainmentOpGenericAdaptor(RangeT values, LateInst op) : Base(op), odsOperands(values) {}
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index) {
    return Base::getODSOperandIndexAndLength(index, odsOperands.size());
  }
 
  RangeT getODSOperands(unsigned index) {
    auto valueRange = getODSOperandIndexAndLength(index);
    return {std::next(odsOperands.begin(), valueRange.first),
             std::next(odsOperands.begin(), valueRange.first + valueRange.second)};
  }
 
  ValueT getLhs() {
    return (*getODSOperands(0).begin());
  }
 
  ValueT getRhs() {
    return (*getODSOperands(1).begin());
  }
 
  RangeT getOperands() {
    return odsOperands;
  }
 
private:
  RangeT odsOperands;
};
class EmitContainmentOpAdaptor : public EmitContainmentOpGenericAdaptor<::mlir::ValueRange> {
public:
  using EmitContainmentOpGenericAdaptor::EmitContainmentOpGenericAdaptor;
  EmitContainmentOpAdaptor(EmitContainmentOp op);
 
  ::llvm::LogicalResult verify(::mlir::Location loc);
};
class EmitContainmentOp : public ::mlir::Op<EmitContainmentOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::NOperands<2>::Impl, ::llzk::function::ConstraintGen, ::mlir::OpTrait::OpInvariants, ::llzk::constrain::ConstraintOpInterface::Trait, ::mlir::SymbolUserOpInterface::Trait> {
public:
  using Op::Op;
  using Op::print;
  using Adaptor = EmitContainmentOpAdaptor;
  template <typename RangeT>
  using GenericAdaptor = EmitContainmentOpGenericAdaptor<RangeT>;
  using FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute>>;
  static ::llvm::ArrayRef<::llvm::StringRef> getAttributeNames() {
    return {};
  }
 
  static constexpr ::llvm::StringLiteral getOperationName() {
    return ::llvm::StringLiteral("constrain.in");
  }
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index) {
    return {index, 1};
  }
 
  ::mlir::Operation::operand_range 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> getLhs() {
    return ::llvm::cast<::mlir::TypedValue<::llzk::array::ArrayType>>(*getODSOperands(0).begin());
  }
 
  ::mlir::TypedValue<::mlir::Type> getRhs() {
    return ::llvm::cast<::mlir::TypedValue<::mlir::Type>>(*getODSOperands(1).begin());
  }
 
  ::mlir::OpOperand &getLhsMutable() {
    auto range = getODSOperandIndexAndLength(0);
    return getOperation()->getOpOperand(range.first);
  }
 
  ::mlir::OpOperand &getRhsMutable() {
    auto range = getODSOperandIndexAndLength(1);
    return getOperation()->getOpOperand(range.first);
  }
 
  std::pair<unsigned, unsigned> getODSResultIndexAndLength(unsigned index) {
    return {index, 1};
  }
 
  ::mlir::Operation::result_range getODSResults(unsigned index) {
    auto valueRange = getODSResultIndexAndLength(index);
    return {std::next(getOperation()->result_begin(), valueRange.first),
             std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
  }
 
  static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value lhs, ::mlir::Value rhs);
  static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value lhs, ::mlir::Value rhs);
  static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
  ::llvm::LogicalResult verifyInvariantsImpl();
  ::llvm::LogicalResult verifyInvariants();
  ::llvm::LogicalResult verify();
  ::llvm::LogicalResult verifySymbolUses(::mlir::SymbolTableCollection &symbolTable);
  static ::mlir::ParseResult parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result);
  void print(::mlir::OpAsmPrinter &_odsPrinter);
public:
};
} // namespace constrain
} // namespace llzk
MLIR_DECLARE_EXPLICIT_TYPE_ID(::llzk::constrain::EmitContainmentOp)
 
namespace llzk {
namespace constrain {
 
//===----------------------------------------------------------------------===//
// ::llzk::constrain::EmitEqualityOp declarations
//===----------------------------------------------------------------------===//
 
namespace detail {
class EmitEqualityOpGenericAdaptorBase {
public:
protected:
  ::mlir::DictionaryAttr odsAttrs;
  ::std::optional<::mlir::OperationName> odsOpName;
  ::mlir::RegionRange odsRegions;
public:
  EmitEqualityOpGenericAdaptorBase(::mlir::DictionaryAttr attrs = {}, const ::mlir::EmptyProperties &properties = {}, ::mlir::RegionRange regions = {}) : odsAttrs(attrs), odsRegions(regions) {  if (odsAttrs)
      odsOpName.emplace("constrain.eq", odsAttrs.getContext());
  }
 
  EmitEqualityOpGenericAdaptorBase(::mlir::Operation *op) : odsAttrs(op->getRawDictionaryAttrs()), odsOpName(op->getName()), odsRegions(op->getRegions()) {}
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index, unsigned odsOperandsSize) {
    return {index, 1};
  }
 
  ::mlir::DictionaryAttr getAttributes() {
    return odsAttrs;
  }
 
};
} // namespace detail
template <typename RangeT>
class EmitEqualityOpGenericAdaptor : public detail::EmitEqualityOpGenericAdaptorBase {
  using ValueT = ::llvm::detail::ValueOfRange<RangeT>;
  using Base = detail::EmitEqualityOpGenericAdaptorBase;
public:
  EmitEqualityOpGenericAdaptor(RangeT values, ::mlir::DictionaryAttr attrs = {}, const ::mlir::EmptyProperties &properties = {}, ::mlir::RegionRange regions = {}) : Base(attrs, properties, regions), odsOperands(values) {}
 
  EmitEqualityOpGenericAdaptor(RangeT values, ::mlir::DictionaryAttr attrs, ::mlir::OpaqueProperties properties, ::mlir::RegionRange regions = {}) : EmitEqualityOpGenericAdaptor(values, attrs, (properties ? *properties.as<::mlir::EmptyProperties *>() : ::mlir::EmptyProperties{}), regions) {}
 
  EmitEqualityOpGenericAdaptor(RangeT values, const EmitEqualityOpGenericAdaptorBase &base) : Base(base), odsOperands(values) {}
 
  template <typename LateInst = EmitEqualityOp, typename = std::enable_if_t<std::is_same_v<LateInst, EmitEqualityOp>>>
  EmitEqualityOpGenericAdaptor(RangeT values, LateInst op) : Base(op), odsOperands(values) {}
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index) {
    return Base::getODSOperandIndexAndLength(index, odsOperands.size());
  }
 
  RangeT getODSOperands(unsigned index) {
    auto valueRange = getODSOperandIndexAndLength(index);
    return {std::next(odsOperands.begin(), valueRange.first),
             std::next(odsOperands.begin(), valueRange.first + valueRange.second)};
  }
 
  ValueT getLhs() {
    return (*getODSOperands(0).begin());
  }
 
  ValueT getRhs() {
    return (*getODSOperands(1).begin());
  }
 
  RangeT getOperands() {
    return odsOperands;
  }
 
private:
  RangeT odsOperands;
};
class EmitEqualityOpAdaptor : public EmitEqualityOpGenericAdaptor<::mlir::ValueRange> {
public:
  using EmitEqualityOpGenericAdaptor::EmitEqualityOpGenericAdaptor;
  EmitEqualityOpAdaptor(EmitEqualityOp op);
 
  ::llvm::LogicalResult verify(::mlir::Location loc);
};
class EmitEqualityOp : public ::mlir::Op<EmitEqualityOp, ::mlir::OpTrait::ZeroRegions, ::mlir::OpTrait::ZeroResults, ::mlir::OpTrait::ZeroSuccessors, ::mlir::OpTrait::NOperands<2>::Impl, ::llzk::function::ConstraintGen, ::mlir::OpTrait::OpInvariants, ::mlir::OpTrait::IsCommutative, ::mlir::OpTrait::Elementwise, ::mlir::OpTrait::Scalarizable, ::mlir::OpTrait::Vectorizable, ::mlir::OpTrait::Tensorizable, ::llzk::constrain::ConstraintOpInterface::Trait, ::mlir::SymbolUserOpInterface::Trait> {
public:
  using Op::Op;
  using Op::print;
  using Adaptor = EmitEqualityOpAdaptor;
  template <typename RangeT>
  using GenericAdaptor = EmitEqualityOpGenericAdaptor<RangeT>;
  using FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute>>;
  static ::llvm::ArrayRef<::llvm::StringRef> getAttributeNames() {
    return {};
  }
 
  static constexpr ::llvm::StringLiteral getOperationName() {
    return ::llvm::StringLiteral("constrain.eq");
  }
 
  std::pair<unsigned, unsigned> getODSOperandIndexAndLength(unsigned index) {
    return {index, 1};
  }
 
  ::mlir::Operation::operand_range 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<::mlir::Type> getLhs() {
    return ::llvm::cast<::mlir::TypedValue<::mlir::Type>>(*getODSOperands(0).begin());
  }
 
  ::mlir::TypedValue<::mlir::Type> getRhs() {
    return ::llvm::cast<::mlir::TypedValue<::mlir::Type>>(*getODSOperands(1).begin());
  }
 
  ::mlir::OpOperand &getLhsMutable() {
    auto range = getODSOperandIndexAndLength(0);
    return getOperation()->getOpOperand(range.first);
  }
 
  ::mlir::OpOperand &getRhsMutable() {
    auto range = getODSOperandIndexAndLength(1);
    return getOperation()->getOpOperand(range.first);
  }
 
  std::pair<unsigned, unsigned> getODSResultIndexAndLength(unsigned index) {
    return {index, 1};
  }
 
  ::mlir::Operation::result_range getODSResults(unsigned index) {
    auto valueRange = getODSResultIndexAndLength(index);
    return {std::next(getOperation()->result_begin(), valueRange.first),
             std::next(getOperation()->result_begin(), valueRange.first + valueRange.second)};
  }
 
  static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::Value lhs, ::mlir::Value rhs);
  static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value lhs, ::mlir::Value rhs);
  static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = {});
  ::llvm::LogicalResult verifyInvariantsImpl();
  ::llvm::LogicalResult verifyInvariants();
  ::llvm::LogicalResult verifySymbolUses(::mlir::SymbolTableCollection &symbolTable);
  static ::mlir::ParseResult parse(::mlir::OpAsmParser &parser, ::mlir::OperationState &result);
  void print(::mlir::OpAsmPrinter &_odsPrinter);
public:
  static ::mlir::Type inferRHS(::mlir::Type lhsType);
 
  static ::mlir::ParseResult parseInferredOrParsedType(::mlir::OpAsmParser &parser,
      ::mlir::Type &rhsType, ::mlir::Type lhsType
  ) {
    if (mlir::succeeded(parser.parseOptionalComma())) {
      // If there is a comma, parse the `rhsType`
      mlir::Type type;
      if (parser.parseCustomTypeWithFallback(type)) {
        return mlir::failure();
      }
      rhsType = type;
    } else {
      // Otherwise, infer the `rhsType` from `lhsType`
      rhsType = inferRHS(lhsType);
    }
    return mlir::success();
  }
 
  static void printInferredOrParsedType(::mlir::OpAsmPrinter &printer,
      ::mlir::Operation *op, ::mlir::Type rhsType, ::mlir::Type lhsType
  ) {
    printer << ", ";
    printer.printStrippedAttrOrType(rhsType);
  }
};
} // namespace constrain
} // namespace llzk
MLIR_DECLARE_EXPLICIT_TYPE_ID(::llzk::constrain::EmitEqualityOp)
 
 
#endif  // GET_OP_CLASSES