INNER CODE UNIT · Rust

rank

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

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

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

View source record →

📰 Research Paper
Loading…
⏳ Fetching content…