Struct burn_wgpu::compute::WgpuStorage

pub struct WgpuStorage { /* private fields */ }

Buffer storage for wgpu.

Implementations

impl WgpuStorage

Keeps actual wgpu buffer references in a hashmap with ids as key.

pub fn new(device: Arc<Device>) -> Self

Create a new storage on the given device.

pub fn perform_deallocations(&mut self)

Actually deallocates buffers tagged to be deallocated.

Trait Implementations

impl ComputeStorage for WgpuStorage

type Resource = WgpuResource

The resource associated type determines the way data is implemented and how it can be accessed by kernels.

fn get(&mut self, handle: &StorageHandle) -> Self::Resource

Returns the underlying resource for a specified storage handle

fn alloc(&mut self, size: usize) -> StorageHandle

Allocates size units of memory and returns a handle to it

fn dealloc(&mut self, id: StorageId)

Deallocates the memory pointed by the given storage id.

impl Debug for WgpuStorage

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

Formats the value using the given formatter.