Function resolve_gpus

vmm/src/main_service.rs:86–132

(gpu_cfg: &rpc::GpuConfig)

}

pub fn resolve_gpus(gpu_cfg: &rpc::GpuConfig) -> Result<GpuConfig> {
    // Check the attach mode to determine how to handle GPUs
    match gpu_cfg.attach_mode.as_str() {
        "listed" => {
            // If the mode is "listed", use the GPUs specified in the request
            let gpus = gpu_cfg
                .gpus
                .iter()
                .map(|g| GpuSpec {
                    slot: g.slot.clone(),
                })
                .collect();

            Ok(GpuConfig {
                attach_mode: AttachMode::Listed,
                gpus,
                bridges: Vec::new(),
            })
        }
        "all" => {
            // If the mode is "all", find all NVIDIA GPUs and NVSwitches
            let devices = lspci::lspci_filtered(|dev| {
                // Check if it's an NVIDIA device (vendor ID 10de)
                dev.vendor_id == "10de"
            })
            .context("Failed to list PCI devices")?;

            let mut gpus = Vec::new();
            let mut bridges = Vec::new();

            for dev in devices {
                // Check if it's a GPU (3D controller) or NVSwitch (Bridge)
                if dev.class.contains("3D controller") {
                    gpus.push(GpuSpec { slot: dev.slot });
                } else if dev.class.contains("Bridge") {
                    bridges.push(GpuSpec { slot: dev.slot });
                }
            }
            Ok(GpuConfig {
                attach_mode: AttachMode::All,
                gpus,
                bridges,
            })
        }
        _ => bail!("Invalid GPU attach mode: {}", gpu_cfg.attach_mode),
    }
}

// Shared function to create manifest from VM configuration
pub fn create_manifest_from_vm_config(