corpus-ipc 0.1.0

Inter-Process Communication (IPC) library for bridging Rust to external compute engines via ZeroMQ.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// SPDX-License-Identifier: MIT OR Apache-2.0

use super::*;

#[test]
fn aggregate_accepts_exact_max() {
    let limits = limits();
    let aggregate = GradientBatch {
        session_id: "s".into(),
        batch_id: 1,
        gradients: vec![GradientUpdate {
            layer_id: "a".into(),
            gradients: vec![0.0; 4],
            eligibility_trace: Some(vec![0.0; 4]),
        }],
    };
    let tight = ProtocolLimits {
        max_aggregate_records: 9,
        ..limits
    };
    aggregate.validate_with(tight).expect("aggregate max");
}

#[test]
fn aggregate_rejects_max_plus_one() {
    let limits = limits();
    let mut aggregate = GradientBatch {
        session_id: "s".into(),
        batch_id: 1,
        gradients: vec![GradientUpdate {
            layer_id: "a".into(),
            gradients: vec![0.0; 4],
            eligibility_trace: Some(vec![0.0; 4]),
        }],
    };
    let tight = ProtocolLimits {
        max_aggregate_records: 9,
        ..limits
    };
    aggregate.gradients[0].gradients.push(0.0);
    let err = aggregate.validate_with(tight).unwrap_err();
    assert_limit_exceeded(&err, "aggregate");
}