Expand description
GPU acceleration support for SIMD operations
This module provides CUDA and OpenCL kernel interfaces for GPU-accelerated machine learning operations with fallback to CPU SIMD implementations.
Modules§
Structs§
- GpuBuffer
- GPU memory buffer wrapper
- GpuContext
- GPU context for managing resources
- GpuDevice
- GPU device information
- GpuManager
- GPU manager for handling multiple devices and backends
- GpuStream
- GPU stream for asynchronous operations
- Kernel
Config - GPU kernel launch parameters
Enums§
- GpuBackend
- GPU computation backends
Statics§
Traits§
- GpuOperations
- GPU operations interface
Functions§
- get_
gpu_ devices - Get available GPU devices
- initialize_
gpu - Initialize GPU support
- is_
gpu_ available - Check if GPU acceleration is available