1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
//! Vertical (lane-wise) vector comparisons returning vector masks.
macro_rules! impl_cmp_vertical {
(
[$elem_ty:ident; $elem_count:expr]:
$id:ident,
$mask_ty:ident,
$is_mask:expr,($true:expr, $false:expr) | $test_tt:tt
) => {
impl $id {
/// Lane-wise equality comparison.
#[inline]
pub fn eq(self, other: Self) -> $mask_ty {
use crate::llvm::simd_eq;
Simd(unsafe { simd_eq(self.0, other.0) })
}
/// Lane-wise inequality comparison.
#[inline]
pub fn ne(self, other: Self) -> $mask_ty {
use crate::llvm::simd_ne;
Simd(unsafe { simd_ne(self.0, other.0) })
}
/// Lane-wise less-than comparison.
#[inline]
pub fn lt(self, other: Self) -> $mask_ty {
use crate::llvm::{simd_gt, simd_lt};
if $is_mask {
Simd(unsafe { simd_gt(self.0, other.0) })
} else {
Simd(unsafe { simd_lt(self.0, other.0) })
}
}
/// Lane-wise less-than-or-equals comparison.
#[inline]
pub fn le(self, other: Self) -> $mask_ty {
use crate::llvm::{simd_ge, simd_le};
if $is_mask {
Simd(unsafe { simd_ge(self.0, other.0) })
} else {
Simd(unsafe { simd_le(self.0, other.0) })
}
}
/// Lane-wise greater-than comparison.
#[inline]
pub fn gt(self, other: Self) -> $mask_ty {
use crate::llvm::{simd_gt, simd_lt};
if $is_mask {
Simd(unsafe { simd_lt(self.0, other.0) })
} else {
Simd(unsafe { simd_gt(self.0, other.0) })
}
}
/// Lane-wise greater-than-or-equals comparison.
#[inline]
pub fn ge(self, other: Self) -> $mask_ty {
use crate::llvm::{simd_ge, simd_le};
if $is_mask {
Simd(unsafe { simd_le(self.0, other.0) })
} else {
Simd(unsafe { simd_ge(self.0, other.0) })
}
}
}
test_if!{
$test_tt:
paste::item! {
pub mod [<$id _cmp_vertical>] {
use super::*;
#[cfg_attr(not(target_arch = "wasm32"), test)] #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
fn cmp() {
let a = $id::splat($false);
let b = $id::splat($true);
let r = a.lt(b);
let e = $mask_ty::splat(true);
assert!(r == e);
let r = a.le(b);
assert!(r == e);
let e = $mask_ty::splat(false);
let r = a.gt(b);
assert!(r == e);
let r = a.ge(b);
assert!(r == e);
let r = a.eq(b);
assert!(r == e);
let mut a = a;
let mut b = b;
let mut e = e;
for i in 0..$id::lanes() {
if i % 2 == 0 {
a = a.replace(i, $false);
b = b.replace(i, $true);
e = e.replace(i, true);
} else {
a = a.replace(i, $true);
b = b.replace(i, $false);
e = e.replace(i, false);
}
}
let r = a.lt(b);
assert!(r == e);
}
}
}
}
};
}