Struct GpuContext 

pub struct GpuContext { /* private fields */ }

GPU context for managing device operations

Implementations

impl GpuContext

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

Create a new GPU context

pub fn device_info(&self) -> &GpuDeviceInfo

Get device information

pub fn backend(&self) -> &GpuBackend

Get the backend type

pub fn is_available(&self) -> bool

Check if GPU is available and functional

pub fn make_classification_gpu( &self, n_samples: usize, n_features: usize, n_classes: usize, n_clusters_per_class: usize, n_informative: usize, random_state: Option<u64>, ) -> Result<Dataset>

Generate classification dataset on GPU

pub fn make_regression_gpu( &self, n_samples: usize, n_features: usize, n_informative: usize, noise: f64, random_state: Option<u64>, ) -> Result<Dataset>

Generate regression dataset on GPU

pub fn make_blobs_gpu( &self, n_samples: usize, n_features: usize, n_centers: usize, cluster_std: f64, random_state: Option<u64>, ) -> Result<Dataset>

Generate clustering dataset (blobs) on GPU

pub fn gpu_matrix_multiply( &self, a: &Array2<f64>, b: &Array2<f64>, ) -> Result<Array2<f64>>

Perform matrix operations on GPU

pub fn gpu_elementwise_op<F>( &self, data: &Array2<f64>, op: F, ) -> Result<Array2<f64>>

where F: Fn(f64) -> f64 + Send + Sync,

Apply element-wise operations on GPU