Skip to content

Commit

Permalink
Add vminq_f32 and vmaxq_f32 (rust-lang#905)
Browse files Browse the repository at this point in the history
  • Loading branch information
jrmuizel authored Sep 14, 2020
1 parent 016eff9 commit e451716
Showing 1 changed file with 43 additions and 0 deletions.
43 changes: 43 additions & 0 deletions crates/core_arch/src/arm/neon/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -188,6 +188,13 @@ extern "C" {
#[cfg_attr(target_arch = "arm", link_name = "llvm.arm.neon.vpadd.v16i8")]
#[cfg_attr(target_arch = "aarch64", link_name = "llvm.aarch64.neon.addp.v16i8")]
fn vpaddq_s8_(a: int8x16_t, b: int8x16_t) -> int8x16_t;

#[cfg_attr(target_arch = "arm", link_name = "llvm.arm.neon.vmaxs.v4f32")]
#[cfg_attr(target_arch = "aarch64", link_name = "llvm.aarch64.neon.fmax.v4f32")]
fn vmaxq_f32_(a: float32x4_t, b: float32x4_t) -> float32x4_t;
#[cfg_attr(target_arch = "arm", link_name = "llvm.arm.neon.vmins.v4f32")]
#[cfg_attr(target_arch = "aarch64", link_name = "llvm.aarch64.neon.fmin.v4f32")]
fn vminq_f32_(a: float32x4_t, b: float32x4_t) -> float32x4_t;
}

#[cfg(target_arch = "arm")]
Expand Down Expand Up @@ -1836,6 +1843,26 @@ pub unsafe fn vcvtq_u32_f32(a: float32x4_t) -> uint32x4_t {
transmute(simd_cast::<_, u32x4>(transmute::<_, f32x4>(a)))
}

/// Floating-point minimum (vector).
#[inline]
#[target_feature(enable = "neon")]
#[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmin.f32"))]
#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(fmin))]
pub unsafe fn vminq_f32(a: float32x4_t, b: float32x4_t) -> float32x4_t {
vminq_f32_(a, b)
}

/// Floating-point maxmimum (vector).
#[inline]
#[target_feature(enable = "neon")]
#[cfg_attr(target_arch = "arm", target_feature(enable = "v7"))]
#[cfg_attr(all(test, target_arch = "arm"), assert_instr("vmax.f32"))]
#[cfg_attr(all(test, target_arch = "aarch64"), assert_instr(fmax))]
pub unsafe fn vmaxq_f32(a: float32x4_t, b: float32x4_t) -> float32x4_t {
vmaxq_f32_(a, b)
}

#[cfg(test)]
mod tests {
use super::*;
Expand Down Expand Up @@ -4314,6 +4341,22 @@ mod tests {
let e = u8x8::new(3, 7, 11, 15, 61, 65, 69, 73);
assert_eq!(r, e);
}
#[simd_test(enable = "neon")]
unsafe fn test_vminq_f32() {
let a = f32x4::new(1., -2., 3., -4.);
let b = f32x4::new(0., 3., 2., 8.);
let e = f32x4::new(0., -2., 2., -4.);
let r: f32x4 = transmute(vminq_f32(transmute(a), transmute(b)));
assert_eq!(r, e);
}
#[simd_test(enable = "neon")]
unsafe fn test_vmaxq_f32() {
let a = f32x4::new(1., -2., 3., -4.);
let b = f32x4::new(0., 3., 2., 8.);
let e = f32x4::new(1., 3., 3., 8.);
let r: f32x4 = transmute(vmaxq_f32(transmute(a), transmute(b)));
assert_eq!(r, e);
}
}

#[cfg(test)]
Expand Down

0 comments on commit e451716

Please sign in to comment.