fn detect_gpus()

in src/main.rs [572:580]


fn detect_gpus() -> anyhow::Result<Vec<Arc<CudaDevice>>> {
    let num_gpus = CudaDevice::count()? as usize;
    let mut devices = Vec::new();
    for i in 0..num_gpus {
        let dev = CudaDevice::new(i)?;
        devices.push(dev);
    }
    Ok(devices)
}