archx 3.0.1

High-performance CPU/GPU adaptive optimization library with SIMD and Multithreading
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

/// Represents specific CPU instruction set extensions.
#[derive(Debug, Clone, Copy, Default, serde::Serialize)]
pub struct CpuFeatures {
    /// Streaming SIMD Extensions 2
    pub sse2: bool,
    /// Advanced Vector Extensions
    pub avx: bool,
    /// Advanced Vector Extensions 2
    pub avx2: bool,
    /// Advanced Vector Extensions 512 Foundation
    pub avx512f: bool,
    /// ARM NEON (Advanced SIMD)
    pub neon: bool,
}

impl CpuFeatures {
    /// Detects available CPU features at runtime.
    pub fn detect() -> Self {
        let mut features = CpuFeatures::default();

        #[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
        {
            features.sse2 = std::is_x86_feature_detected!("sse2");
            features.avx = std::is_x86_feature_detected!("avx");
            features.avx2 = std::is_x86_feature_detected!("avx2");
            features.avx512f = std::is_x86_feature_detected!("avx512f");
        }

        #[cfg(target_arch = "aarch64")]
        {
            // Neon is mandatory on AArch64, but we can check specifically if needed
            // or just set to true on this platform.
            features.neon = true; 
        }

        features
    }
}