pub struct GpuMemoryManager { /* private fields */ }Implementations§
Source§impl GpuMemoryManager
impl GpuMemoryManager
pub fn new(config: &GpuAccelerationConfig) -> Self
pub fn allocate_gpu_memory(&mut self, size: usize) -> Result<()>
pub fn transfer_to_gpu( &mut self, data: &ArrayView2<'_, f64>, ) -> Result<GpuTensor>
pub fn transfer_to_cpu(&self, gputensor: &GpuTensor) -> Result<Array2<f64>>
pub fn get_memory_stats(&self) -> GpuMemoryStats
pub fn get_peak_memory_usage(&self) -> f64
Trait Implementations§
Auto Trait Implementations§
impl Freeze for GpuMemoryManager
impl RefUnwindSafe for GpuMemoryManager
impl Send for GpuMemoryManager
impl Sync for GpuMemoryManager
impl Unpin for GpuMemoryManager
impl UnwindSafe for GpuMemoryManager
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
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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.