INNER CODE UNIT · Rust

lsa

NVIDIA/nccl · contrib/nccl4rust/crates/nccl-device/src/lib.rs:153

pub fn lsa(comm: DevComm<'_>) -> Team {
    Team(sys::ncclTeam_t {
        n_ranks: lsa_size(comm),
        rank: lsa_rank(comm),
        stride: 1,
    })
}

/// Return the rail team containing the communicator's rank.
///
/// A live NCCL device communicator always has a positive LSA size, which is
/// required by the divisions used to derive the rail size and rank.
#[inline(always)]
pub fn rail(comm: DevComm<'_>) -> Team {
    let n_ranks = size(comm);
    let rank = rank(comm);
    let lsa_size = lsa_size(comm);
    Team(sys::ncclTeam_t {

View source record →