Struct SharedGpuContext 

pub struct SharedGpuContext { /* private fields */ }

Shared GPU context for efficient resource management across all crates Now supports both single-GPU (legacy) and multi-GPU operations

Implementations

impl SharedGpuContext

pub async fn global() -> UnifiedGpuResult<&'static Self>

Get the global shared GPU context (singleton pattern) Note: This creates a new context each time for now. In production, this would be a proper singleton with atomic initialization.

pub async fn with_multi_gpu() -> UnifiedGpuResult<Self>

Create a new shared GPU context with multi-GPU support

pub fn device(&self) -> &Device

Get the device

pub fn queue(&self) -> &Queue

Get the queue

pub fn adapter_info(&self) -> &AdapterInfo

Get adapter info

pub fn get_buffer( &self, size: u64, usage: BufferUsages, label: Option<&str>, ) -> Buffer

Get or create a buffer from the pool

pub fn return_buffer(&self, buffer: Buffer, size: u64, usage: BufferUsages)

Return a buffer to the pool for reuse

pub fn get_compute_pipeline( &self, shader_key: &str, shader_source: &str, entry_point: &str, ) -> UnifiedGpuResult<Arc<ComputePipeline>>

Get or create a compute pipeline from cache

pub fn buffer_pool_stats(&self) -> BufferPoolStats

Get buffer pool statistics

pub fn uptime(&self) -> Duration

Get uptime of this context

pub fn get_optimal_workgroup( &self, operation: &str, data_size: usize, ) -> (u32, u32, u32)

Get optimal workgroup configuration for given operation type and data size

pub fn get_workgroup_declaration( &self, operation: &str, data_size: usize, ) -> String

Generate optimized WGSL workgroup declaration for operation

pub fn is_multi_gpu_enabled(&self) -> bool

Check if multi-GPU mode is enabled

pub async fn device_count(&self) -> usize

Get the number of available GPU devices

pub async fn get_device_info(&self) -> Vec<(DeviceId, String, String)>

Get information about all GPU devices

pub async fn get_device(&self, device_id: DeviceId) -> Option<Arc<GpuDevice>>

Get a specific GPU device by ID

pub async fn optimal_device_for_operation( &self, operation: &str, _data_size: usize, ) -> DeviceId

Get the optimal device for a specific operation

pub async fn distribute_workload( &self, workload: Workload, ) -> UnifiedGpuResult<Vec<DeviceWorkload>>

Distribute a workload across multiple GPUs

pub async fn execute_multi_gpu_workload( &self, workload_id: String, workload: Workload, ) -> UnifiedGpuResult<Vec<Vec<u8>>>

Execute a workload on multiple GPUs and aggregate results

pub async fn get_gpu_utilization(&self) -> HashMap<DeviceId, f32>

Get real-time GPU utilization across all devices

pub async fn get_multi_gpu_stats(&self) -> MultiGpuStats

Get performance statistics for multi-GPU operations

pub async fn set_load_balancing_strategy( &self, _strategy: LoadBalancingStrategy, ) -> UnifiedGpuResult<()>

Set load balancing strategy for multi-GPU operations

pub async fn add_gpu_device( &self, device: Arc<GpuDevice>, ) -> UnifiedGpuResult<()>

Add a new GPU device to the multi-GPU context (hot-plugging support)

pub async fn remove_gpu_device( &self, device_id: DeviceId, ) -> UnifiedGpuResult<()>

Remove a GPU device from the multi-GPU context

Trait Implementations

impl Clone for SharedGpuContext

fn clone(&self) -> SharedGpuContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.