Deprecated: The each() function is deprecated. This message will be suppressed on further calls in /home/zhenxiangba/zhenxiangba.com/public_html/phproxy-improved-master/index.php on line 456
multiversion 0.5.1 - Docs.rs
[go: Go Back, main page]

multiversion 0.5.1

Easy function multiversioning
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use multiversion::multiversion;

#[multiversion]
#[clone(target = "[x86|x86_64]+avx")]
#[clone(target = "[x86|x86_64]+sse")]
#[clone(target = "[arm|aarch64]+neon")]
pub fn pub_add(a: &mut [f32], b: &[f32]) {
    a.iter_mut().zip(b.iter()).for_each(|(a, b)| *a += b);
}

#[multiversion]
#[clone(target = "[x86|x86_64]+avx")]
#[clone(target = "[x86|x86_64]+sse")]
#[clone(target = "[arm|aarch64]+neon")]
fn priv_add(a: &mut [f32], b: &[f32]) {
    a.iter_mut().zip(b.iter()).for_each(|(a, b)| *a += b);
}

#[multiversion]
#[clone(target = "[x86|x86_64]+avx")]
#[clone(target = "[x86|x86_64]+sse")]
#[clone(target = "[arm|aarch64]+neon")]
#[inline]
pub unsafe fn pub_unsafe_add(a: &mut [f32], b: &[f32]) {
    a.iter_mut().zip(b.iter()).for_each(|(a, b)| *a += b);
}

#[multiversion]
#[clone(target = "[x86|x86_64]+avx")]
#[clone(target = "[x86|x86_64]+sse")]
#[clone(target = "[arm|aarch64]+neon")]
#[inline]
unsafe fn priv_unsafe_add(a: &mut [f32], b: &[f32]) {
    a.iter_mut().zip(b.iter()).for_each(|(a, b)| *a += b);
}

struct Adder(f32);

impl Adder {
    #[multiversion]
    #[clone(target = "[x86|x86_64]+avx")]
    #[clone(target = "[x86|x86_64]+sse")]
    #[clone(target = "[arm|aarch64]+neon")]
    #[inline]
    pub fn pub_add(&self, x: &mut [f32]) {
        x.iter_mut().for_each(|x| *x += self.0);
    }

    #[multiversion]
    #[clone(target = "[x86|x86_64]+avx")]
    #[clone(target = "[x86|x86_64]+sse")]
    #[clone(target = "[arm|aarch64]+neon")]
    #[inline]
    fn priv_add(&self, x: &mut [f32]) {
        x.iter_mut().for_each(|x| *x += self.0);
    }

    #[multiversion]
    #[clone(target = "[x86|x86_64]+avx")]
    #[clone(target = "[x86|x86_64]+sse")]
    #[clone(target = "[arm|aarch64]+neon")]
    #[inline]
    pub unsafe fn pub_unsafe_add(&self, x: &mut [f32]) {
        x.iter_mut().for_each(|x| *x += self.0);
    }

    #[multiversion]
    #[clone(target = "[x86|x86_64]+avx")]
    #[clone(target = "[x86|x86_64]+sse")]
    #[clone(target = "[arm|aarch64]+neon")]
    #[inline]
    unsafe fn priv_unsafe_add(&self, x: &mut [f32]) {
        x.iter_mut().for_each(|x| *x += self.0);
    }
}

mod test {
    use super::*;

    #[test]
    fn test_add() {
        let mut a = vec![0f32, 2f32, 4f32];
        let b = vec![1f32, 1f32, 1f32];
        pub_add(&mut a, &b);
        assert_eq!(a, vec![1f32, 3f32, 5f32]);
        priv_add(&mut a, &b);
        assert_eq!(a, vec![2f32, 4f32, 6f32]);
        unsafe {
            pub_unsafe_add(&mut a, &b);
        }
        assert_eq!(a, vec![3f32, 5f32, 7f32]);
        unsafe {
            priv_unsafe_add(&mut a, &b);
        }
        assert_eq!(a, vec![4f32, 6f32, 8f32]);
    }

    #[test]
    fn test_add_associated() {
        let adder = Adder(1.);
        let mut a = vec![0f32, 2f32, 4f32];
        adder.pub_add(&mut a);
        assert_eq!(a, vec![1f32, 3f32, 5f32]);
        adder.priv_add(&mut a);
        assert_eq!(a, vec![2f32, 4f32, 6f32]);
        unsafe {
            adder.pub_unsafe_add(&mut a);
        }
        assert_eq!(a, vec![3f32, 5f32, 7f32]);
        unsafe {
            adder.priv_unsafe_add(&mut a);
        }
        assert_eq!(a, vec![4f32, 6f32, 8f32]);
    }
}