Enum HostBuffer 

pub enum HostBuffer {
    U8(Vec<u8>),
    I8(Vec<i8>),
    U32(Vec<u32>),
    I16(Vec<i16>),
    I32(Vec<i32>),
    I64(Vec<i64>),
    BF16(Vec<bf16>),
    F16(Vec<f16>),
    F32(Vec<f32>),
    F64(Vec<f64>),
    F8E4M3(Vec<F8E4M3>),
    F6E2M3(Vec<u8>),
    F6E3M2(Vec<u8>),
    F4(Vec<u8>),
    F8E8M0(Vec<u8>),
    Bool(Vec<u8>),
}

Host-addressable tensor storage holding a typed vector of elements.

This is the universal interchange format for moving tensor data between backends: every BackendStorage can produce one via to_host_buffer(), and every BackendDevice can ingest one via storage_from_host_buffer().

Previously named CpuStorage — that name remains as a type alias.

Variants

U8(Vec<u8>)

I8(Vec<i8>)

U32(Vec<u32>)

I16(Vec<i16>)

I32(Vec<i32>)

I64(Vec<i64>)

BF16(Vec<bf16>)

F16(Vec<f16>)

F32(Vec<f32>)

F64(Vec<f64>)

F8E4M3(Vec<F8E4M3>)

F6E2M3(Vec<u8>)

F6E3M2(Vec<u8>)

F4(Vec<u8>)

F8E8M0(Vec<u8>)

Bool(Vec<u8>)

Boolean mask storage (GAP-168(c)). One byte per element (0/1), byte-identical to U8 — but a DISTINCT variant so a Bool buffer can never be silently consumed as U8. The comparison kernels write 0/1 bytes here; where_cond/masked_fill read it as a byte mask.

Implementations

impl HostBuffer

pub fn dtype(&self) -> DType

Returns the DType of the elements stored.

pub fn as_slice<D>(&self) -> Result<&[D], Error>

where D: HostDType,

Returns a typed slice of the stored data.

pub fn concat(storages: &[HostBuffer]) -> Result<HostBuffer, Error>

Concatenates multiple storages of the same dtype into one.

pub fn as_ref(&self) -> HostBufferRef<'_>

Borrow this buffer as a HostBufferRef (zero-copy).

The ref form is the common shape for HostStorage::as_host_buffer_ref; owned-buffer implementors delegate through here.

Trait Implementations

impl Clone for HostBuffer

fn clone(&self) -> HostBuffer

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for HostBuffer

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

Formats the value using the given formatter.

impl From<CpuStorage> for HostBuffer

fn from(s: CpuStorage) -> HostBuffer

Converts to this type from the input type.