Helper to call DefaultStrategy with concrete workers
(
workers: &[W],
num_layers: usize,
master_tflops: f64,
layer_size_bytes: u64,
master_max_layers: usize,
layer_prefix: &str,
)
| 332 | |
| 333 | // Helper to call DefaultStrategy with concrete workers |
| 334 | fn compute_layer_assignments<W: WorkerCapacity>( |
| 335 | workers: &[W], |
| 336 | num_layers: usize, |
| 337 | master_tflops: f64, |
| 338 | layer_size_bytes: u64, |
| 339 | master_max_layers: usize, |
| 340 | layer_prefix: &str, |
| 341 | ) -> Vec<(usize, Vec<String>)> { |
| 342 | let dyn_workers: Vec<&dyn WorkerCapacity> = workers.iter().map(|w| w as &dyn WorkerCapacity).collect(); |
| 343 | DefaultStrategy.assign_layers(&dyn_workers, num_layers, master_tflops, layer_size_bytes, master_max_layers, layer_prefix) |
| 344 | } |
| 345 | |
| 346 | // ── layer_prefix_for_config tests ────────────────────── |
| 347 |