INNER CODE UNIT · Rust

nccl4rust_dev_comm_n_ranks

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

    pub fn nccl4rust_dev_comm_n_ranks(comm: *const ncclDevComm_t) -> i32;
    pub fn nccl4rust_dev_comm_lsa_rank(comm: *const ncclDevComm_t) -> i32;
    pub fn nccl4rust_dev_comm_lsa_size(comm: *const ncclDevComm_t) -> i32;

    pub fn nccl4rust_team_rank_to_world(
        comm: *const ncclDevComm_t,
        team_n_ranks: i32,
        team_rank: i32,
        team_stride: i32,
        rank: i32,
    ) -> i32;
    pub fn nccl4rust_team_rank_to_lsa(
        comm: *const ncclDevComm_t,
        team_n_ranks: i32,
        team_rank: i32,
        team_stride: i32,
        rank: i32,
    ) -> i32;

View source record →