Struct WgpuServer 

pub struct WgpuServer {
    pub compilation_options: WgpuCompilationOptions,
    /* private fields */
}

Wgpu compute server.

Fields

compilation_options: WgpuCompilationOptions

Implementations

impl WgpuServer

pub fn create_pipeline( &mut self, kernel: CompiledKernel<AutoCompiler>, mode: ExecutionMode, ) -> Arc<ComputePipeline>

impl WgpuServer

pub fn new( memory_properties: MemoryDeviceProperties, memory_config: MemoryConfiguration, compilation_options: WgpuCompilationOptions, device: Device, queue: Queue, tasks_max: usize, backend: Backend, timing_method: TimingMethod, utilities: ServerUtilities<Self>, ) -> Self

Create a new server.

Trait Implementations

impl ComputeServer for WgpuServer

fn sync(&mut self, stream_id: StreamId) -> DynFut<()>

Returns the total time of GPU work this sync completes.

type Kernel = Box<dyn CubeTask<AutoCompiler>>

The kernel type defines the computation algorithms.

type Storage = WgpuStorage

The storage type defines how data is stored and accessed.

type Info = Backend

Information that can be retrieved for the runtime.

fn logger(&self) -> Arc<ServerLogger>

Retrieve the server logger.

fn utilities(&self) -> Arc<ServerUtilities<Self>>

Retrieve the server utilities.

fn create( &mut self, descriptors: Vec<AllocationDescriptor<'_>>, stream_id: StreamId, ) -> Result<Vec<Allocation>, IoError>

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

fn read<'a>( &mut self, descriptors: Vec<CopyDescriptor<'a>>, stream_id: StreamId, ) -> DynFut<Result<Vec<Bytes>, IoError>>

Given bindings, returns the owned resources as bytes.

fn write( &mut self, descriptors: Vec<(CopyDescriptor<'_>, &[u8])>, stream_id: StreamId, ) -> Result<(), IoError>

Writes the specified bytes into the buffers given

fn get_resource( &mut self, binding: Binding, stream_id: StreamId, ) -> BindingResource<WgpuResource>

Given a resource handle, returns the storage resource.

unsafe fn execute( &mut self, kernel: Self::Kernel, count: CubeCount, bindings: Bindings, mode: ExecutionMode, stream_id: StreamId, )

Executes the kernel over the given memory handles.

fn flush(&mut self, stream_id: StreamId)

Flush all outstanding tasks in the server.

fn start_profile(&mut self, stream_id: StreamId) -> ProfilingToken

Enable collecting timestamps.

fn end_profile( &mut self, stream_id: StreamId, token: ProfilingToken, ) -> Result<ProfileDuration, ProfileError>

Disable collecting timestamps.

fn memory_usage(&mut self, stream_id: StreamId) -> MemoryUsage

The current memory usage of the server.

fn memory_cleanup(&mut self, stream_id: StreamId)

Ask the server to release memory that it can release.

fn allocation_mode(&mut self, mode: MemoryAllocationMode, stream_id: StreamId)

Update the memory mode of allocation in the server.

fn create_with_data( &mut self, data: &[u8], stream_id: StreamId, ) -> Result<Handle, IoError>

Utility to create a new buffer and immediately copy contiguous data into it

impl Debug for WgpuServer

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

Formats the value using the given formatter.

impl DeviceState for WgpuServer

fn init(device_id: DeviceId) -> Self

Initialize a new state on the given device.

impl ServerCommunication for WgpuServer

const SERVER_COMM_ENABLED: bool = false

Indicates whether server-to-server communication is enabled for this implementation.

fn copy( server_src: &mut Self, server_dst: &mut Self, src: CopyDescriptor<'_>, stream_id_src: StreamId, stream_id_dst: StreamId, ) -> Result<Allocation, IoError>

Copies data from a source server to a destination server.