pb

type AllReduceRequest struct {
	Name   string  `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Tensor *Tensor `protobuf:"bytes,2,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

type AllReduceResponse struct {
	Name   string  `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Tensor *Tensor `protobuf:"bytes,2,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

type BarrierRequest struct {
	Rank int32 `protobuf:"varint,1,opt,name=rank,proto3" json:"rank,omitempty"`
	// contains filtered or unexported fields
}

type BarrierResponse struct {
	// contains filtered or unexported fields
}

type BroadcastRequest struct {
	Name   string  `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Tensor *Tensor `protobuf:"bytes,2,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

type BroadcastResponse struct {
	Tensor *Tensor `protobuf:"bytes,1,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

type CoordinatorClient interface {
	// RegisterWorker allows a worker to join the cluster.
	RegisterWorker(ctx context.Context, in *RegisterWorkerRequest, opts ...grpc.CallOption) (*RegisterWorkerResponse, error)
	// UnregisterWorker allows a worker to leave the cluster.
	UnregisterWorker(ctx context.Context, in *UnregisterWorkerRequest, opts ...grpc.CallOption) (*UnregisterWorkerResponse, error)
	// Heartbeat is sent by workers to indicate that they are still alive.
	Heartbeat(ctx context.Context, in *HeartbeatRequest, opts ...grpc.CallOption) (*HeartbeatResponse, error)
	// StartCheckpoint initiates a distributed checkpoint.
	StartCheckpoint(ctx context.Context, in *StartCheckpointRequest, opts ...grpc.CallOption) (*StartCheckpointResponse, error)
	// EndCheckpoint is called by workers to report the completion of a checkpoint.
	EndCheckpoint(ctx context.Context, in *EndCheckpointRequest, opts ...grpc.CallOption) (*EndCheckpointResponse, error)
}

type CoordinatorServer interface {
	// RegisterWorker allows a worker to join the cluster.
	RegisterWorker(context.Context, *RegisterWorkerRequest) (*RegisterWorkerResponse, error)
	// UnregisterWorker allows a worker to leave the cluster.
	UnregisterWorker(context.Context, *UnregisterWorkerRequest) (*UnregisterWorkerResponse, error)
	// Heartbeat is sent by workers to indicate that they are still alive.
	Heartbeat(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error)
	// StartCheckpoint initiates a distributed checkpoint.
	StartCheckpoint(context.Context, *StartCheckpointRequest) (*StartCheckpointResponse, error)
	// EndCheckpoint is called by workers to report the completion of a checkpoint.
	EndCheckpoint(context.Context, *EndCheckpointRequest) (*EndCheckpointResponse, error)
	// contains filtered or unexported methods
}

type DistributedServiceClient interface {
	// AllReduce performs a single all-reduce operation on a tensor.
	AllReduce(ctx context.Context, opts ...grpc.CallOption) (grpc.BidiStreamingClient[AllReduceRequest, AllReduceResponse], error)
	// Barrier provides a synchronization point for all workers.
	Barrier(ctx context.Context, in *BarrierRequest, opts ...grpc.CallOption) (*BarrierResponse, error)
	// Broadcast sends a tensor from the root to all other workers.
	Broadcast(ctx context.Context, in *BroadcastRequest, opts ...grpc.CallOption) (*BroadcastResponse, error)
}

type DistributedServiceServer interface {
	// AllReduce performs a single all-reduce operation on a tensor.
	AllReduce(grpc.BidiStreamingServer[AllReduceRequest, AllReduceResponse]) error
	// Barrier provides a synchronization point for all workers.
	Barrier(context.Context, *BarrierRequest) (*BarrierResponse, error)
	// Broadcast sends a tensor from the root to all other workers.
	Broadcast(context.Context, *BroadcastRequest) (*BroadcastResponse, error)
	// contains filtered or unexported methods
}

type DistributedService_AllReduceClient = grpc.BidiStreamingClient[AllReduceRequest, AllReduceResponse]

type DistributedService_AllReduceServer = grpc.BidiStreamingServer[AllReduceRequest, AllReduceResponse]

type EndCheckpointRequest struct {
	WorkerId     string `protobuf:"bytes,1,opt,name=worker_id,json=workerId,proto3" json:"worker_id,omitempty"`
	Epoch        int64  `protobuf:"varint,2,opt,name=epoch,proto3" json:"epoch,omitempty"`
	CheckpointId string `protobuf:"bytes,3,opt,name=checkpoint_id,json=checkpointId,proto3" json:"checkpoint_id,omitempty"`
	// contains filtered or unexported fields
}

type EndCheckpointResponse struct {
	// contains filtered or unexported fields
}

type HeartbeatRequest struct {
	WorkerId string `protobuf:"bytes,1,opt,name=worker_id,json=workerId,proto3" json:"worker_id,omitempty"`
	// contains filtered or unexported fields
}

type HeartbeatResponse struct {
	Status string `protobuf:"bytes,1,opt,name=status,proto3" json:"status,omitempty"`
	// contains filtered or unexported fields
}

type RegisterWorkerRequest struct {
	WorkerId string `protobuf:"bytes,1,opt,name=worker_id,json=workerId,proto3" json:"worker_id,omitempty"`
	Address  string `protobuf:"bytes,2,opt,name=address,proto3" json:"address,omitempty"`
	// contains filtered or unexported fields
}

type RegisterWorkerResponse struct {
	Rank  int32    `protobuf:"varint,1,opt,name=rank,proto3" json:"rank,omitempty"`
	Peers []string `protobuf:"bytes,2,rep,name=peers,proto3" json:"peers,omitempty"`
	// contains filtered or unexported fields
}

type StartCheckpointRequest struct {
	Epoch int64  `protobuf:"varint,1,opt,name=epoch,proto3" json:"epoch,omitempty"`
	Path  string `protobuf:"bytes,2,opt,name=path,proto3" json:"path,omitempty"`
	// contains filtered or unexported fields
}

type StartCheckpointResponse struct {
	CheckpointId string `protobuf:"bytes,1,opt,name=checkpoint_id,json=checkpointId,proto3" json:"checkpoint_id,omitempty"`
	// contains filtered or unexported fields
}

type Tensor struct {
	Shape []int32   `protobuf:"varint,1,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	Data  []float32 `protobuf:"fixed32,2,rep,packed,name=data,proto3" json:"data,omitempty"` // Using float32 for simplicity of transport
	// contains filtered or unexported fields
}

type UnimplementedCoordinatorServer struct{}

type UnimplementedDistributedServiceServer struct{}

type UnregisterWorkerRequest struct {
	WorkerId string `protobuf:"bytes,1,opt,name=worker_id,json=workerId,proto3" json:"worker_id,omitempty"`
	// contains filtered or unexported fields
}

type UnregisterWorkerResponse struct {
	// contains filtered or unexported fields
}

type UnsafeCoordinatorServer interface {
	// contains filtered or unexported methods
}

type UnsafeDistributedServiceServer interface {
	// contains filtered or unexported methods
}