Function effective_driver_gpu_count

crates/openshell-core/src/gpu.rs:40–51

Return the effective compute-driver GPU count. A present GPU requirement with an omitted count requests one GPU. # Errors Returns an error when a GPU requirement explicitly requests zero GPUs.

(
    gpu: Option<&DriverGpuResourceRequirements>,
)

/// # Errors
/// Returns an error when a GPU requirement explicitly requests zero GPUs.
pub fn effective_driver_gpu_count(
    gpu: Option<&DriverGpuResourceRequirements>,
) -> Result<Option<u32>, String> {
    let Some(gpu) = gpu else {
        return Ok(None);
    };
    let count = gpu.count.unwrap_or(1);
    if count == 0 {
        return Err("gpu count must be greater than 0".to_string());
    }
    Ok(Some(count))
}

/// Return the requested compute-driver GPU requirements, if present.
#[must_use]