trueno 0.18.0

High-performance SIMD compute library with GPU support, LLM inference engine, and GGUF model loading
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

//! H0-MON-30 through H0-MON-61: Memory, thermal, power, and GPU metrics tests

use std::time::Duration;

use crate::monitor::*;

// =========================================================================
// H0-MON-30: GpuMemoryMetrics
// =========================================================================

#[test]
fn h0_mon_30_memory_metrics_basic() {
    let mem = GpuMemoryMetrics::new(24_000_000_000, 8_000_000_000, 16_000_000_000);

    assert_eq!(mem.total, 24_000_000_000);
    assert_eq!(mem.used, 8_000_000_000);
    assert_eq!(mem.free, 16_000_000_000);
}

#[test]
fn h0_mon_31_memory_metrics_usage_percent() {
    let mem = GpuMemoryMetrics::new(100, 25, 75);
    assert!((mem.usage_percent() - 25.0).abs() < 0.01);
}

#[test]
fn h0_mon_32_memory_metrics_usage_percent_zero_total() {
    let mem = GpuMemoryMetrics::new(0, 0, 0);
    assert!((mem.usage_percent() - 0.0).abs() < 0.01);
}

#[test]
fn h0_mon_33_memory_metrics_mb_helpers() {
    let mem = GpuMemoryMetrics::new(
        24 * 1024 * 1024 * 1024,
        8 * 1024 * 1024 * 1024,
        16 * 1024 * 1024 * 1024,
    );

    assert_eq!(mem.used_mb(), 8 * 1024);
    assert_eq!(mem.free_mb(), 16 * 1024);
}

// =========================================================================
// H0-MON-40: GpuThermalMetrics
// =========================================================================

#[test]
fn h0_mon_40_thermal_safe() {
    let thermal = GpuThermalMetrics { temperature_celsius: 50, ..Default::default() };
    assert!(thermal.is_safe());
    assert!(!thermal.is_critical());
    assert_eq!(thermal.status(), "COOL");
}

#[test]
fn h0_mon_41_thermal_warm() {
    let thermal = GpuThermalMetrics { temperature_celsius: 65, ..Default::default() };
    assert!(thermal.is_safe());
    assert!(!thermal.is_critical());
    assert_eq!(thermal.status(), "WARM");
}

#[test]
fn h0_mon_42_thermal_hot() {
    let thermal = GpuThermalMetrics { temperature_celsius: 82, ..Default::default() };
    assert!(!thermal.is_safe());
    assert!(!thermal.is_critical());
    assert_eq!(thermal.status(), "HOT");
}

#[test]
fn h0_mon_43_thermal_critical() {
    let thermal = GpuThermalMetrics { temperature_celsius: 95, ..Default::default() };
    assert!(!thermal.is_safe());
    assert!(thermal.is_critical());
    assert_eq!(thermal.status(), "CRITICAL");
}

// =========================================================================
// H0-MON-50: GpuPowerMetrics
// =========================================================================

#[test]
fn h0_mon_50_power_usage_percent() {
    let power =
        GpuPowerMetrics { power_draw_watts: 225.0, power_limit_watts: 450.0, power_state: 0 };
    assert!((power.usage_percent() - 50.0).abs() < 0.01);
}

#[test]
fn h0_mon_51_power_usage_percent_zero_limit() {
    let power = GpuPowerMetrics { power_draw_watts: 100.0, power_limit_watts: 0.0, power_state: 0 };
    assert!((power.usage_percent() - 0.0).abs() < 0.01);
}

// =========================================================================
// H0-MON-60: GpuMetrics
// =========================================================================

#[test]
fn h0_mon_60_metrics_creation() {
    let mem = GpuMemoryMetrics::new(1000, 500, 500);
    let metrics = GpuMetrics::new(0, mem);

    assert_eq!(metrics.device_index, 0);
    assert_eq!(metrics.memory.total, 1000);
    assert!(metrics.thermal.is_none());
    assert!(metrics.power.is_none());
}

#[test]
fn h0_mon_61_metrics_age() {
    let mem = GpuMemoryMetrics::new(1000, 500, 500);
    let metrics = GpuMetrics::new(0, mem);

    // Age should be very small immediately after creation
    assert!(metrics.age() < Duration::from_millis(100));
}