ccl_generated.h

Go to the documentation of this file.

// automatically generated by the FlatBuffers compiler, do not modify
 
 
#ifndef FLATBUFFERS_GENERATED_CCL_TT_TARGET_TTNN_H_
#define FLATBUFFERS_GENERATED_CCL_TT_TARGET_TTNN_H_
 
#include "flatbuffers/flatbuffers.h"
 
// Ensure the included flatbuffers.h is the same version as when this file was
// generated, otherwise it may not be compatible.
static_assert(FLATBUFFERS_VERSION_MAJOR == 24 &&
              FLATBUFFERS_VERSION_MINOR == 3 &&
              FLATBUFFERS_VERSION_REVISION == 25,
             "Non-compatible flatbuffers version included");
 
#include "ttmlir/Target/Common/types_generated.h"
#include "ttmlir/Target/TTNN/types_generated.h"
 
namespace tt {
namespace target {
namespace ttnn {
 
struct AllGatherOp;
struct AllGatherOpBuilder;
 
struct CollectivePermuteOp;
struct CollectivePermuteOpBuilder;
 
struct MeshShardOp;
struct MeshShardOpBuilder;
 
struct ReduceScatterOp;
struct ReduceScatterOpBuilder;
 
struct AllGatherOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef AllGatherOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_IN = 4,
    VT_OUT = 6,
    VT_DEVICE = 8,
    VT_ALL_GATHER_DIM = 10,
    VT_CLUSTER_AXIS = 12,
    VT_NUM_LINKS = 14
  };
  const tt::target::ttnn::TensorRef *in() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_IN);
  }
  const tt::target::ttnn::TensorRef *out() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
  }
  const tt::target::DeviceRef *device() const {
    return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
  }
  int32_t all_gather_dim() const {
    return GetField<int32_t>(VT_ALL_GATHER_DIM, 0);
  }
  uint32_t cluster_axis() const {
    return GetField<uint32_t>(VT_CLUSTER_AXIS, 0);
  }
  uint32_t num_links() const {
    return GetField<uint32_t>(VT_NUM_LINKS, 0);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_IN) &&
           verifier.VerifyTable(in()) &&
           VerifyOffset(verifier, VT_OUT) &&
           verifier.VerifyTable(out()) &&
           VerifyOffset(verifier, VT_DEVICE) &&
           verifier.VerifyTable(device()) &&
           VerifyField<int32_t>(verifier, VT_ALL_GATHER_DIM, 4) &&
           VerifyField<uint32_t>(verifier, VT_CLUSTER_AXIS, 4) &&
           VerifyField<uint32_t>(verifier, VT_NUM_LINKS, 4) &&
           verifier.EndTable();
  }
};
 
struct AllGatherOpBuilder {
  typedef AllGatherOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_in(::flatbuffers::Offset<tt::target::ttnn::TensorRef> in) {
    fbb_.AddOffset(AllGatherOp::VT_IN, in);
  }
  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
    fbb_.AddOffset(AllGatherOp::VT_OUT, out);
  }
  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
    fbb_.AddOffset(AllGatherOp::VT_DEVICE, device);
  }
  void add_all_gather_dim(int32_t all_gather_dim) {
    fbb_.AddElement<int32_t>(AllGatherOp::VT_ALL_GATHER_DIM, all_gather_dim, 0);
  }
  void add_cluster_axis(uint32_t cluster_axis) {
    fbb_.AddElement<uint32_t>(AllGatherOp::VT_CLUSTER_AXIS, cluster_axis, 0);
  }
  void add_num_links(uint32_t num_links) {
    fbb_.AddElement<uint32_t>(AllGatherOp::VT_NUM_LINKS, num_links, 0);
  }
  explicit AllGatherOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<AllGatherOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<AllGatherOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<AllGatherOp> CreateAllGatherOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    int32_t all_gather_dim = 0,
    uint32_t cluster_axis = 0,
    uint32_t num_links = 0) {
  AllGatherOpBuilder builder_(_fbb);
  builder_.add_num_links(num_links);
  builder_.add_cluster_axis(cluster_axis);
  builder_.add_all_gather_dim(all_gather_dim);
  builder_.add_device(device);
  builder_.add_out(out);
  builder_.add_in(in);
  return builder_.Finish();
}
 
struct AllGatherOp::Traits {
  using type = AllGatherOp;
  static auto constexpr Create = CreateAllGatherOp;
};
 
struct CollectivePermuteOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef CollectivePermuteOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_IN = 4,
    VT_OUT = 6,
    VT_DEVICE = 8,
    VT_SOURCE_TARGET_PAIRS = 10
  };
  const tt::target::ttnn::TensorRef *in() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_IN);
  }
  const tt::target::ttnn::TensorRef *out() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
  }
  const tt::target::DeviceRef *device() const {
    return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
  }
  const ::flatbuffers::Vector<int64_t> *source_target_pairs() const {
    return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SOURCE_TARGET_PAIRS);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_IN) &&
           verifier.VerifyTable(in()) &&
           VerifyOffset(verifier, VT_OUT) &&
           verifier.VerifyTable(out()) &&
           VerifyOffset(verifier, VT_DEVICE) &&
           verifier.VerifyTable(device()) &&
           VerifyOffset(verifier, VT_SOURCE_TARGET_PAIRS) &&
           verifier.VerifyVector(source_target_pairs()) &&
           verifier.EndTable();
  }
};
 
struct CollectivePermuteOpBuilder {
  typedef CollectivePermuteOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_in(::flatbuffers::Offset<tt::target::ttnn::TensorRef> in) {
    fbb_.AddOffset(CollectivePermuteOp::VT_IN, in);
  }
  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
    fbb_.AddOffset(CollectivePermuteOp::VT_OUT, out);
  }
  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
    fbb_.AddOffset(CollectivePermuteOp::VT_DEVICE, device);
  }
  void add_source_target_pairs(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> source_target_pairs) {
    fbb_.AddOffset(CollectivePermuteOp::VT_SOURCE_TARGET_PAIRS, source_target_pairs);
  }
  explicit CollectivePermuteOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<CollectivePermuteOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<CollectivePermuteOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<CollectivePermuteOp> CreateCollectivePermuteOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> source_target_pairs = 0) {
  CollectivePermuteOpBuilder builder_(_fbb);
  builder_.add_source_target_pairs(source_target_pairs);
  builder_.add_device(device);
  builder_.add_out(out);
  builder_.add_in(in);
  return builder_.Finish();
}
 
struct CollectivePermuteOp::Traits {
  using type = CollectivePermuteOp;
  static auto constexpr Create = CreateCollectivePermuteOp;
};
 
inline ::flatbuffers::Offset<CollectivePermuteOp> CreateCollectivePermuteOpDirect(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    const std::vector<int64_t> *source_target_pairs = nullptr) {
  auto source_target_pairs__ = source_target_pairs ? _fbb.CreateVector<int64_t>(*source_target_pairs) : 0;
  return tt::target::ttnn::CreateCollectivePermuteOp(
      _fbb,
      in,
      out,
      device,
      source_target_pairs__);
}
 
struct MeshShardOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef MeshShardOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_IN = 4,
    VT_OUT = 6,
    VT_DEVICE = 8,
    VT_SHARD_DIRECTION = 10,
    VT_SHARD_TYPE = 12,
    VT_SHARD_SHAPE = 14,
    VT_SHARD_DIMS = 16
  };
  const tt::target::ttnn::TensorRef *in() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_IN);
  }
  const tt::target::ttnn::TensorRef *out() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
  }
  const tt::target::DeviceRef *device() const {
    return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
  }
  tt::target::ttnn::MeshShardDirection shard_direction() const {
    return static_cast<tt::target::ttnn::MeshShardDirection>(GetField<uint32_t>(VT_SHARD_DIRECTION, 0));
  }
  tt::target::ttnn::MeshShardType shard_type() const {
    return static_cast<tt::target::ttnn::MeshShardType>(GetField<uint32_t>(VT_SHARD_TYPE, 0));
  }
  const ::flatbuffers::Vector<int64_t> *shard_shape() const {
    return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHARD_SHAPE);
  }
  const ::flatbuffers::Vector<int64_t> *shard_dims() const {
    return GetPointer<const ::flatbuffers::Vector<int64_t> *>(VT_SHARD_DIMS);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_IN) &&
           verifier.VerifyTable(in()) &&
           VerifyOffset(verifier, VT_OUT) &&
           verifier.VerifyTable(out()) &&
           VerifyOffset(verifier, VT_DEVICE) &&
           verifier.VerifyTable(device()) &&
           VerifyField<uint32_t>(verifier, VT_SHARD_DIRECTION, 4) &&
           VerifyField<uint32_t>(verifier, VT_SHARD_TYPE, 4) &&
           VerifyOffset(verifier, VT_SHARD_SHAPE) &&
           verifier.VerifyVector(shard_shape()) &&
           VerifyOffset(verifier, VT_SHARD_DIMS) &&
           verifier.VerifyVector(shard_dims()) &&
           verifier.EndTable();
  }
};
 
struct MeshShardOpBuilder {
  typedef MeshShardOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_in(::flatbuffers::Offset<tt::target::ttnn::TensorRef> in) {
    fbb_.AddOffset(MeshShardOp::VT_IN, in);
  }
  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
    fbb_.AddOffset(MeshShardOp::VT_OUT, out);
  }
  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
    fbb_.AddOffset(MeshShardOp::VT_DEVICE, device);
  }
  void add_shard_direction(tt::target::ttnn::MeshShardDirection shard_direction) {
    fbb_.AddElement<uint32_t>(MeshShardOp::VT_SHARD_DIRECTION, static_cast<uint32_t>(shard_direction), 0);
  }
  void add_shard_type(tt::target::ttnn::MeshShardType shard_type) {
    fbb_.AddElement<uint32_t>(MeshShardOp::VT_SHARD_TYPE, static_cast<uint32_t>(shard_type), 0);
  }
  void add_shard_shape(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shard_shape) {
    fbb_.AddOffset(MeshShardOp::VT_SHARD_SHAPE, shard_shape);
  }
  void add_shard_dims(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shard_dims) {
    fbb_.AddOffset(MeshShardOp::VT_SHARD_DIMS, shard_dims);
  }
  explicit MeshShardOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<MeshShardOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<MeshShardOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<MeshShardOp> CreateMeshShardOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    tt::target::ttnn::MeshShardDirection shard_direction = tt::target::ttnn::MeshShardDirection::FullToShardShape,
    tt::target::ttnn::MeshShardType shard_type = tt::target::ttnn::MeshShardType::Identity,
    ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shard_shape = 0,
    ::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> shard_dims = 0) {
  MeshShardOpBuilder builder_(_fbb);
  builder_.add_shard_dims(shard_dims);
  builder_.add_shard_shape(shard_shape);
  builder_.add_shard_type(shard_type);
  builder_.add_shard_direction(shard_direction);
  builder_.add_device(device);
  builder_.add_out(out);
  builder_.add_in(in);
  return builder_.Finish();
}
 
struct MeshShardOp::Traits {
  using type = MeshShardOp;
  static auto constexpr Create = CreateMeshShardOp;
};
 
inline ::flatbuffers::Offset<MeshShardOp> CreateMeshShardOpDirect(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    tt::target::ttnn::MeshShardDirection shard_direction = tt::target::ttnn::MeshShardDirection::FullToShardShape,
    tt::target::ttnn::MeshShardType shard_type = tt::target::ttnn::MeshShardType::Identity,
    const std::vector<int64_t> *shard_shape = nullptr,
    const std::vector<int64_t> *shard_dims = nullptr) {
  auto shard_shape__ = shard_shape ? _fbb.CreateVector<int64_t>(*shard_shape) : 0;
  auto shard_dims__ = shard_dims ? _fbb.CreateVector<int64_t>(*shard_dims) : 0;
  return tt::target::ttnn::CreateMeshShardOp(
      _fbb,
      in,
      out,
      device,
      shard_direction,
      shard_type,
      shard_shape__,
      shard_dims__);
}
 
struct ReduceScatterOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef ReduceScatterOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_IN = 4,
    VT_OUT = 6,
    VT_DEVICE = 8,
    VT_SCATTER_DIM = 10,
    VT_REDUCE_TYPE = 12,
    VT_CLUSTER_AXIS = 14,
    VT_NUM_LINKS = 16
  };
  const tt::target::ttnn::TensorRef *in() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_IN);
  }
  const tt::target::ttnn::TensorRef *out() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_OUT);
  }
  const tt::target::DeviceRef *device() const {
    return GetPointer<const tt::target::DeviceRef *>(VT_DEVICE);
  }
  int32_t scatter_dim() const {
    return GetField<int32_t>(VT_SCATTER_DIM, 0);
  }
  uint32_t reduce_type() const {
    return GetField<uint32_t>(VT_REDUCE_TYPE, 0);
  }
  uint32_t cluster_axis() const {
    return GetField<uint32_t>(VT_CLUSTER_AXIS, 0);
  }
  uint32_t num_links() const {
    return GetField<uint32_t>(VT_NUM_LINKS, 0);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_IN) &&
           verifier.VerifyTable(in()) &&
           VerifyOffset(verifier, VT_OUT) &&
           verifier.VerifyTable(out()) &&
           VerifyOffset(verifier, VT_DEVICE) &&
           verifier.VerifyTable(device()) &&
           VerifyField<int32_t>(verifier, VT_SCATTER_DIM, 4) &&
           VerifyField<uint32_t>(verifier, VT_REDUCE_TYPE, 4) &&
           VerifyField<uint32_t>(verifier, VT_CLUSTER_AXIS, 4) &&
           VerifyField<uint32_t>(verifier, VT_NUM_LINKS, 4) &&
           verifier.EndTable();
  }
};
 
struct ReduceScatterOpBuilder {
  typedef ReduceScatterOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_in(::flatbuffers::Offset<tt::target::ttnn::TensorRef> in) {
    fbb_.AddOffset(ReduceScatterOp::VT_IN, in);
  }
  void add_out(::flatbuffers::Offset<tt::target::ttnn::TensorRef> out) {
    fbb_.AddOffset(ReduceScatterOp::VT_OUT, out);
  }
  void add_device(::flatbuffers::Offset<tt::target::DeviceRef> device) {
    fbb_.AddOffset(ReduceScatterOp::VT_DEVICE, device);
  }
  void add_scatter_dim(int32_t scatter_dim) {
    fbb_.AddElement<int32_t>(ReduceScatterOp::VT_SCATTER_DIM, scatter_dim, 0);
  }
  void add_reduce_type(uint32_t reduce_type) {
    fbb_.AddElement<uint32_t>(ReduceScatterOp::VT_REDUCE_TYPE, reduce_type, 0);
  }
  void add_cluster_axis(uint32_t cluster_axis) {
    fbb_.AddElement<uint32_t>(ReduceScatterOp::VT_CLUSTER_AXIS, cluster_axis, 0);
  }
  void add_num_links(uint32_t num_links) {
    fbb_.AddElement<uint32_t>(ReduceScatterOp::VT_NUM_LINKS, num_links, 0);
  }
  explicit ReduceScatterOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<ReduceScatterOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<ReduceScatterOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<ReduceScatterOp> CreateReduceScatterOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> in = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> out = 0,
    ::flatbuffers::Offset<tt::target::DeviceRef> device = 0,
    int32_t scatter_dim = 0,
    uint32_t reduce_type = 0,
    uint32_t cluster_axis = 0,
    uint32_t num_links = 0) {
  ReduceScatterOpBuilder builder_(_fbb);
  builder_.add_num_links(num_links);
  builder_.add_cluster_axis(cluster_axis);
  builder_.add_reduce_type(reduce_type);
  builder_.add_scatter_dim(scatter_dim);
  builder_.add_device(device);
  builder_.add_out(out);
  builder_.add_in(in);
  return builder_.Finish();
}
 
struct ReduceScatterOp::Traits {
  using type = ReduceScatterOp;
  static auto constexpr Create = CreateReduceScatterOp;
};
 
}  // namespace ttnn
}  // namespace target
}  // namespace tt
 
#endif  // FLATBUFFERS_GENERATED_CCL_TT_TARGET_TTNN_H_