Struct UpdateWeightsFromDistributedReqInput 

pub struct UpdateWeightsFromDistributedReqInput {
    pub rid: Option<StrOrList>,
    pub http_worker_ipc: Option<String>,
    pub names: Vec<String>,
    pub dtype_names: Vec<String>,
    pub shapes: Vec<Vec<i64>>,
    pub group_name: String,
    pub flush_cache: bool,
    pub weight_version: Option<String>,
}

Receive new weights from the trainer’s NCCL group.

Fields

rid: Option<StrOrList>

Request id; control operations rarely set it.

http_worker_ipc: Option<String>

In-process HTTP worker address; unused across processes.

names: Vec<String>

Parameter names, in broadcast order.

dtype_names: Vec<String>

Their dtypes.

shapes: Vec<Vec<i64>>

Their shapes.

group_name: String

The weight-update group.

flush_cache: bool

Drop the cached KV first.

weight_version: Option<String>

The L3 namespace afterwards; None keeps the current one.

Trait Implementations

impl Clone for UpdateWeightsFromDistributedReqInput

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for UpdateWeightsFromDistributedReqInput

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for UpdateWeightsFromDistributedReqInput

fn eq(&self, other: &Self) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for UpdateWeightsFromDistributedReqInput

impl Wire for UpdateWeightsFromDistributedReqInput

fn encode(&self, encoder: &mut Encoder)

Append this value to a message.

fn decode(decoder: &mut Decoder<'_>) -> Result<Self>

Read one value of this type.