flodl 0.7.0

floDl — a flow-graph deep learning framework built on libtorch
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

//! DDP run-mode orchestrator: spawns GPU worker threads and a coordinator thread.

mod rank_entry;
mod coord_config;
mod single_host;
mod handle;
mod builder;

pub use handle::DdpHandle;
pub use builder::DdpBuilder;
/// Crate-wide export: every force-exiting cluster role (launcher, relay,
/// agent, rank death, cooperative Worker drop) must terminate through this
/// instead of `std::process::exit` — see its doc for the libtorch
/// static-teardown GP-fault it avoids.
pub(crate) use handle::clean_process_exit;

#[cfg(test)]
pub(super) use coord_config::build_coord_config_from_builder;