Struct MemoryPinningManager 

pub struct MemoryPinningManager { /* private fields */ }

Memory pinning manager that selects appropriate pinner based on device

This manager provides a unified interface for memory pinning across different device types, automatically selecting the appropriate pinner implementation based on the target device.

Examples

use torsh_data::dataloader::memory::MemoryPinningManager;
use torsh_core::device::DeviceType;

let mut manager = MemoryPinningManager::new();

// Check if pinning is supported for CUDA
let cuda_supported = manager.supports_pinning(Some(DeviceType::Cuda(0)));

Implementations

impl MemoryPinningManager

pub fn new() -> Self

Create a new memory pinning manager

Returns

A new MemoryPinningManager with default settings

pub fn pin_memory<T>( &mut self, data: Tensor<T>, target_device: Option<DeviceType>, ) -> Result<Tensor<T>>

where T: TensorElement,

Pin memory for the appropriate device

This method automatically selects the appropriate pinning implementation based on the target device type.

Arguments

• data - The tensor data to pin
• target_device - The target device for the data

Returns

The tensor with memory pinning applied if supported

Examples

use torsh_data::dataloader::memory::MemoryPinningManager;
use torsh_core::device::DeviceType;
use torsh_tensor::Tensor;

let mut manager = MemoryPinningManager::new();
let tensor = Tensor::zeros(&[2, 3]);
let pinned = manager.pin_memory(tensor, Some(DeviceType::Cpu))?;

pub fn pin_vector_memory<T>( &mut self, data: Vec<T>, target_device: Option<DeviceType>, ) -> Result<Vec<T>>

Pin memory for vector data

A convenience method for pinning vector data that doesn’t require tensor operations.

Arguments

• data - The vector data to pin
• target_device - The target device for the data

Returns

The vector with memory pinning applied if supported

pub fn supports_pinning(&self, target_device: Option<DeviceType>) -> bool

Check if pinning is supported for the target device

Arguments

• target_device - The device to check pinning support for

Returns

True if pinning is supported for the target device, false otherwise

pub fn available_pinners(&self) -> String

Get information about available pinning implementations

Returns

String describing available pinning capabilities

Trait Implementations

impl Debug for MemoryPinningManager

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

Formats the value using the given formatter.

impl Default for MemoryPinningManager

fn default() -> Self

Returns the “default value” for a type.