1pub mod accelerator;
32pub mod autotune;
33pub mod cluster;
34pub mod dashboard;
35pub mod discovery;
36pub mod health;
37pub mod host;
38pub mod kv_cache;
39pub mod load_balance;
40pub mod mdns;
41pub mod models;
42pub mod planning;
43pub mod protocol;
44pub mod ring;
45pub mod runtime;
46pub mod system_profile;
47pub mod watcher;
48#[doc(hidden)]
49pub mod xdp;
50
51pub use accelerator::ExecutionBackend;
53pub use autotune::AutoTuner;
54pub use cluster::{ClusterState, NodeMetrics, NodeStatus};
55pub use discovery::{broadcast_and_collect, UdpDiscoveryConfig, DEFAULT_DISCOVERY_PORT};
56pub use host::{
57 detect_local_node_resources, detect_runtime_profile, detect_runtime_profile_full,
58 detect_runtime_profile_with_mode, infer_compute_capability_from_name, is_apple_silicon,
59 AccelerationMode, GpuBackend, ProbeMode, RuntimeProfile,
60};
61pub use kv_cache::{KVCacheConfig, KVCacheEntry, LayerKvCache};
62pub use planning::{
63 assign_layers_sequentially, assign_layers_with_fault_tolerance_and_runtime,
64 assign_layers_with_runtime_profile, chunk_assignments_for_workers, select_quantization_mode,
65 LayerAssignment, LayerSpec, PlacementPlan, PlanningTuning, QuantizationMode,
66};
67pub use protocol::NodeResources;
68pub use ring::{RingConfig, SpscRingBuffer};
69pub use runtime::{
70 build_token_schedule, execute_pipeline, execute_pipeline_tcp_loopback,
71 execute_pipeline_tcp_loopback_with_config, BridgeAddr, BridgeListener, BridgeStream,
72 DeviceKind, ExecutionResult, PipelinePlan, StageExecutionStats, StagePlacement,
73 TcpTransportConfig, TokenStep, TransportKind,
74};
75pub use system_profile::{
76 CpuFeatures, CpuInfo, GpuInfo, MemoryInfo, NetworkInfo, NpuInfo, SystemProfile,
77};
78pub use watcher::{ProfileChange, SystemProfileWatcher};