INNER CODE UNIT · Rust

world

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

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

#[inline(always)]
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.

View source record →