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)
}