Finish avx-512 support by bundling safe_arch upgrade

Author: HadrienG2

src/arch/x86_family.rs

@@ -15,6 +15,8 @@ use target_arch::__m256i;
 use target_arch::{__m128d, __m128i};
 #[cfg(any(target_feature = "avx", doc))]
 use target_arch::{__m256, __m256d};
+#[cfg(any(target_feature = "avx512f", doc))]
+use target_arch::{__m512, __m512d, __m512i};
 
 pessimize_asm_values!(allow(missing_docs) { reg_byte: (i8, u8), reg: (i16, u16, i32, u32, isize, usize) });
 
@@ -93,8 +95,8 @@ pessimize_asm_values!(
 );
 
 /// AVX-512 specific functionality
-#[doc(cfg(target_feature = "avx512f"))]
 #[cfg(any(target_feature = "avx512f", doc))]
+#[cfg_attr(feature = "nightly", doc(cfg(target_feature = "avx512f")))]
 pub mod avx512 {
     use super::*;
     use crate::Pessimize;
@@ -146,7 +148,7 @@ pub mod avx512 {
             $(
                 // This is one of the primitive Pessimize impls on which
                 // the PessimizeCast/BorrowPessimize stack is built
-                #[$doc_cfg]
+                #[cfg_attr(feature = "nightly", $doc_cfg)]
                 unsafe impl Pessimize for Mask<$mask_impl> {
                     #[inline]
                     fn hide(mut self) -> Self {
@@ -200,6 +202,8 @@ mod safe_arch_types {
     use safe_arch::{m128d, m128i};
     #[cfg(any(target_feature = "avx", doc))]
     use safe_arch::{m256, m256d};
+    #[cfg(any(target_feature = "avx512f", doc))]
+    use safe_arch::{m512, m512d, m512i};
 
     #[cfg(any(target_feature = "sse", doc))]
     pessimize_newtypes!(
@@ -230,6 +234,16 @@ mod safe_arch_types {
         doc(cfg(all(feature = "safe_arch", target_feature = "avx2")))
         { m256i{ __m256i } }
     );
+
+    #[cfg(any(target_feature = "avx512f", doc))]
+    pessimize_newtypes!(
+        doc(cfg(all(feature = "safe_arch", target_feature = "avx512f")))
+        {
+            m512{ __m512 },
+            m512d{ __m512d },
+            m512i{ __m512i }
+        }
+    );
 }
 
 // Support portable_simd if enabled
@@ -725,15 +739,18 @@ mod tests {
         }
     }
 
-    // This is nightly-only even though the rest of avx512 is not nightly only
-    // anymore because we can't easily test without safe_arch or portable_simd
-    // and portable_simd doesn't have support for AVX-512 yet.
-    #[cfg(all(feature = "nightly", target_feature = "avx512f"))]
-    mod avx512 {
-        use super::*;
-
-        #[test]
-        fn avx512f() {
+    #[cfg(target_feature = "avx512f")]
+    #[test]
+    fn avx512f() {
+        use safe_arch::{m512, m512d, m512i};
+        test_simd::<i32, 16, m512i>(i32::MIN, i32::MAX);
+        test_simd::<u32, 16, m512i>(u32::MIN, u32::MAX);
+        test_simd::<f32, 16, m512>(f32::MIN, f32::MAX);
+        test_simd::<i64, 8, m512i>(i64::MIN, i64::MAX);
+        test_simd::<u64, 8, m512i>(u64::MIN, u64::MAX);
+        test_simd::<f64, 8, m512d>(f64::MIN, f64::MAX);
+        #[cfg(feature = "nightly")]
+        {
             portable_simd_tests!(
                 (i32, 16),
                 (u32, 16),
@@ -743,21 +760,21 @@ mod tests {
                 (f64, 8)
             );
             portable_mask_tests!((i32, 16), (i64, 8));
-            #[cfg(target_arch = "x86")]
-            {
-                portable_simd_tests!((isize, 16), (usize, 16));
-                portable_mask_tests!((isize, 16));
-            }
-            #[cfg(target_arch = "x86_64")]
-            {
-                portable_simd_tests!((isize, 8), (usize, 8));
-                portable_mask_tests!((isize, 8));
-            }
+            portable_simd_tests!((isize, 8), (usize, 8));
+            portable_mask_tests!((isize, 8));
         }
+    }
 
-        #[test]
-        #[ignore]
-        fn avx512f_optim() {
+    #[cfg(target_feature = "avx512f")]
+    #[test]
+    #[ignore]
+    fn avx512f_optim() {
+        use safe_arch::{m512, m512d, m512i};
+        test_unoptimized_value_type::<m512>();
+        test_unoptimized_value_type::<m512d>();
+        test_unoptimized_value_type::<m512i>();
+        #[cfg(feature = "nightly")]
+        {
             portable_simd_tests_optim!(
                 (i32, 16),
                 (u32, 16),
@@ -767,26 +784,32 @@ mod tests {
                 (f64, 8)
             );
             portable_mask_tests_optim!((i32, 16), (i64, 8));
-            #[cfg(target_arch = "x86")]
-            {
-                portable_simd_tests_optim!((isize, 16), (usize, 16));
-                portable_mask_tests_optim!((isize, 16));
-            }
-            #[cfg(target_arch = "x86_64")]
-            {
-                portable_simd_tests_optim!((isize, 8), (usize, 8));
-                portable_mask_tests_optim!((isize, 8));
-            }
+            portable_simd_tests_optim!((isize, 8), (usize, 8));
+            portable_mask_tests_optim!((isize, 8));
         }
+    }
 
-        #[cfg(target_feature = "avx512bw")]
-        #[test]
-        fn avx512bw() {
+    #[cfg(target_feature = "avx512bw")]
+    #[test]
+    fn avx512bw() {
+        test_simd::<i8, 64, m512i>(i8::MIN, i8::MAX);
+        test_simd::<u8, 64, m512i>(u8::MIN, u8::MAX);
+        test_simd::<i16, 32, m512i>(i16::MIN, i16::MAX);
+        test_simd::<u16, 32, m512i>(u16::MIN, u16::MAX);
+        #[cfg(feature = "nightly")]
+        {
             portable_simd_tests!((i8, 64), (u8, 64), (i16, 32), (u16, 32));
             portable_mask_tests!((i8, 64), (i16, 32));
         }
+    }
+
+    // This is nightly-only even though the rest of avx512 is not nightly only
+    // anymore because we can't easily test without safe_arch or portable_simd
+    // and portable_simd doesn't have support for AVX-512 yet.
+    #[cfg(all(feature = "nightly", target_feature = "avx512bw"))]
+    mod avx512 {
+        use super::*;
 
-        #[cfg(target_feature = "avx512bw")]
         #[test]
         #[ignore]
         fn avx512bw_optim() {