ai-hwaccel 1.1.0

Universal AI hardware accelerator detection, capability querying, and workload planning for Rust
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

//! Tests for container/VM/cloud environment detection.

use crate::*;

#[test]
fn environment_detection_runs() {
    // This just verifies the detection code doesn't panic.
    let registry = AcceleratorRegistry::detect();
    let env = registry.system_io().environment.as_ref();
    // On a dev machine, we expect at least the environment to be detected.
    assert!(env.is_some());
}

#[test]
fn environment_serde_roundtrip() {
    let env = RuntimeEnvironment {
        is_docker: true,
        is_kubernetes: false,
        kubernetes_namespace: None,
        cloud_instance: Some(CloudInstanceMeta {
            provider: "aws".into(),
            instance_type: Some("p4d.24xlarge".into()),
            instance_id: Some("i-1234567890abcdef0".into()),
            region: Some("us-east-1".into()),
            zone: None,
        }),
    };
    let json = serde_json::to_string(&env).unwrap();
    let back: RuntimeEnvironment = serde_json::from_str(&json).unwrap();
    assert_eq!(env, back);
}

#[test]
fn environment_serde_minimal() {
    let env = RuntimeEnvironment {
        is_docker: false,
        is_kubernetes: false,
        kubernetes_namespace: None,
        cloud_instance: None,
    };
    let json = serde_json::to_string(&env).unwrap();
    // Optional fields should be omitted.
    assert!(!json.contains("kubernetes_namespace"));
    assert!(!json.contains("cloud_instance"));
    let back: RuntimeEnvironment = serde_json::from_str(&json).unwrap();
    assert_eq!(env, back);
}

#[test]
fn system_io_without_environment_deserializes() {
    // Old JSON without environment field should deserialize fine.
    let json = r#"{"interconnects":[],"storage":[]}"#;
    let sio: SystemIo = serde_json::from_str(json).unwrap();
    assert!(sio.environment.is_none());
}