Struct GpuMemoryBudget 

pub struct GpuMemoryBudget {
    pub limits: GpuMemoryLimits,
    pub allocated_bytes: u64,
    pub storage_bindings_used: u32,
}

Tracks GPU buffer allocations and enforces limits.

Fields

limits: GpuMemoryLimits

allocated_bytes: u64

Total bytes currently allocated across all buffers.

storage_bindings_used: u32

Number of storage buffer bindings in use.

Implementations

impl GpuMemoryBudget

pub fn new(limits: GpuMemoryLimits) -> Self

pub fn webgpu_default() -> Self

WebGPU default limits.

pub fn check_buffer(&self, size_bytes: u64) -> Result<(), MemoryError>

Check if a buffer allocation would exceed limits.

pub fn check_storage_binding(&self) -> Result<(), MemoryError>

Check if a storage binding can be added.

pub fn check_workgroup( &self, size: [u32; 3], count: [u32; 3], ) -> Result<(), MemoryError>

Check workgroup configuration validity.

pub fn estimate_orbital_grid_memory( &self, n_basis: usize, grid_points: usize, max_primitives_per_basis: usize, ) -> Result<OrbitalGridMemoryEstimate, MemoryError>

Pre-flight check for an orbital grid computation.

Returns the estimated total GPU memory needed in bytes.

pub fn estimate_d4_dispersion_memory( &self, n_atoms: usize, ) -> Result<PairwiseMemoryEstimate, MemoryError>

Estimate memory for D4 dispersion pairwise computation.

pub fn optimal_grid_dispatch(&self, dims: [u32; 3]) -> ([u32; 3], [u32; 3])

Compute optimal workgroup dispatch for a 3D grid.

pub fn optimal_1d_dispatch(&self, n: u32) -> (u32, u32)

Compute optimal workgroup dispatch for a 1D array.

Trait Implementations

impl Clone for GpuMemoryBudget

fn clone(&self) -> GpuMemoryBudget

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GpuMemoryBudget

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.