moirai-utils 0.5.0

Utility functions and data structures for Moirai concurrency library
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

//! Private native ISA backends for the generic SIMD surface.

#[cfg(target_arch = "aarch64")]
mod aarch64;
#[cfg(target_arch = "x86_64")]
mod x86;

/// Returns true when the current x86 CPU supports AVX2.
#[inline]
pub fn has_avx2_support() -> bool {
    #[cfg(all(target_arch = "x86_64", feature = "std", not(target_arch = "wasm32")))]
    {
        std::is_x86_feature_detected!("avx2")
    }

    #[cfg(not(all(target_arch = "x86_64", feature = "std", not(target_arch = "wasm32"))))]
    {
        false
    }
}

/// Returns true when the current AArch64 CPU supports NEON.
#[inline]
pub fn has_neon_support() -> bool {
    #[cfg(target_arch = "aarch64")]
    {
        true
    }

    #[cfg(not(target_arch = "aarch64"))]
    {
        false
    }
}

#[cfg(target_arch = "x86_64")]
pub(crate) use x86::{
    add, add_wide, dot, dot_wide, matrix_mul_square, mul, mul_wide, squared_diff_sum,
    squared_diff_sum_wide, sum, sum_wide, LANES,
};

#[cfg(target_arch = "aarch64")]
pub(crate) use aarch64::{add, dot, matrix_mul_square, mul, squared_diff_sum, sum, LANES};