Trait GpuBackendTrait

Source
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>;
}
Expand description

Trait for GPU backend implementations

Required Methods§

Implementors§