Enum RawBuffer 

pub enum RawBuffer {
    Cpu {
        data: UnsafeCell<AlignedBuffer>,
        cpu_accessible: bool,
    },
    Mmap {
        data: Mmap,
        size: usize,
    },
}

Opaque handle to device memory.

Safety

RawBuffer uses UnsafeCell for interior mutability without locking overhead. Thread safety is guaranteed at a higher level by the scheduler:

1. Allocation: OnceLock in BufferData ensures single initialization
2. Buffer Access: The scheduler guarantees exclusive access to each buffer during kernel execution - no two kernels access the same buffer concurrently
3. Kernel Execution: Raw pointers passed to JIT code; Rust doesn’t access buffer data during execution

This design follows Tinygrad’s approach where buffer synchronization is the scheduler’s responsibility, not the buffer’s.

Variants

Cpu

Fields

data: UnsafeCell<AlignedBuffer>

cpu_accessible: bool

Mmap

Memory-mapped file region (read-only). Used by DISK device.

Fields

data: Mmap

size: usize

Implementations

impl RawBuffer

pub fn size(&self) -> usize

Get the size of the buffer in bytes.

pub fn cpu_accessible(&self) -> bool

Get whether this buffer is CPU-accessible.

Trait Implementations

impl Debug for RawBuffer

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

Formats the value using the given formatter.

impl Send for RawBuffer

impl Sync for RawBuffer