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

// Network Traffic Module

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

pub type FlowId = u64;

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct TrafficFlow {
    pub flow_id: FlowId,
    pub bandwidth_mbps: f64,
}

#[derive(Debug, Clone, Default)]
pub struct TrafficManager {
    pub flows: HashMap<FlowId, TrafficFlow>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum TrafficPattern {
    Uniform,
    Burst,
    Adaptive,
}

impl Default for TrafficPattern {
    fn default() -> Self { Self::Uniform }
}