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
|
//@ only-x86_64
//@ build-pass
use std::arch::x86_64::*;
use std::fmt::Debug;
use std::ops::*;
pub trait Simd {
type Vf32: Copy + Debug + Add<Self::Vf32, Output = Self::Vf32> + Add<f32, Output = Self::Vf32>;
unsafe fn set1_ps(a: f32) -> Self::Vf32;
unsafe fn add_ps(a: Self::Vf32, b: Self::Vf32) -> Self::Vf32;
}
#[derive(Copy, Debug, Clone)]
pub struct F32x4(pub __m128);
impl Add<F32x4> for F32x4 {
type Output = F32x4;
fn add(self, rhs: F32x4) -> F32x4 {
F32x4(unsafe { _mm_add_ps(self.0, rhs.0) })
}
}
impl Add<f32> for F32x4 {
type Output = F32x4;
fn add(self, rhs: f32) -> F32x4 {
F32x4(unsafe { _mm_add_ps(self.0, _mm_set1_ps(rhs)) })
}
}
pub struct Sse2;
impl Simd for Sse2 {
type Vf32 = F32x4;
#[inline(always)]
unsafe fn set1_ps(a: f32) -> Self::Vf32 {
F32x4(_mm_set1_ps(a))
}
#[inline(always)]
unsafe fn add_ps(a: Self::Vf32, b: Self::Vf32) -> Self::Vf32 {
F32x4(_mm_add_ps(a.0, b.0))
}
}
unsafe fn test<S: Simd>() -> S::Vf32 {
let a = S::set1_ps(3.0);
let b = S::set1_ps(2.0);
let result = a + b;
result
}
fn main() {
println!("{:?}", unsafe { test::<Sse2>() });
}
|
