optirs-tpu 0.3.1

OptiRS TPU coordination and pod management
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

// Communication Module

use crate::pod_coordination::types::*;
use serde::{Deserialize, Serialize};

pub mod authentication;
pub mod authorization;
pub mod buffer_management;
pub mod buffers;
pub mod certificate_management;
pub mod communication_core;
pub mod compression;
pub mod core;
pub mod encryption;
pub mod identity_management;
pub mod key_management;
pub mod monitoring;
pub mod network_config;
pub mod policy_management;
pub mod protocols;
pub mod qos;
pub mod reliability;
pub mod routing;
pub mod scheduling;
pub mod security;
pub mod security_monitoring;

pub use authentication::*;
pub use authorization::*;
pub use buffer_management::*;
pub use buffers::*;
pub use certificate_management::*;
pub use communication_core::*;
pub use compression::*;
pub use core::*;
pub use encryption::*;
pub use identity_management::*;
pub use key_management::*;
pub use monitoring::*;
pub use network_config::*;
pub use policy_management::*;
pub use protocols::*;
pub use qos::*;
pub use reliability::*;
pub use routing::*;
pub use scheduling::*;
pub use security::*;
pub use security_monitoring::*;

// Define missing types
pub type CommunicationId = u64;

#[derive(Debug, Clone, Default)]
pub struct ActiveCommunication {
    pub id: CommunicationId,
    pub active: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub enum BufferStatus {
    #[default]
    Empty,
    Partial,
    Full,
}

#[derive(Debug, Clone, Default)]
pub struct CommunicationBuffer {
    pub status: BufferStatus,
    pub size: usize,
}

#[derive(Debug, Clone, Default)]
pub struct CommunicationManager {
    pub active_comms: Vec<ActiveCommunication>,
}

#[derive(Debug, Clone, Default)]
pub struct CommunicationProgress {
    pub bytes_sent: u64,
    pub bytes_total: u64,
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub enum CompressionAlgorithm {
    #[default]
    None,
    Gzip,
    Lz4,
    Zstd,
}

#[derive(Debug, Clone, Default)]
pub struct CompressionInfo {
    pub algorithm: CompressionAlgorithm,
    pub ratio: f64,
}

#[derive(Debug, Clone, Default)]
pub struct CommunicationStatistics {
    pub total_bytes: u64,
    pub total_messages: u64,
}