kv_cache_generated.h

Go to the documentation of this file.

// automatically generated by the FlatBuffers compiler, do not modify
 
 
#ifndef FLATBUFFERS_GENERATED_KVCACHE_TT_TARGET_TTNN_H_
#define FLATBUFFERS_GENERATED_KVCACHE_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 FillCacheOp;
struct FillCacheOpBuilder;
 
struct UpdateCacheOp;
struct UpdateCacheOpBuilder;
 
struct FillCacheOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef FillCacheOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_CACHE = 4,
    VT_INPUT = 6,
    VT_BATCH_OFFSET = 8
  };
  const tt::target::ttnn::TensorRef *cache() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_CACHE);
  }
  const tt::target::ttnn::TensorRef *input() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_INPUT);
  }
  uint32_t batch_offset() const {
    return GetField<uint32_t>(VT_BATCH_OFFSET, 0);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_CACHE) &&
           verifier.VerifyTable(cache()) &&
           VerifyOffset(verifier, VT_INPUT) &&
           verifier.VerifyTable(input()) &&
           VerifyField<uint32_t>(verifier, VT_BATCH_OFFSET, 4) &&
           verifier.EndTable();
  }
};
 
struct FillCacheOpBuilder {
  typedef FillCacheOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_cache(::flatbuffers::Offset<tt::target::ttnn::TensorRef> cache) {
    fbb_.AddOffset(FillCacheOp::VT_CACHE, cache);
  }
  void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
    fbb_.AddOffset(FillCacheOp::VT_INPUT, input);
  }
  void add_batch_offset(uint32_t batch_offset) {
    fbb_.AddElement<uint32_t>(FillCacheOp::VT_BATCH_OFFSET, batch_offset, 0);
  }
  explicit FillCacheOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<FillCacheOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<FillCacheOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<FillCacheOp> CreateFillCacheOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> cache = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
    uint32_t batch_offset = 0) {
  FillCacheOpBuilder builder_(_fbb);
  builder_.add_batch_offset(batch_offset);
  builder_.add_input(input);
  builder_.add_cache(cache);
  return builder_.Finish();
}
 
struct FillCacheOp::Traits {
  using type = FillCacheOp;
  static auto constexpr Create = CreateFillCacheOp;
};
 
struct UpdateCacheOp FLATBUFFERS_FINAL_CLASS : private ::flatbuffers::Table {
  typedef UpdateCacheOpBuilder Builder;
  struct Traits;
  enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
    VT_CACHE = 4,
    VT_INPUT = 6,
    VT_UPDATE_INDEX = 8,
    VT_BATCH_OFFSET = 10
  };
  const tt::target::ttnn::TensorRef *cache() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_CACHE);
  }
  const tt::target::ttnn::TensorRef *input() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_INPUT);
  }
  const tt::target::ttnn::TensorRef *update_index() const {
    return GetPointer<const tt::target::ttnn::TensorRef *>(VT_UPDATE_INDEX);
  }
  uint32_t batch_offset() const {
    return GetField<uint32_t>(VT_BATCH_OFFSET, 0);
  }
  bool Verify(::flatbuffers::Verifier &verifier) const {
    return VerifyTableStart(verifier) &&
           VerifyOffset(verifier, VT_CACHE) &&
           verifier.VerifyTable(cache()) &&
           VerifyOffset(verifier, VT_INPUT) &&
           verifier.VerifyTable(input()) &&
           VerifyOffset(verifier, VT_UPDATE_INDEX) &&
           verifier.VerifyTable(update_index()) &&
           VerifyField<uint32_t>(verifier, VT_BATCH_OFFSET, 4) &&
           verifier.EndTable();
  }
};
 
struct UpdateCacheOpBuilder {
  typedef UpdateCacheOp Table;
  ::flatbuffers::FlatBufferBuilder &fbb_;
  ::flatbuffers::uoffset_t start_;
  void add_cache(::flatbuffers::Offset<tt::target::ttnn::TensorRef> cache) {
    fbb_.AddOffset(UpdateCacheOp::VT_CACHE, cache);
  }
  void add_input(::flatbuffers::Offset<tt::target::ttnn::TensorRef> input) {
    fbb_.AddOffset(UpdateCacheOp::VT_INPUT, input);
  }
  void add_update_index(::flatbuffers::Offset<tt::target::ttnn::TensorRef> update_index) {
    fbb_.AddOffset(UpdateCacheOp::VT_UPDATE_INDEX, update_index);
  }
  void add_batch_offset(uint32_t batch_offset) {
    fbb_.AddElement<uint32_t>(UpdateCacheOp::VT_BATCH_OFFSET, batch_offset, 0);
  }
  explicit UpdateCacheOpBuilder(::flatbuffers::FlatBufferBuilder &_fbb)
        : fbb_(_fbb) {
    start_ = fbb_.StartTable();
  }
  ::flatbuffers::Offset<UpdateCacheOp> Finish() {
    const auto end = fbb_.EndTable(start_);
    auto o = ::flatbuffers::Offset<UpdateCacheOp>(end);
    return o;
  }
};
 
inline ::flatbuffers::Offset<UpdateCacheOp> CreateUpdateCacheOp(
    ::flatbuffers::FlatBufferBuilder &_fbb,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> cache = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> input = 0,
    ::flatbuffers::Offset<tt::target::ttnn::TensorRef> update_index = 0,
    uint32_t batch_offset = 0) {
  UpdateCacheOpBuilder builder_(_fbb);
  builder_.add_batch_offset(batch_offset);
  builder_.add_update_index(update_index);
  builder_.add_input(input);
  builder_.add_cache(cache);
  return builder_.Finish();
}
 
struct UpdateCacheOp::Traits {
  using type = UpdateCacheOp;
  static auto constexpr Create = CreateUpdateCacheOp;
};
 
}  // namespace ttnn
}  // namespace target
}  // namespace tt
 
#endif  // FLATBUFFERS_GENERATED_KVCACHE_TT_TARGET_TTNN_H_