Function create_backend

cake-core/src/backends/mod.rs:688–715

Create the appropriate backend for the given device.

(device: &Device)

/// Create the appropriate backend for the given device.
pub fn create_backend(device: &Device) -> Arc<dyn ComputeBackend> {
    match device {
        #[cfg(feature = "cuda")]
        Device::Cuda(_) => Arc::new(CudaBackend::new(device.clone())),
        #[cfg(feature = "metal")]
        Device::Metal(_) => Arc::new(MetalBackend::new(device.clone())),
        _ => {
            // No GPU device — try ROCm first (native AMD), then Vulkan (wgpu)
            #[cfg(feature = "rocm")]
            match RocmBackend::new() {
                Ok(r) => {
                    log::info!("using ROCm backend (rocBLAS GEMM)");
                    return Arc::new(r);
                }
                Err(e) => log::warn!("ROCm init failed ({e}), trying next backend"),
            }
            #[cfg(feature = "vulkan")]
            match VulkanBackend::new() {
                Ok(vk) => {
                    log::info!("using Vulkan backend for GPU-accelerated ops");
                    return Arc::new(vk);
                }
                Err(e) => log::warn!("Vulkan init failed ({e}), falling back to CPU"),
            }
            Arc::new(CpuBackend::new())
        }
    }
}