simdeez 3.0.1

SIMD library to abstract over different instruction sets and widths
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
65

use crate::SimdBaseIo;

pub trait SimdIter: SimdBaseIo {
    #[inline(always)]
    fn iter(&self) -> SimdArrayIterator<'_, Self> {
        SimdArrayIterator {
            simd: self,
            index: 0,
        }
    }

    #[inline(always)]
    fn iter_mut(&mut self) -> SimdArrayMutIterator<'_, Self> {
        SimdArrayMutIterator {
            simd: self,
            index: 0,
        }
    }
}

impl<T: SimdBaseIo> SimdIter for T {}

pub struct SimdArrayIterator<'a, S: SimdBaseIo> {
    simd: &'a S,
    index: usize,
}

impl<S: SimdBaseIo> Iterator for SimdArrayIterator<'_, S> {
    type Item = S::Scalar;

    #[inline(always)]
    fn next(&mut self) -> Option<Self::Item> {
        if self.index >= S::WIDTH {
            return None;
        }

        unsafe {
            let value = self.simd.get_unchecked(self.index);
            self.index += 1;
            Some(value)
        }
    }
}

pub struct SimdArrayMutIterator<'a, S: SimdBaseIo> {
    simd: &'a mut S,
    index: usize,
}

impl<'a, S: SimdBaseIo> Iterator for SimdArrayMutIterator<'a, S> {
    type Item = &'a mut S::Scalar;

    #[inline(always)]
    fn next(&mut self) -> Option<Self::Item> {
        if self.index >= S::WIDTH {
            return None;
        }

        unsafe {
            let value = self.simd.get_unchecked_mut(self.index);
            self.index += 1;
            Some(value)
        }
    }
}