Struct BroadcastRequest 

pub struct BroadcastRequest<T: NcclReduceSupported> {
    pub data: GpuRef<T>,
    pub root: usize,
    pub reply: Sender<Result<(), GpuError>>,
}

In-place broadcast: every rank’s data buffer is overwritten with the contents of data on the rank whose comm.rank() == root.

Fields

data: GpuRef<T>

root: usize

reply: Sender<Result<(), GpuError>>

Trait Implementations

impl<T: NcclReduceSupported> CollectiveDispatch for BroadcastRequest<T>

fn dtype_kind(&self) -> DispatchDType

fn device_id(&self) -> Option<u32>

fn dispatch(self: Box<Self>, ctx: &CollectiveDispatchCtx<'_>)