numrs2 0.3.3

A Rust implementation inspired by NumPy for numerical computing (NumRS2)
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

//! Custom memory allocation for numerical workloads
//!
//! This module provides specialized memory allocators optimized for
//! numerical computing workloads.

pub mod aligned;
pub mod aligned_helpers;
pub mod arena;
pub mod benchmarking;
pub mod cache_optimization;
pub mod enhanced_traits;
pub mod large_scale;
pub mod monitored_allocator;
pub mod out_of_core;
pub mod performance_tuning;
pub mod pool;
pub mod strategy;

// Re-export the main types and functions for convenience
pub use aligned::{AlignedAllocator, AlignmentConfig};
pub use aligned_helpers::{
    aligned_alloc, aligned_alloc_zeroed, aligned_dealloc, cache_line_size, is_aligned,
    optimal_simd_alignment, AlignedBox, AlignedVec,
};
pub use arena::{ArenaAllocator, ArenaConfig};
pub use benchmarking::{benchmark_configs, AllocatorBenchmark, BenchmarkConfig, BenchmarkResults};
pub use cache_optimization::{
    cache_constants, CacheConfig, CacheLevel, CacheMetrics, CacheOptimizationRecommendation,
    CacheOptimizationType, CacheOptimizedAllocator,
};
pub use enhanced_traits::{
    EnhancedAllocatorBridge, IntelligentAllocationStrategy, NumericalArrayAllocator,
};
pub use large_scale::{
    get_global_memory_stats, get_global_spill_stats, init_global_manager,
    load_spilled_data_globally, should_spill_globally, spill_data_globally, with_global_manager,
    with_global_manager_mut, ChunkIterator, LargeScaleConfig, LargeScaleManager, MemoryStats,
    MemoryTracker, SpillStats,
};
pub use monitored_allocator::{
    presets, MonitoredAllocator, MonitoredAllocatorFactory, MonitoringConfig,
};
pub use out_of_core::{CacheStats, CacheStrategy, OutOfCoreArray, OutOfCoreConfig};
pub use performance_tuning::{
    init_global_tuner, with_global_tuner, with_global_tuner_mut, OptimizationRecommendation,
    OptimizationType, PerformanceMetrics, PerformanceTuner, TuningConfig,
};
pub use pool::{PoolAllocator, PoolConfig};
pub use strategy::{get_default_allocator, AllocStrategy, MemoryAllocator};

/// Initialize the global allocator with the preferred strategy
pub fn init_global_allocator(strategy: AllocStrategy) {
    strategy::set_global_allocator(strategy);
}

/// Helper function to get the current global allocator strategy
pub fn get_global_allocator_strategy() -> AllocStrategy {
    strategy::get_global_allocator_strategy()
}

/// Helper function to reset the global allocator to the default strategy
pub fn reset_global_allocator() {
    strategy::reset_global_allocator();
}