Skip to content

Commit

Permalink
Rollup merge of #86393 - yerke:add-test-for-issue-52025, r=JohnTitor
Browse files Browse the repository at this point in the history
Add regression test for issue #52025

Closes #52025

Took the test from #52025
  • Loading branch information
JohnTitor committed Jun 22, 2021
2 parents 44f4a87 + 311f578 commit 0cfaa27
Showing 1 changed file with 57 additions and 0 deletions.
57 changes: 57 additions & 0 deletions src/test/ui/traits/operator-overloading-issue-52025.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,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>() });
}

0 comments on commit 0cfaa27

Please sign in to comment.