torsh-backend 0.1.2

Backend abstraction layer for ToRSh
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

//! Auto-generated module
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)

use super::super::allocation::AllocationStats;

use super::types::{
    CudaMemoryStatisticsManager, DistributionType, EfficiencyTrend, ErrorImpactLevel,
    GlobalMemoryStatistics, MemoryPressureLevel, StatisticsConfig,
};

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_statistics_manager_creation() {
        let config = StatisticsConfig::default();
        let manager = CudaMemoryStatisticsManager::new(config);
        assert!(manager.config.enable_detailed_tracking);
        assert!(manager.config.enable_historical_data);
    }
    #[test]
    fn test_memory_pressure_levels() {
        assert!(MemoryPressureLevel::Critical > MemoryPressureLevel::High);
        assert!(MemoryPressureLevel::High > MemoryPressureLevel::Medium);
        assert!(MemoryPressureLevel::Medium > MemoryPressureLevel::Low);
        assert!(MemoryPressureLevel::Low > MemoryPressureLevel::None);
    }
    #[test]
    fn test_distribution_types() {
        assert_eq!(DistributionType::Normal, DistributionType::Normal);
        assert_ne!(DistributionType::Normal, DistributionType::LogNormal);
    }
    #[test]
    fn test_efficiency_trends() {
        assert_eq!(EfficiencyTrend::Improving, EfficiencyTrend::Improving);
        assert_ne!(EfficiencyTrend::Stable, EfficiencyTrend::Declining);
    }
    #[test]
    fn test_error_impact_levels() {
        assert!(ErrorImpactLevel::Critical > ErrorImpactLevel::High);
        assert!(ErrorImpactLevel::High > ErrorImpactLevel::Moderate);
        assert!(ErrorImpactLevel::Moderate > ErrorImpactLevel::Low);
        assert!(ErrorImpactLevel::Low > ErrorImpactLevel::Minimal);
    }
    #[test]
    fn test_statistics_update() {
        let config = StatisticsConfig::default();
        let manager = CudaMemoryStatisticsManager::new(config);
        let stats = AllocationStats {
            total_allocations: 100,
            current_bytes_allocated: 1024 * 1024,
            ..Default::default()
        };
        manager.update_device_stats(0, &stats);
        let device_stats = manager
            .device_stats
            .read()
            .expect("lock should not be poisoned");
        assert!(device_stats.contains_key(&0));
    }
}
/// Memory usage statistics (alias to GlobalMemoryStatistics)
pub type MemoryUsageStatistics = GlobalMemoryStatistics;