INNER CODE UNIT · Rust

nccl4rust_lsa_reduce_sum_copy_f32_cta

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

    pub fn nccl4rust_lsa_reduce_sum_copy_f32_cta(
        comm: *const ncclDevComm_t,
        src_window: ncclWindow_t,
        src_offset: usize,
        dst_window: ncclWindow_t,
        dst_offset: usize,
        count: usize,
    );
}

const _: () = {
    assert!(core::mem::size_of::<ncclTeam_t>() == 12);
    assert!(core::mem::align_of::<ncclTeam_t>() == 4);
    assert!(core::mem::offset_of!(ncclTeam_t, n_ranks) == 0);
    assert!(core::mem::offset_of!(ncclTeam_t, rank) == 4);
    assert!(core::mem::offset_of!(ncclTeam_t, stride) == 8);
    assert!(core::mem::size_of::<ncclMultimemHandle_t>() == core::mem::size_of::<usize>());
    assert!(core::mem::align_of::<ncclMultimemHandle_t>() == core::mem::align_of::<usize>());

View source record →

📰 Research Paper
Loading…
⏳ Fetching content…