Struct burn::backend::wgpu::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>) -> WgpuStorage

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.

const ALIGNMENT: u64 = 32u64

The alignment memory is allocated with in this storage.

fn get( &mut self, handle: &StorageHandle, ) -> <WgpuStorage as ComputeStorage>::Resource

Returns the underlying resource for a specified storage handle

fn alloc(&mut self, size: u64) -> 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<(), Error>

Formats the value using the given formatter.