macerator 0.3.2

Type and target-generic SIMD
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

use half::f16;
use paste::paste;

use crate::{Scalar, Simd, Vector};

pub trait ReduceAdd: Scalar {
    fn reduce_add<S: Simd>(input: Vector<S, Self>) -> Self;
    fn is_accelerated<S: Simd>() -> bool;
}

impl<S: Simd, T: ReduceAdd> Vector<S, T> {
    #[inline(always)]
    pub fn reduce_add(self) -> T {
        T::reduce_add(self)
    }
}

pub trait ReduceMin: Scalar {
    fn reduce_min<S: Simd>(input: Vector<S, Self>) -> Self;
    fn is_accelerated<S: Simd>() -> bool;
}

impl<S: Simd, T: ReduceMin> Vector<S, T> {
    #[inline(always)]
    pub fn reduce_min(self) -> T {
        T::reduce_min(self)
    }
}

pub trait ReduceMax: Scalar {
    fn reduce_max<S: Simd>(input: Vector<S, Self>) -> Self;
    fn is_accelerated<S: Simd>() -> bool;
}

impl<S: Simd, T: ReduceMax> Vector<S, T> {
    #[inline(always)]
    pub fn reduce_max(self) -> T {
        T::reduce_max(self)
    }
}

macro_rules! impl_reduce {
    ($trait: ident, $name: ident, $($ty: ty),*) => {
        $(paste! {
            impl $trait for $ty {
                #[inline(always)]
                fn [<$trait:snake>]<S: Simd>(input: Vector<S, Self>) -> Self {
                    S::[<$name _ $ty>](*input)
                }
                #[inline(always)]
                fn is_accelerated<S: Simd>() -> bool {
                    S::[<$name _ $ty _supported>]()
                }
            }
        })*
    };
}

impl_reduce!(ReduceAdd, reduce_add, u8, i8, u16, i16, u32, i32, u64, i64, f16, f32, f64);
impl_reduce!(ReduceMin, reduce_min, u8, i8, u16, i16, u32, i32, u64, i64, f16, f32, f64);
impl_reduce!(ReduceMax, reduce_max, u8, i8, u16, i16, u32, i32, u64, i64, f16, f32, f64);