MCPcopy Create free account
hub / github.com/evilsocket/cake / compute_layer_assignments

Function compute_layer_assignments

cake-core/src/cake/sharding/default.rs:334–344  ·  view source on GitHub ↗

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

Source from the content-addressed store, hash-verified

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

Calls 1

assign_layersMethod · 0.80