mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adjust clamp test

Browse Source
This commit is contained in:
Caleb Zulawski 2023-07-22 20:50:49 -04:00
parent 52d6397da7
commit bd4e6616f3
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
crates/core_simd/tests/ops_macros.rs
View File

@ -516,7 +516,7 @@ fn simd_max<const LANES: usize>() {
fn simd_clamp<const LANES: usize>() { fn simd_clamp<const LANES: usize>() {
test_helpers::test_3(&|value: [Scalar; LANES], mut min: [Scalar; LANES], mut max: [Scalar; LANES]| { test_helpers::test_3(&|value: [Scalar; LANES], mut min: [Scalar; LANES], mut max: [Scalar; LANES]| {
use test_helpers::subnormals::FlushSubnormals; use test_helpers::subnormals::flush_in;
for (min, max) in min.iter_mut().zip(max.iter_mut()) { for (min, max) in min.iter_mut().zip(max.iter_mut()) {
if max < min { if max < min {
core::mem::swap(min, max); core::mem::swap(min, max);
@ -535,13 +535,14 @@ fn simd_clamp<const LANES: usize>() {
} }
let mut result_scalar_flush = [Scalar::default(); LANES]; let mut result_scalar_flush = [Scalar::default(); LANES];
for i in 0..LANES { for i in 0..LANES {
result_scalar_flush[i] = value[i]; let mut value = flush_in(value[i]);
if FlushSubnormals::flush(value[i]) < FlushSubnormals::flush(min[i]) { if value < flush_in(min[i]) {
result_scalar_flush[i] = min[i]; value = min[i];
} }
if FlushSubnormals::flush(value[i]) > FlushSubnormals::flush(max[i]) { if value > flush_in(max[i]) {
result_scalar_flush[i] = max[i]; value = max[i];
} }
result_scalar_flush[i] = value
} }
let result_vector = Vector::from_array(value).simd_clamp(min.into(), max.into()).to_array(); let result_vector = Vector::from_array(value).simd_clamp(min.into(), max.into()).to_array();
test_helpers::prop_assert_biteq!(result_vector, result_scalar, result_scalar_flush); test_helpers::prop_assert_biteq!(result_vector, result_scalar, result_scalar_flush);