ruCCL 0.21.1

Ruda collective communication algorithms and orchestration.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use super::{CollectiveAlgorithm, CollectiveTransport};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct CollectiveStats {
    pub algorithm: CollectiveAlgorithm,
    pub transport: CollectiveTransport,
    pub steps: u32,
    pub transferred_bytes: u64,
    pub reduction_kernel_launches: u32,
}

impl CollectiveStats {
    pub fn new(algorithm: CollectiveAlgorithm) -> Self {
        Self {
            algorithm,
            transport: CollectiveTransport::HostStaged,
            steps: 0,
            transferred_bytes: 0,
            reduction_kernel_launches: 0,
        }
    }
}