Module gpu_acceleration

Module gpu_acceleration 

Source
Expand description

GPU acceleration and optimization features for embedding models

This module provides advanced GPU acceleration capabilities including memory pooling, tensor caching, mixed precision, and compute optimization with full SciRS2 integration for maximum performance.

Structs§

AcceleratorStats
Statistics for GPU accelerator
AllocationStats
Memory allocation statistics
BatchSizeOptimizer
Batch size optimizer for maximizing GPU utilization
BatchSizeOptimizerStats
Statistics from batch size optimization
CacheStats
Cache statistics
DefragmentationResult
Results of memory defragmentation operation
DynamicShapeHandler
Dynamic shape handling for variable-size inputs
GpuAccelerationConfig
GPU acceleration configuration
GpuAccelerationManager
Main GPU acceleration manager
GpuMemoryPool
GPU memory pool for efficient memory management
GpuPerformanceStats
GPU performance statistics
MemoryDefragmenter
Memory defragmentation utilities
MixedPrecisionProcessor
Mixed precision training and inference
MultiStreamProcessor
Multi-stream processor for parallel GPU operations
OutOfCoreProcessor
Out-of-core processing for handling datasets larger than GPU memory
SciRS2GpuAccelerator
Advanced GPU accelerator using SciRS2’s full GPU capabilities
TensorCache
Tensor cache for frequently used tensors