Add initial type implementations

2020-09-22 19:41:42 -04:00
parent 8ee9cbaf60
commit 011aafea16
58 changed files with 533 additions and 0 deletions
--- a/crates/core_simd/src/macros.rs
+++ b/crates/core_simd/src/macros.rs
@@ -0,0 +1,140 @@
+macro_rules! from_aligned {
+    { unsafe $from:ty => $to:ty } => {
+        impl core::convert::From<$from> for $to {
+            #[inline]
+            fn from(value: $from) -> $to {
+                assert_eq!(core::mem::size_of::<$from>(), core::mem::size_of::<$to>());
+                assert!(core::mem::align_of::<$from>() >= core::mem::align_of::<$to>());
+                unsafe { core::mem::transmute(value) }
+            }
+        }
+    };
+    { unsafe $a:ty |bidirectional| $b:ty } => {
+        from_aligned!{ unsafe $a => $b }
+        from_aligned!{ unsafe $b => $a }
+    }
+}
+
+macro_rules! from_unaligned {
+    { unsafe $from:ty => $to:ty } => {
+        impl core::convert::From<$from> for $to {
+            #[inline]
+            fn from(value: $from) -> $to {
+                assert_eq!(core::mem::size_of::<$from>(), core::mem::size_of::<$to>());
+                unsafe { (&value as *const $from as *const $to).read_unaligned() }
+            }
+        }
+    }
+}
+
+macro_rules! define_type {
+    { struct $name:ident([$type:ty; $lanes:tt]); } => {
+        define_type! { @impl $name [$type; $lanes] }
+
+        // array references
+        impl AsRef<[$type; $lanes]> for $name {
+            #[inline]
+            fn as_ref(&self) -> &[$type; $lanes] {
+                unsafe { &*(self as *const _ as *const _) }
+            }
+        }
+
+        impl AsMut<[$type; $lanes]> for $name {
+            #[inline]
+            fn as_mut(&mut self) -> &mut [$type; $lanes] {
+                unsafe { &mut *(self as *mut _ as *mut _) }
+            }
+        }
+
+        // slice references
+        impl AsRef<[$type]> for $name {
+            #[inline]
+            fn as_ref(&self) -> &[$type] {
+                AsRef::<[$type; $lanes]>::as_ref(self)
+            }
+        }
+
+        impl AsMut<[$type]> for $name {
+            #[inline]
+            fn as_mut(&mut self) -> &mut [$type] {
+                AsMut::<[$type; $lanes]>::as_mut(self)
+            }
+        }
+
+        // vector to array
+        from_aligned! { unsafe $name => [$type; $lanes] }
+
+        // array to vector
+        from_unaligned! { unsafe [$type; $lanes] => $name }
+
+        // splat
+        impl From<$type> for $name {
+            fn from(value: $type) -> Self {
+                Self::splat(value)
+            }
+        }
+    };
+    { @impl $name:ident [$type:ty; 1] } => {
+        define_type! { @impl $name | $type | $type, | v0, }
+    };
+    { @impl $name:ident [$type:ty; 2] } => {
+        define_type! { @impl $name | $type | $type, $type, | v0, v1, }
+    };
+    { @impl $name:ident [$type:ty; 4] } => {
+        define_type! { @impl $name | $type |
+            $type, $type, $type, $type, |
+            v0, v1, v2, v3,
+        }
+    };
+    { @impl $name:ident [$type:ty; 8] } => {
+        define_type! { @impl $name | $type |
+            $type, $type, $type, $type, $type, $type, $type, $type, |
+            v0, v1, v2, v3, v4, v5, v6, v7,
+        }
+    };
+    { @impl $name:ident [$type:ty; 16] } => {
+        define_type! { @impl $name | $type |
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, |
+            v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, v10, v11, v12, v13, v14, v15,
+        }
+    };
+    { @impl $name:ident [$type:ty; 32] } => {
+        define_type! { @impl $name | $type |
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type,
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, |
+            v0,  v1,  v2,  v3,  v4,  v5,  v6,  v7,  v8,  v9,  v10, v11, v12, v13, v14, v15,
+            v16, v17, v18, v19, v20, v21, v22, v23, v24, v25, v26, v27, v28, v29, v30, v31,
+        }
+    };
+    { @impl $name:ident [$type:ty; 64] } => {
+        define_type! { @impl $name | $type |
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type,
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type,
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type,
+            $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, $type, |
+            v0,  v1,  v2,  v3,  v4,  v5,  v6,  v7,  v8,  v9,  v10, v11, v12, v13, v14, v15,
+            v16, v17, v18, v19, v20, v21, v22, v23, v24, v25, v26, v27, v28, v29, v30, v31,
+            v32, v33, v34, v35, v36, v37, v38, v39, v40, v41, v42, v43, v44, v45, v46, v47,
+            v48, v49, v50, v51, v52, v53, v54, v55, v56, v57, v58, v59, v60, v61, v62, v63,
+        }
+    };
+    { @impl $name:ident | $type:ty | $($itype:ty,)* | $($ivar:ident,)* } => {
+        #[allow(non_camel_case_types)]
+        #[derive(Copy, Clone, Debug, Default, PartialEq, PartialOrd)]
+        #[repr(simd)]
+        pub struct $name($($itype),*);
+
+        impl $name {
+            #[inline]
+            pub fn splat(value: $type) -> Self {
+                Self($(value as $itype),*)
+            }
+
+            #[allow(clippy::too_many_arguments)]
+            #[inline]
+            pub fn new($($ivar: $itype),*) -> Self {
+                Self($($ivar),*)
+            }
+        }
+    }
+}