ccl_generated.h File Reference

#include "flatbuffers/flatbuffers.h"
#include "ttmlir/Target/Common/types_generated.h"
#include "ttmlir/Target/TTNN/types_generated.h"

Go to the source code of this file.

Classes
struct	tt::target::ttnn::FLATBUFFERS_FINAL_CLASS
struct	tt::target::ttnn::AllGatherOpBuilder
struct	tt::target::ttnn::AllGatherOp::Traits
struct	tt::target::ttnn::CollectivePermuteOpBuilder
struct	tt::target::ttnn::CollectivePermuteOp::Traits
struct	tt::target::ttnn::MeshShardOpBuilder
struct	tt::target::ttnn::MeshShardOp::Traits
struct	tt::target::ttnn::ReduceScatterOpBuilder
struct	tt::target::ttnn::ReduceScatterOp::Traits

Namespaces
	tt
	tt::target
	tt::target::ttnn

Functions
inline ::flatbuffers::Offset< AllGatherOp >	tt::target::ttnn::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)
inline ::flatbuffers::Offset< CollectivePermuteOp >	tt::target::ttnn::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)
inline ::flatbuffers::Offset< CollectivePermuteOp >	tt::target::ttnn::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)
inline ::flatbuffers::Offset< MeshShardOp >	tt::target::ttnn::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)
inline ::flatbuffers::Offset< MeshShardOp >	tt::target::ttnn::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)
inline ::flatbuffers::Offset< ReduceScatterOp >	tt::target::ttnn::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)

Variables
	VT_IN = 4
	VT_OUT = 6
	VT_DEVICE = 8
	VT_ALL_GATHER_DIM = 10
	VT_CLUSTER_AXIS = 12
	VT_SHARD_DIRECTION = 10
	VT_SHARD_TYPE = 12
	VT_SHARD_SHAPE = 14
	VT_SCATTER_DIM = 10
	VT_REDUCE_TYPE = 12

Variable Documentation

VT_ALL_GATHER_DIM

VT_ALL_GATHER_DIM = 10

VT_CLUSTER_AXIS

VT_CLUSTER_AXIS = 12

VT_DEVICE

VT_DEVICE = 8

VT_IN

VT_IN = 4

VT_OUT

VT_OUT = 6

VT_REDUCE_TYPE

VT_REDUCE_TYPE = 12

VT_SCATTER_DIM

VT_SCATTER_DIM = 10

VT_SHARD_DIRECTION

VT_SHARD_DIRECTION = 10

VT_SHARD_SHAPE

VT_SHARD_SHAPE = 14

VT_SHARD_TYPE

VT_SHARD_TYPE = 12