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