pub struct MultiGpuCoordinator { /* private fields */ }Expand description
Multi-GPU coordinator for parallel processing
Implementations§
Source§impl MultiGpuCoordinator
impl MultiGpuCoordinator
Sourcepub fn submit_task(&mut self, task: GpuTask) -> Result<(), SimdError>
pub fn submit_task(&mut self, task: GpuTask) -> Result<(), SimdError>
Add a task to the scheduler
Sourcepub fn execute_all(&mut self) -> Result<Vec<CompletedTask>, SimdError>
pub fn execute_all(&mut self) -> Result<Vec<CompletedTask>, SimdError>
Execute all pending tasks
Sourcepub fn distributed_matrix_multiply(
&mut self,
a: &[f32],
b: &[f32],
a_rows: usize,
a_cols: usize,
b_cols: usize,
) -> Result<Vec<f32>, SimdError>
pub fn distributed_matrix_multiply( &mut self, a: &[f32], b: &[f32], a_rows: usize, a_cols: usize, b_cols: usize, ) -> Result<Vec<f32>, SimdError>
Execute a distributed matrix multiplication
Sourcepub fn set_load_balancing(&mut self, strategy: LoadBalancingStrategy)
pub fn set_load_balancing(&mut self, strategy: LoadBalancingStrategy)
Set load balancing strategy
Sourcepub fn get_device_stats(&self) -> HashMap<u32, DeviceStats>
pub fn get_device_stats(&self) -> HashMap<u32, DeviceStats>
Get device utilization statistics
Auto Trait Implementations§
impl !RefUnwindSafe for MultiGpuCoordinator
impl !Send for MultiGpuCoordinator
impl !Sync for MultiGpuCoordinator
impl !UnwindSafe for MultiGpuCoordinator
impl Freeze for MultiGpuCoordinator
impl Unpin for MultiGpuCoordinator
impl UnsafeUnpin for MultiGpuCoordinator
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more