llml 0.2.3

Implementation of basic math data types with high level frontend and low level backend
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

x86_use!();
use std::{ops::Mul, intrinsics::transmute};
use crate::others::{Complxf, Complxd};
use super::{vec2::EucVecf2, _mm_combine_ps, _mm_low_high_ps, EucVecd2};

impl Mul for Complxf {
    type Output = Self;

    #[inline(always)]
    fn mul (self, rhs: Self) -> Self::Output {
        unsafe {
            let v1 = _mm_shuffle_ps(self.0.0, self.0.0, _MM_SHUFFLE(1, 1, 0, 0));
            let v2 = _mm_combine_ps(rhs.0.0, _mm_set_ps(0., 0., rhs.re(), -rhs.im()));
            let m1 = _mm_mul_ps(v1, v2);

            let (v1, v2) = _mm_low_high_ps(m1);
            Self(EucVecf2(_mm_add_ps(v1, v2)))
        }
    }
}

impl Mul for Complxd {
    type Output = Self;

    #[inline(always)]
    fn mul (self, rhs: Self) -> Self::Output {
        unsafe {
            let m1 = _mm_mul_pd(_mm_set1_pd(self.re()), rhs.0.0);
            let m2 = _mm_mul_pd(_mm_set1_pd(self.im()), transmute([-rhs.im(), rhs.re()]));
            Self(EucVecd2(_mm_add_pd(m1, m2)))
        }
    }
}