Struct GpuOptimizer 

pub struct GpuOptimizer<O, A>
where
    O: Optimizer<A, Ix1>,
    A: Float + ScalarOperand + Debug,
{ /* private fields */ }

GPU-accelerated optimizer wrapper

Wraps any CPU optimizer to provide GPU acceleration using SciRS2’s GPU abstractions. Automatically handles host-device data transfer and GPU memory management.

Examples

use optirs_core::optimizers::SGD;
use optirs_core::gpu_optimizer::{GpuOptimizer, GpuConfig};
use scirs2_core::ndarray::Array1;

let optimizer = SGD::new(0.01);
let config = GpuConfig::default();

// Create GPU-accelerated optimizer
let mut gpu_opt = GpuOptimizer::new(optimizer, config)?;

// Use like a normal optimizer - GPU acceleration is automatic
let params = Array1::from_vec(vec![1.0, 2.0, 3.0]);
let grads = Array1::from_vec(vec![0.1, 0.2, 0.3]);

let updated = gpu_opt.step(&params, &grads)?;

Implementations

impl<O, A> GpuOptimizer<O, A>

where O: Optimizer<A, Ix1> + Clone, A: Float + ScalarOperand + Debug,

pub fn new(base_optimizer: O, config: GpuConfig) -> Result<Self>

Creates a new GPU-accelerated optimizer

Arguments

• base_optimizer - The CPU optimizer to accelerate
• config - GPU configuration settings

Returns

A GPU-accelerated optimizer or an error if GPU initialization fails

pub fn with_default_config(base_optimizer: O) -> Result<Self>

Creates a new GPU optimizer with default configuration

pub fn step( &mut self, params: &Array1<A>, gradients: &Array1<A>, ) -> Result<Array1<A>>

Perform GPU-accelerated optimization step

Arguments

• params - Current parameters
• gradients - Gradients

Returns

Updated parameters after GPU-accelerated optimization

pub fn to_gpu(&self, _data: &ArrayView1<'_, A>) -> Result<()>

Transfer array to GPU

Note: Full implementation would use scirs2_core::gpu::GpuBuffer

pub fn from_gpu(&self) -> Result<Array1<A>>

Transfer array from GPU

Note: Full implementation would use scirs2_core::gpu::GpuBuffer

pub fn is_gpu_available(&self) -> bool

Check if GPU is available and initialized

pub fn gpu_backend(&self) -> Option<&str>

Get GPU backend name

pub fn config(&self) -> &GpuConfig

Get GPU configuration

pub fn set_use_tensor_cores(&mut self, enable: bool)

Enable/disable tensor core acceleration

pub fn set_use_mixed_precision(&mut self, enable: bool)

Enable/disable mixed-precision training

pub fn estimate_gpu_memory( num_params: usize, dtype_size: usize, optimizer_states: usize, ) -> usize

Get estimated GPU memory usage for given parameter count