rust-lang/rust
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add rounding to integers

Browse Source
This commit is contained in:
Caleb Zulawski
2020-10-11 13:22:26 -04:00
parent ffedbe5b15
commit 75fdacde1c
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
crates/core_simd/src/round.rs
View File

@@ -1,10 +1,12 @@
macro_rules! implement { macro_rules! implement {
{ {
impl $type:ident { impl $type:ident {
int_type = $int_type:ident,
floor = $floor_intrinsic:literal, floor = $floor_intrinsic:literal,
ceil = $ceil_intrinsic:literal, ceil = $ceil_intrinsic:literal,
round = $round_intrinsic:literal, round = $round_intrinsic:literal,
trunc = $trunc_intrinsic:literal, trunc = $trunc_intrinsic:literal,
round_to_int = $round_to_int_intrinsic:literal,
} }
} => { } => {
mod $type { mod $type {
@@ -18,6 +20,8 @@ macro_rules! implement {
fn round_intrinsic(x: crate::$type) -> crate::$type; fn round_intrinsic(x: crate::$type) -> crate::$type;
#[link_name = $trunc_intrinsic] #[link_name = $trunc_intrinsic]
fn trunc_intrinsic(x: crate::$type) -> crate::$type; fn trunc_intrinsic(x: crate::$type) -> crate::$type;
#[link_name = $round_to_int_intrinsic]
fn round_to_int_intrinsic(x: crate::$type) -> crate::$int_type;
} }
impl crate::$type { impl crate::$type {
@@ -55,6 +59,13 @@ macro_rules! implement {
pub fn fract(self) -> Self { pub fn fract(self) -> Self {
self - self.trunc() self - self.trunc()
} }
/// Returns the nearest integer to each lane. Round half-way cases away from 0.0.
#[must_use = "method returns a new vector and does not mutate the original value"]
#[inline]
pub fn round_to_int(self) -> crate::$int_type {
unsafe { round_to_int_intrinsic(self) }
}
} }
} }
} }
@@ -62,63 +73,77 @@ macro_rules! implement {
implement! { implement! {
impl f32x2 { impl f32x2 {
int_type = i32x2,
floor = "llvm.floor.v2f32", floor = "llvm.floor.v2f32",
ceil = "llvm.ceil.v2f32", ceil = "llvm.ceil.v2f32",
round = "llvm.round.v2f32", round = "llvm.round.v2f32",
trunc = "llvm.trunc.v2f32", trunc = "llvm.trunc.v2f32",
round_to_int = "llvm.lround.i32.v2f32",
} }
} }
implement! { implement! {
impl f32x4 { impl f32x4 {
int_type = i32x4,
floor = "llvm.floor.v4f32", floor = "llvm.floor.v4f32",
ceil = "llvm.ceil.v4f32", ceil = "llvm.ceil.v4f32",
round = "llvm.round.v4f32", round = "llvm.round.v4f32",
trunc = "llvm.trunc.v4f32", trunc = "llvm.trunc.v4f32",
round_to_int = "llvm.lround.i32.v4f32",
} }
} }
implement! { implement! {
impl f32x8 { impl f32x8 {
int_type = i32x8,
floor = "llvm.floor.v8f32", floor = "llvm.floor.v8f32",
ceil = "llvm.ceil.v8f32", ceil = "llvm.ceil.v8f32",
round = "llvm.round.v8f32", round = "llvm.round.v8f32",
trunc = "llvm.trunc.v8f32", trunc = "llvm.trunc.v8f32",
round_to_int = "llvm.lround.i32.v8f32",
} }
} }
implement! { implement! {
impl f32x16 { impl f32x16 {
int_type = i32x16,
floor = "llvm.floor.v16f32", floor = "llvm.floor.v16f32",
ceil = "llvm.ceil.v16f32", ceil = "llvm.ceil.v16f32",
round = "llvm.round.v16f32", round = "llvm.round.v16f32",
trunc = "llvm.trunc.v16f32", trunc = "llvm.trunc.v16f32",
round_to_int = "llvm.lround.i32.v16f32",
} }
} }
implement! { implement! {
impl f64x2 { impl f64x2 {
int_type = i64x2,
floor = "llvm.floor.v2f64", floor = "llvm.floor.v2f64",
ceil = "llvm.ceil.v2f64", ceil = "llvm.ceil.v2f64",
round = "llvm.round.v2f64", round = "llvm.round.v2f64",
trunc = "llvm.trunc.v2f64", trunc = "llvm.trunc.v2f64",
round_to_int = "llvm.lround.i64.v2f64",
} }
} }
implement! { implement! {
impl f64x4 { impl f64x4 {
int_type = i64x4,
floor = "llvm.floor.v4f64", floor = "llvm.floor.v4f64",
ceil = "llvm.ceil.v4f64", ceil = "llvm.ceil.v4f64",
round = "llvm.round.v4f64", round = "llvm.round.v4f64",
trunc = "llvm.trunc.v4f64", trunc = "llvm.trunc.v4f64",
round_to_int = "llvm.lround.i64.v4f64",
} }
} }
implement! { implement! {
impl f64x8 { impl f64x8 {
int_type = i64x8,
floor = "llvm.floor.v8f64", floor = "llvm.floor.v8f64",
ceil = "llvm.ceil.v8f64", ceil = "llvm.ceil.v8f64",
round = "llvm.round.v8f64", round = "llvm.round.v8f64",
trunc = "llvm.trunc.v8f64", trunc = "llvm.trunc.v8f64",
round_to_int = "llvm.lround.i64.v8f64",
} }
} }