/*===- TableGen'erated file -------------------------------------*- C++ -*-===*\
|*                                                                            *|
|* AttrDef Declarations                                                       *|
|*                                                                            *|
|* Automatically generated file, do not edit!                                 *|
|*                                                                            *|
\*===----------------------------------------------------------------------===*/

#ifdef GET_ATTRDEF_CLASSES
#undef GET_ATTRDEF_CLASSES


namespace mlir {
class AsmParser;
class AsmPrinter;
} // namespace mlir
namespace mlir {
namespace mesh {
class PartialAttr;
class MeshShardingAttr;
namespace detail {
struct PartialAttrStorage;
} // namespace detail
class PartialAttr : public ::mlir::Attribute::AttrBase<PartialAttr, ::mlir::Attribute, detail::PartialAttrStorage> {
public:
  using Base::Base;
  static constexpr ::llvm::StringLiteral name = "mesh.partial";
  static PartialAttr get(::mlir::MLIRContext *context, ::mlir::mesh::Partial value);
  static constexpr ::llvm::StringLiteral getMnemonic() {
    return {"partial"};
  }

  static ::mlir::Attribute parse(::mlir::AsmParser &odsParser, ::mlir::Type odsType);
  void print(::mlir::AsmPrinter &odsPrinter) const;
  ::mlir::mesh::Partial getValue() const;
};
namespace detail {
struct MeshShardingAttrStorage;
} // namespace detail
class MeshShardingAttr : public ::mlir::Attribute::AttrBase<MeshShardingAttr, ::mlir::Attribute, detail::MeshShardingAttrStorage> {
public:
  using Base::Base;
  bool operator==(::mlir::Attribute rhs) const;
  bool operator==(::mlir::mesh::MeshShardingAttr rhs) const;
  static constexpr ::llvm::StringLiteral name = "mesh.shard";
  using Base::getChecked;
  static MeshShardingAttr get(::mlir::MLIRContext *context, ::mlir::FlatSymbolRefAttr cluster, ::llvm::ArrayRef<MeshAxesAttr> split_axes, ::llvm::ArrayRef<MeshAxis> partial_axes, ::mlir::mesh::Partial partial_type);
  static MeshShardingAttr getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, ::mlir::FlatSymbolRefAttr cluster, ::llvm::ArrayRef<MeshAxesAttr> split_axes, ::llvm::ArrayRef<MeshAxis> partial_axes, ::mlir::mesh::Partial partial_type);
  static MeshShardingAttr get(::mlir::MLIRContext *context, FlatSymbolRefAttr cluster, ArrayRef<SmallVector<MeshAxis>> split_axes, ArrayRef<MeshAxis> partial_axes, mesh::Partial partial_type);
  static MeshShardingAttr getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, FlatSymbolRefAttr cluster, ArrayRef<SmallVector<MeshAxis>> split_axes, ArrayRef<MeshAxis> partial_axes, mesh::Partial partial_type);
  static MeshShardingAttr get(::mlir::MLIRContext *context, FlatSymbolRefAttr cluster, ArrayRef<SmallVector<MeshAxis>> split_axes);
  static MeshShardingAttr getChecked(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, FlatSymbolRefAttr cluster, ArrayRef<SmallVector<MeshAxis>> split_axes);
  static ::mlir::LogicalResult verify(::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::FlatSymbolRefAttr cluster, ::llvm::ArrayRef<MeshAxesAttr> split_axes, ::llvm::ArrayRef<MeshAxis> partial_axes, ::mlir::mesh::Partial partial_type);
  static constexpr ::llvm::StringLiteral getMnemonic() {
    return {"shard"};
  }

  static ::mlir::Attribute parse(::mlir::AsmParser &odsParser, ::mlir::Type odsType);
  void print(::mlir::AsmPrinter &odsPrinter) const;
  ::mlir::FlatSymbolRefAttr getCluster() const;
  ::llvm::ArrayRef<MeshAxesAttr> getSplitAxes() const;
  ::llvm::ArrayRef<MeshAxis> getPartialAxes() const;
  ::mlir::mesh::Partial getPartialType() const;
};
} // namespace mesh
} // namespace mlir
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::mesh::PartialAttr)
MLIR_DECLARE_EXPLICIT_TYPE_ID(::mlir::mesh::MeshShardingAttr)

#endif  // GET_ATTRDEF_CLASSES

