Module backends 

GPU Backend Abstraction Layer

This module provides a unified interface for different GPU backends, supporting CUDA, ROCm, Metal, WebGPU, and CPU fallback.

Structs

BackendFactory

GPU backend factory

CompiledKernel

Compiled kernel representation

CpuBackend

CPU backend (fallback implementation)

CudaBackend

CUDA backend implementation

DeviceCapabilities

GPU device capabilities

DeviceMemory

Device memory handle

MetalBackend

Metal backend implementation

RocmBackend

ROCm backend implementation

WgpuBackend

WebGPU backend implementation

Enums

BackendError

Errors that can occur with GPU backends

GpuBackend

GPU backend types supported by the optimizer

KernelArg

Kernel argument types

Traits

Backend

Trait for GPU backend implementations