Function default_cdi_gpu_device_id

e2e/rust/tests/gpu/device_selection.rs:205–236

(device_ids: &[String], allow_all_devices: bool)

}

fn default_cdi_gpu_device_id(device_ids: &[String], allow_all_devices: bool) -> Option<String> {
    let mut indexed = device_ids
        .iter()
        .filter_map(|device_id| {
            let suffix = device_id.strip_prefix(CDI_GPU_DEVICE_PREFIX)?;
            let index = suffix.parse::<u64>().ok()?;
            Some((index, device_id.clone()))
        })
        .collect::<Vec<_>>();
    if !indexed.is_empty() {
        indexed.sort_by(|left, right| left.0.cmp(&right.0).then_with(|| left.1.cmp(&right.1)));
        return indexed.into_iter().map(|(_, device_id)| device_id).next();
    }

    let mut named = device_ids
        .iter()
        .filter(|device_id| {
            device_id.starts_with(CDI_GPU_DEVICE_PREFIX) && device_id.as_str() != CDI_GPU_DEVICE_ALL
        })
        .cloned()
        .collect::<Vec<_>>();
    if !named.is_empty() {
        named.sort();
        return named.into_iter().next();
    }

    (allow_all_devices
        && device_ids
            .iter()
            .any(|device_id| device_id == CDI_GPU_DEVICE_ALL))
    .then(|| CDI_GPU_DEVICE_ALL.to_string())
}

fn has_cdi_gpu_device(device_id: &str) -> bool {
    discovered_cdi_gpu_device_ids()