Struct InterGpuTransfer 

pub struct InterGpuTransfer { /* private fields */ }

Inter-GPU data transfer manager.

Implementations

impl InterGpuTransfer

pub fn new(manager: Arc<MultiGpuManager>) -> Self

Create a new inter-GPU transfer manager.

pub async fn copy_buffer( &self, src_device: usize, dst_device: usize, data: &[u8], ) -> GpuResult<()>

Copy data between GPUs.

Errors

Returns an error if transfer fails or devices are invalid.

pub async fn broadcast(&self, data: &[u8]) -> GpuResult<()>

Broadcast data to all GPUs.

Errors

Returns an error if any transfer fails.

pub async fn gather(&self, dst_device: usize) -> GpuResult<Vec<Vec<u8>>>

Gather data from all GPUs to one device.

Errors

Returns an error if any transfer fails.