Struct burn::backend::wgpu::WgpuServer

pub struct WgpuServer<C>
where
    C: WgpuCompiler,
{ /* private fields */ }

Wgpu compute server.

Implementations

impl<C> WgpuServer<C>

where C: WgpuCompiler,

pub fn new( memory_management: MemoryManagement<WgpuStorage>, device: Arc<Device>, queue: Arc<Queue>, tasks_max: usize, ) -> WgpuServer<C>

Create a new server.

Trait Implementations

impl<C> ComputeServer for WgpuServer<C>

where C: WgpuCompiler,

fn create(&mut self, data: &[u8]) -> Handle

When we create a new handle from existing data, we use custom allocations so that we don’t have to execute the current pending tasks.

This is important, otherwise the compute passes are going to be too small and we won’t be able to fully utilize the GPU.

fn sync(&mut self) -> impl Future<Output = ()> + 'static

Returns the total time of GPU work this sync completes.

fn sync_elapsed( &mut self, ) -> impl Future<Output = Result<Duration, TimestampsError>> + 'static

Returns the total time of GPU work this sync completes.

type Kernel = Box<dyn CubeTask<C>>

The kernel type defines the computation algorithms.

type Storage = WgpuStorage

The storage type defines how data is stored and accessed.

type Feature = Feature

The type of the features supported by the server.

fn read( &mut self, binding: Binding, ) -> impl Future<Output = Vec<u8>> + Send + 'static

Given a handle, returns the owned resource as bytes.

fn get_resource(&mut self, binding: Binding) -> BindingResource<WgpuServer<C>>

Given a resource handle, returns the storage resource.

fn empty(&mut self, size: usize) -> Handle

Reserves size bytes in the storage, and returns a handle over them.

unsafe fn execute( &mut self, kernel: <WgpuServer<C> as ComputeServer>::Kernel, count: CubeCount, bindings: Vec<Binding>, mode: ExecutionMode, )

Executes the kernel over the given memory handles.

fn flush(&mut self)

Flush all outstanding tasks in the server.

fn memory_usage(&self) -> MemoryUsage

The current memory usage of the server.

fn enable_timestamps(&mut self)

Enable collecting timestamps.

fn disable_timestamps(&mut self)

Disable collecting timestamps.

impl<C> Debug for WgpuServer<C>

where C: Debug + WgpuCompiler,

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

Formats the value using the given formatter.