Module gpu

Module gpu 

Source
Expand description

GPU Acceleration for Embedding Computations

This module provides GPU-accelerated implementations of embedding operations using scirs2-linalg GPU features for CUDA, OpenCL, ROCm, and Metal backends.

Structs§

AdaptiveBatchConfig
Adaptive batch sizing configuration
AdaptiveEmbeddingAccelerator
Adaptive acceleration that chooses between GPU and CPU based on problem size
GpuEmbeddingAccelerator
Enhanced GPU-accelerated embedding computations with memory pooling and adaptive batching
GpuError
GpuMemoryPool
Memory pool for GPU buffers
GpuPerformanceReport
Comprehensive GPU performance report
GpuPerformanceStats
GPU performance statistics

Type Aliases§

GpuArray
GpuContext