4 #ifndef FLATBUFFERS_GENERATED_TYPES_TT_TARGET_METAL_H_
5 #define FLATBUFFERS_GENERATED_TYPES_TT_TARGET_METAL_H_
7 #include "flatbuffers/flatbuffers.h"
11 static_assert(FLATBUFFERS_VERSION_MAJOR == 24 &&
12 FLATBUFFERS_VERSION_MINOR == 3 &&
13 FLATBUFFERS_VERSION_REVISION == 25,
14 "Non-compatible flatbuffers version included");
23 struct ShardSpecBuilder;
25 struct ShardSpecBuffer;
26 struct ShardSpecBufferBuilder;
28 struct ShardedBufferConfig;
29 struct ShardedBufferConfigBuilder;
31 struct CircularBufferConfig;
32 struct CircularBufferConfigBuilder;
35 struct BufferDescBuilder;
38 struct BufferRefBuilder;
43 struct ShardSpec FLATBUFFERS_FINAL_CLASS :
private ::flatbuffers::Table {
47 return "tt.target.metal.ShardSpec";
49 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
53 const ::flatbuffers::Vector<const tt::target::Dim2dRange *> *
core_range_set()
const {
54 return GetPointer<const ::flatbuffers::Vector<const tt::target::Dim2dRange *> *>(
VT_CORE_RANGE_SET);
59 bool Verify(::flatbuffers::Verifier &verifier)
const {
60 return VerifyTableStart(verifier) &&
62 verifier.VerifyVector(core_range_set()) &&
70 ::flatbuffers::FlatBufferBuilder &
fbb_;
72 void add_core_range_set(::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::Dim2dRange *>> core_range_set) {
82 ::flatbuffers::Offset<ShardSpec>
Finish() {
84 auto o = ::flatbuffers::Offset<ShardSpec>(end);
90 ::flatbuffers::FlatBufferBuilder &_fbb,
91 ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::Dim2dRange *>> core_range_set = 0,
92 const tt::target::Dim2d *shard_shape =
nullptr) {
105 ::flatbuffers::FlatBufferBuilder &_fbb,
106 const std::vector<tt::target::Dim2dRange> *core_range_set =
nullptr,
107 const tt::target::Dim2d *shard_shape =
nullptr) {
108 auto core_range_set__ = core_range_set ? _fbb.CreateVectorOfStructs<tt::target::Dim2dRange>(*core_range_set) : 0;
119 return "tt.target.metal.ShardSpecBuffer";
121 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
124 VT_TENSOR_SHAPE_IN_PAGES = 8
127 return GetPointer<const tt::target::metal::ShardSpec *>(
VT_SHARD_SPEC);
133 return GetStruct<const tt::target::Dim2d *>(VT_TENSOR_SHAPE_IN_PAGES);
135 bool Verify(::flatbuffers::Verifier &verifier)
const {
136 return VerifyTableStart(verifier) &&
138 verifier.VerifyTable(shard_spec()) &&
139 VerifyField<tt::target::Dim2d>(verifier,
VT_PAGE_SHAPE, 4) &&
140 VerifyField<tt::target::Dim2d>(verifier, VT_TENSOR_SHAPE_IN_PAGES, 4) &&
147 ::flatbuffers::FlatBufferBuilder &
fbb_;
149 void add_shard_spec(::flatbuffers::Offset<tt::target::metal::ShardSpec> shard_spec) {
156 fbb_.AddStruct(ShardSpecBuffer::VT_TENSOR_SHAPE_IN_PAGES, tensor_shape_in_pages);
162 ::flatbuffers::Offset<ShardSpecBuffer>
Finish() {
164 auto o = ::flatbuffers::Offset<ShardSpecBuffer>(end);
170 ::flatbuffers::FlatBufferBuilder &_fbb,
171 ::flatbuffers::Offset<tt::target::metal::ShardSpec> shard_spec = 0,
172 const tt::target::Dim2d *page_shape =
nullptr,
173 const tt::target::Dim2d *tensor_shape_in_pages =
nullptr) {
190 return "tt.target.metal.ShardedBufferConfig";
192 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
195 VT_SHARD_SPEC_BUFFER = 8
198 return GetField<uint64_t>(
VT_SIZE, 0);
204 return GetPointer<const tt::target::metal::ShardSpecBuffer *>(VT_SHARD_SPEC_BUFFER);
206 bool Verify(::flatbuffers::Verifier &verifier)
const {
207 return VerifyTableStart(verifier) &&
208 VerifyField<uint64_t>(verifier,
VT_SIZE, 8) &&
210 VerifyOffset(verifier, VT_SHARD_SPEC_BUFFER) &&
211 verifier.VerifyTable(shard_spec_buffer()) &&
218 ::flatbuffers::FlatBufferBuilder &
fbb_;
227 fbb_.AddOffset(ShardedBufferConfig::VT_SHARD_SPEC_BUFFER, shard_spec_buffer);
233 ::flatbuffers::Offset<ShardedBufferConfig>
Finish() {
235 auto o = ::flatbuffers::Offset<ShardedBufferConfig>(end);
241 ::flatbuffers::FlatBufferBuilder &_fbb,
243 uint64_t page_size = 0,
244 ::flatbuffers::Offset<tt::target::metal::ShardSpecBuffer> shard_spec_buffer = 0) {
253 using type = ShardedBufferConfig;
261 return "tt.target.metal.CircularBufferConfig";
263 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
269 const ::flatbuffers::Vector<const tt::target::Dim2dRange *> *
core_range_set()
const {
270 return GetPointer<const ::flatbuffers::Vector<const tt::target::Dim2dRange *> *>(
VT_CORE_RANGE_SET);
279 return GetField<uint64_t>(VT_NUM_BUFFERS, 0);
281 bool Verify(::flatbuffers::Verifier &verifier)
const {
282 return VerifyTableStart(verifier) &&
284 verifier.VerifyVector(core_range_set()) &&
287 VerifyField<uint64_t>(verifier, VT_NUM_BUFFERS, 8) &&
294 ::flatbuffers::FlatBufferBuilder &
fbb_;
296 void add_core_range_set(::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::Dim2dRange *>> core_range_set) {
306 fbb_.AddElement<uint64_t>(CircularBufferConfig::VT_NUM_BUFFERS, num_buffers, 0);
312 ::flatbuffers::Offset<CircularBufferConfig>
Finish() {
314 auto o = ::flatbuffers::Offset<CircularBufferConfig>(end);
320 ::flatbuffers::FlatBufferBuilder &_fbb,
321 ::flatbuffers::Offset<::flatbuffers::Vector<const tt::target::Dim2dRange *>> core_range_set = 0,
322 uint64_t total_size = 0,
323 uint64_t page_size = 0,
324 uint64_t num_buffers = 0) {
334 using type = CircularBufferConfig;
339 ::flatbuffers::FlatBufferBuilder &_fbb,
340 const std::vector<tt::target::Dim2dRange> *core_range_set =
nullptr,
341 uint64_t total_size = 0,
342 uint64_t page_size = 0,
343 uint64_t num_buffers = 0) {
344 auto core_range_set__ = core_range_set ? _fbb.CreateVectorOfStructs<tt::target::Dim2dRange>(*core_range_set) : 0;
357 return "tt.target.metal.BufferDesc";
359 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
365 VT_CIRCULAR_BUFFER_CONFIG = 14
367 const ::flatbuffers::Vector<int32_t> *
shape()
const {
368 return GetPointer<const ::flatbuffers::Vector<int32_t> *>(
VT_SHAPE);
383 return GetPointer<const tt::target::metal::CircularBufferConfig *>(VT_CIRCULAR_BUFFER_CONFIG);
385 bool Verify(::flatbuffers::Verifier &verifier)
const {
386 return VerifyTableStart(verifier) &&
388 verifier.VerifyVector(shape()) &&
389 VerifyField<tt::target::Dim2d>(verifier,
VT_TILE_SHAPE, 4) &&
393 verifier.VerifyTable(sharded_buffer_config()) &&
394 VerifyOffset(verifier, VT_CIRCULAR_BUFFER_CONFIG) &&
395 verifier.VerifyTable(circular_buffer_config()) &&
402 ::flatbuffers::FlatBufferBuilder &
fbb_;
404 void add_shape(::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape) {
420 fbb_.AddOffset(BufferDesc::VT_CIRCULAR_BUFFER_CONFIG, circular_buffer_config);
426 ::flatbuffers::Offset<BufferDesc>
Finish() {
428 auto o = ::flatbuffers::Offset<BufferDesc>(end);
434 ::flatbuffers::FlatBufferBuilder &_fbb,
435 ::flatbuffers::Offset<::flatbuffers::Vector<int32_t>> shape = 0,
436 const tt::target::Dim2d *tile_shape =
nullptr,
439 ::flatbuffers::Offset<tt::target::metal::ShardedBufferConfig> sharded_buffer_config = 0,
440 ::flatbuffers::Offset<tt::target::metal::CircularBufferConfig> circular_buffer_config = 0) {
457 ::flatbuffers::FlatBufferBuilder &_fbb,
458 const std::vector<int32_t> *shape =
nullptr,
459 const tt::target::Dim2d *tile_shape =
nullptr,
462 ::flatbuffers::Offset<tt::target::metal::ShardedBufferConfig> sharded_buffer_config = 0,
463 ::flatbuffers::Offset<tt::target::metal::CircularBufferConfig> circular_buffer_config = 0) {
464 auto shape__ = shape ? _fbb.CreateVector<int32_t>(*shape) : 0;
471 sharded_buffer_config,
472 circular_buffer_config);
479 return "tt.target.metal.BufferRef";
481 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
492 const tt::target::metal::BufferDesc *
desc()
const {
493 return GetPointer<const tt::target::metal::BufferDesc *>(VT_DESC);
495 bool Verify(::flatbuffers::Verifier &verifier)
const {
496 return VerifyTableStart(verifier) &&
498 VerifyField<uint64_t>(verifier,
VT_ADDRESS, 8) &&
499 VerifyOffset(verifier, VT_DESC) &&
500 verifier.VerifyTable(desc()) &&
507 ::flatbuffers::FlatBufferBuilder &
fbb_;
515 void add_desc(::flatbuffers::Offset<tt::target::metal::BufferDesc> desc) {
516 fbb_.AddOffset(BufferRef::VT_DESC, desc);
522 ::flatbuffers::Offset<BufferRef>
Finish() {
524 auto o = ::flatbuffers::Offset<BufferRef>(end);
530 ::flatbuffers::FlatBufferBuilder &_fbb,
531 uint32_t global_id = 0,
532 uint64_t address = 0,
533 ::flatbuffers::Offset<tt::target::metal::BufferDesc> desc = 0) {
550 return "tt.target.metal.CBRef";
552 enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE {
557 return GetField<uint32_t>(
VT_PORT, 0);
560 return GetPointer<const tt::target::metal::BufferRef *>(VT_BUFFER_REF);
562 bool Verify(::flatbuffers::Verifier &verifier)
const {
563 return VerifyTableStart(verifier) &&
564 VerifyField<uint32_t>(verifier,
VT_PORT, 4) &&
565 VerifyOffset(verifier, VT_BUFFER_REF) &&
566 verifier.VerifyTable(buffer_ref()) &&
573 ::flatbuffers::FlatBufferBuilder &
fbb_;
578 void add_buffer_ref(::flatbuffers::Offset<tt::target::metal::BufferRef> buffer_ref) {
579 fbb_.AddOffset(CBRef::VT_BUFFER_REF, buffer_ref);
587 auto o = ::flatbuffers::Offset<CBRef>(end);
593 ::flatbuffers::FlatBufferBuilder &_fbb,
595 ::flatbuffers::Offset<tt::target::metal::BufferRef> buffer_ref = 0) {
VT_SHARD_SHAPE
Definition: ccl_generated.h:249
MemorySpace
Definition: types_generated.h:183
DataType
Definition: types_generated.h:81
Definition: debug_info_generated.h:18
Definition: debug_info_generated.h:36