distributed.rs

// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

use super::*;

mod leader;
mod utils;
mod worker;

pub use leader::KvbmLeader;
pub use utils::{get_leader_zmq_ack_url, get_leader_zmq_pub_url};
pub use worker::{KvbmWorker, PyLayoutType, VllmTensor};
#[cfg(feature = "nccl")]
pub use worker::{PyNcclBootstrap, PyNcclCommRef};
#[cfg(not(feature = "nccl"))]
pub use worker::{PyNcclBootstrap, PyNcclCommRef};