fn get_gpu_memory()

in src/main.rs [523:527]

• 5 lines of code
• 1 McCabe index (conditional complexity)

fn get_gpu_memory(gpu: Arc<CudaDevice>) -> anyhow::Result<(usize, usize)> {
    CudaDevice::new(gpu.ordinal())?;
    let mem_info = mem_get_info()?;
    Ok(mem_info)
}