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
//! Implements vertical (lane-wise) floating-point `sqrt`.

macro_rules! impl_math_float_sqrt {
    ([$elem_ty:ident; $elem_count:expr]: $id:ident | $test_tt:tt) => {
        impl $id {
            #[inline]
            pub fn sqrt(self) -> Self {
                use crate::codegen::math::float::sqrt::Sqrt;
                Sqrt::sqrt(self)
            }
        }

        test_if!{
            $test_tt:
            paste::item! {
                pub mod [<$id _math_sqrt>] {
                    use super::*;
                    #[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
                    fn sqrt() {
                        use crate::$elem_ty::consts::SQRT_2;
                        let z = $id::splat(0 as $elem_ty);
                        let o = $id::splat(1 as $elem_ty);
                        assert_eq!(z, z.sqrt());
                        assert_eq!(o, o.sqrt());

                        let t = $id::splat(2 as $elem_ty);
                        let e = $id::splat(SQRT_2);
                        assert_eq!(e, t.sqrt());

                    }
                }
            }
        }
    };
}