INNER CODE UNIT · Rust

rank_to_lsa

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

pub fn rank_to_lsa(comm: DevComm<'_>, team: Team, rank: i32) -> Option<i32> {
    if rank < 0 || rank >= team.size() {
        None
    } else {
        let translated = rank_to_lsa_device(comm, team.size(), team.rank(), team.stride(), rank);
        if translated < 0 || translated >= lsa_size(comm) {
            None
        } else {
            Some(translated)
        }
    }
}

#[inline(always)]
fn rank_to_lsa_device(
    comm: DevComm<'_>,
    team_n_ranks: i32,
    team_rank: i32,

View source record →