INNER CODE UNIT · Rust

nccl4rust_team_rank_to_lsa

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

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

    pub fn nccl4rust_get_local_pointer(window: ncclWindow_t, offset: usize) -> *mut c_void;
    pub fn nccl4rust_get_lsa_pointer(window: ncclWindow_t, offset: usize, peer: i32)
    -> *mut c_void;
    pub fn nccl4rust_get_peer_pointer(
        window: ncclWindow_t,
        offset: usize,
        peer: i32,
    ) -> *mut c_void;
    pub fn nccl4rust_get_peer_pointer_team(
        window: ncclWindow_t,

View source record →