INNER CODE UNIT · Rust

size

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

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

#[inline(always)]
pub fn lsa_rank(comm: DevComm<'_>) -> i32 {
    unsafe { sys::nccl4rust_dev_comm_lsa_rank(comm.as_raw()) }
}

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

View source record →