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

Fix masks

Browse Source
This commit is contained in:
Caleb Zulawski 2020-12-14 00:27:20 -05:00
parent 0ddf7acc89
commit 2720ccc5a7
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
crates/core_simd/src/masks/full_masks/mod.rs
View File

@ -22,7 +22,13 @@ macro_rules! define_mask {
impl<const $lanes: usize> $name<$lanes> {
/// Construct a mask by setting all lanes to the given value.
pub fn splat(value: bool) -> Self {
Self(<$type>::splat(value.into()))
Self(<$type>::splat(
if value {
-1
} else {
0
}
))
}
/// Tests the value of the specified lane.
@ -31,7 +37,7 @@ macro_rules! define_mask {
/// Panics if `lane` is greater than or equal to the number of lanes in the vector.
#[inline]
pub fn test(&self, lane: usize) -> bool {
self.0[lane] > 0
self.0[lane] == -1
}
/// Sets the value of the specified lane.
@ -41,7 +47,7 @@ macro_rules! define_mask {
#[inline]
pub fn set(&mut self, lane: usize, value: bool) {
self.0[lane] = if value {
!0
-1
} else {
0
}
@ -57,7 +63,7 @@ macro_rules! define_mask {
impl<const $lanes: usize> core::convert::TryFrom<$type> for $name<$lanes> {
type Error = TryFromMaskError;
fn try_from(value: $type) -> Result<Self, Self::Error> {
if value.as_slice().iter().all(|x| *x == 0 || !*x == 0) {
if value.as_slice().iter().all(|x| *x == 0 || *x == -1) {
Ok(Self(value))
} else {
Err(TryFromMaskError(()))