third_party/rust/packed_simd/src/api/ops/scalar_shifts.rs


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

//! Vertical (lane-wise) vector-scalar shifts operations.

macro_rules! impl_ops_scalar_shifts {
    ([$elem_ty:ident; $elem_count:expr]: $id:ident | $test_tt:tt) => {
        impl crate::ops::Shl<u32> for $id {
            type Output = Self;
            #[inline]
            fn shl(self, other: u32) -> Self {
                self << $id::splat(other as $elem_ty)
            }
        }
        impl crate::ops::Shr<u32> for $id {
            type Output = Self;
            #[inline]
            fn shr(self, other: u32) -> Self {
                self >> $id::splat(other as $elem_ty)
            }
        }

        impl crate::ops::ShlAssign<u32> for $id {
            #[inline]
            fn shl_assign(&mut self, other: u32) {
                *self = *self << other;
            }
        }
        impl crate::ops::ShrAssign<u32> for $id {
            #[inline]
            fn shr_assign(&mut self, other: u32) {
                *self = *self >> other;
            }
        }
        test_if!{
            $test_tt:
            paste::item! {
                pub mod [<$id _ops_scalar_shifts>] {
                    use super::*;
                    #[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
                    #[cfg_attr(any(target_arch = "s390x", target_arch = "sparc64"),
                               allow(unreachable_code,
                                     unused_variables,
                                     unused_mut)
                    )]
                    // ^^^ FIXME: https://github.com/rust-lang/rust/issues/55344
                    fn ops_scalar_shifts() {
                        let z = $id::splat(0 as $elem_ty);
                        let o = $id::splat(1 as $elem_ty);
                        let t = $id::splat(2 as $elem_ty);
                        let f = $id::splat(4 as $elem_ty);

                        {
                            let zi = 0 as u32;
                            let oi = 1 as u32;
                            let ti = 2 as u32;
                            let maxi
                                = (mem::size_of::<$elem_ty>() * 8 - 1) as u32;

                            // shr
                            assert_eq!(z >> zi, z);
                            assert_eq!(z >> oi, z);
                            assert_eq!(z >> ti, z);
                            assert_eq!(z >> ti, z);

                            #[cfg(any(target_arch = "s390x", target_arch = "sparc64"))] {
                                // FIXME: https://github.com/rust-lang-nursery/packed_simd/issues/13
                                return;
                            }

                            assert_eq!(o >> zi, o);
                            assert_eq!(t >> zi, t);
                            assert_eq!(f >> zi, f);
                            assert_eq!(f >> maxi, z);

                            assert_eq!(o >> oi, z);
                            assert_eq!(t >> oi, o);
                            assert_eq!(t >> ti, z);
                            assert_eq!(f >> oi, t);
                            assert_eq!(f >> ti, o);
                            assert_eq!(f >> maxi, z);

                            // shl
                            assert_eq!(z << zi, z);
                            assert_eq!(o << zi, o);
                            assert_eq!(t << zi, t);
                            assert_eq!(f << zi, f);
                            assert_eq!(f << maxi, z);

                            assert_eq!(o << oi, t);
                            assert_eq!(o << ti, f);
                            assert_eq!(t << oi, f);

                            {  // shr_assign
                                let mut v = o;
                                v >>= oi;
                                assert_eq!(v, z);
                            }
                            {  // shl_assign
                                let mut v = o;
                                v <<= oi;
                                assert_eq!(v, t);
                            }
                        }
                    }
                }
            }
        }
    };
}