Struct CpuBuffer 

pub struct CpuBuffer<T> { /* private fields */ }

CPU buffer implementation using Vec

Implementations

impl<T: Clone + Send + Sync> CpuBuffer<T>

pub fn new(data: Vec<T>) -> Self

pub fn zeros(len: usize) -> Self

where T: Default,

Trait Implementations

impl<T: Clone + Send + Sync + 'static> TensorBuffer for CpuBuffer<T>

type Elem = T

fn device(&self) -> &Device

Get the device where this buffer is allocated

fn len(&self) -> usize

Get the number of elements in the buffer

fn size_bytes(&self) -> usize

Get the size in bytes

fn clone_buffer(&self) -> Result<Box<dyn TensorBuffer<Elem = Self::Elem>>>

Clone the buffer (may involve device memory copy)

fn view( &self, offset: usize, len: usize, ) -> Result<Box<dyn TensorBuffer<Elem = Self::Elem>>>

Create a view into this buffer (zero-copy when possible)

fn to_cpu(&self) -> Result<Vec<Self::Elem>>

Convert to a CPU buffer for host operations

unsafe fn as_ptr(&self) -> *const Self::Elem

Get raw pointer (unsafe - for FFI and low-level ops)

unsafe fn as_mut_ptr(&mut self) -> *mut Self::Elem

Get mutable raw pointer (unsafe - for FFI and low-level ops)

fn is_empty(&self) -> bool

Check if the buffer is empty