(name: &str, tflops: f32, vram_gb: f64)
| 315 | use std::path::Path; |
| 316 | |
| 317 | fn make_worker(name: &str, tflops: f32, vram_gb: f64) -> DiscoveredWorker { |
| 318 | DiscoveredWorker { |
| 319 | name: name.to_string(), |
| 320 | host: "127.0.0.1".to_string(), |
| 321 | port: 10128, |
| 322 | gpus: vec![GpuInfo { |
| 323 | name: "NVIDIA RTX Test".to_string(), |
| 324 | vram_bytes: (vram_gb * 1024.0 * 1024.0 * 1024.0) as u64, |
| 325 | tflops, |
| 326 | }], |
| 327 | backend: "cuda".to_string(), |
| 328 | hostname: name.to_string(), |
| 329 | os: "linux".to_string(), |
| 330 | } |
| 331 | } |
| 332 | |
| 333 | // Helper to call DefaultStrategy with concrete workers |
| 334 | fn compute_layer_assignments<W: WorkerCapacity>( |
nothing calls this directly
no outgoing calls
no test coverage detected