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

Remove scalar Sum/Product over questionable order of operations

Browse Source
This commit is contained in:
Caleb Zulawski 2021-06-21 21:08:26 +00:00
parent 7b66032ed5
commit 708ae61841
1 changed files with 13 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
crates/core_simd/src/iter.rs
View File

@ -1,5 +1,5 @@
macro_rules! impl_traits {
{ $type:ident, $scalar:ty } => {
{ $type:ident } => {
impl<const LANES: usize> core::iter::Sum<Self> for crate::$type<LANES>
where
Self: crate::LanesAtMost32,
@ -18,24 +18,6 @@ fn product<I: core::iter::Iterator<Item = Self>>(iter: I) -> Self {
}
}
impl<const LANES: usize> core::iter::Sum<crate::$type<LANES>> for $scalar
where
crate::$type<LANES>: crate::LanesAtMost32,
{
fn sum<I: core::iter::Iterator<Item = crate::$type<LANES>>>(iter: I) -> Self {
iter.sum::<crate::$type<LANES>>().horizontal_sum()
}
}
impl<const LANES: usize> core::iter::Product<crate::$type<LANES>> for $scalar
where
crate::$type<LANES>: crate::LanesAtMost32,
{
fn product<I: core::iter::Iterator<Item = crate::$type<LANES>>>(iter: I) -> Self {
iter.product::<crate::$type<LANES>>().horizontal_product()
}
}
impl<'a, const LANES: usize> core::iter::Sum<&'a Self> for crate::$type<LANES>
where
Self: crate::LanesAtMost32,
@ -53,36 +35,18 @@ fn product<I: core::iter::Iterator<Item = &'a Self>>(iter: I) -> Self {
iter.fold(Default::default(), core::ops::Mul::mul)
}
}
impl<'a, const LANES: usize> core::iter::Sum<&'a crate::$type<LANES>> for $scalar
where
crate::$type<LANES>: crate::LanesAtMost32,
{
fn sum<I: core::iter::Iterator<Item = &'a crate::$type<LANES>>>(iter: I) -> Self {
iter.sum::<crate::$type<LANES>>().horizontal_sum()
}
}
impl<'a, const LANES: usize> core::iter::Product<&'a crate::$type<LANES>> for $scalar
where
crate::$type<LANES>: crate::LanesAtMost32,
{
fn product<I: core::iter::Iterator<Item = &'a crate::$type<LANES>>>(iter: I) -> Self {
iter.product::<crate::$type<LANES>>().horizontal_product()
}
}
}
}
impl_traits! { SimdF32, f32 }
impl_traits! { SimdF64, f64 }
impl_traits! { SimdU8, u8 }
impl_traits! { SimdU16, u16 }
impl_traits! { SimdU32, u32 }
impl_traits! { SimdU64, u64 }
impl_traits! { SimdUsize, usize }
impl_traits! { SimdI8, i8 }
impl_traits! { SimdI16, i16 }
impl_traits! { SimdI32, i32 }
impl_traits! { SimdI64, i64 }
impl_traits! { SimdIsize, isize }
impl_traits! { SimdF32 }
impl_traits! { SimdF64 }
impl_traits! { SimdU8 }
impl_traits! { SimdU16 }
impl_traits! { SimdU32 }
impl_traits! { SimdU64 }
impl_traits! { SimdUsize }
impl_traits! { SimdI8 }
impl_traits! { SimdI16 }
impl_traits! { SimdI32 }
impl_traits! { SimdI64 }
impl_traits! { SimdIsize }