INNER CODE UNIT · Rust

lsa_size

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

pub fn lsa_size(comm: DevComm<'_>) -> i32 {
    unsafe { sys::nccl4rust_dev_comm_lsa_size(comm.as_raw()) }
}

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

View source record →