Trait GpuBackendTrait

pub trait GpuBackendTrait: Send + Sync {
    // Required methods
    fn initialize(&self) -> NeuralResult<()>;
    fn is_available(&self) -> bool;
    fn get_device_info(&self) -> DeviceInfo;
    fn create_buffer(&self, size: usize) -> NeuralResult<BufferHandle>;
    fn execute_kernel(
        &self,
        kernel: &CompiledKernel,
        inputs: &[BufferHandle],
    ) -> NeuralResult<BufferHandle>;
}

Trait for GPU backend implementations

Required Methods

fn initialize(&self) -> NeuralResult<()>

fn is_available(&self) -> bool

fn get_device_info(&self) -> DeviceInfo

fn create_buffer(&self, size: usize) -> NeuralResult<BufferHandle>

fn execute_kernel( &self, kernel: &CompiledKernel, inputs: &[BufferHandle], ) -> NeuralResult<BufferHandle>

Implementors

impl GpuBackendTrait for WebGpuBackend