AVX512: _mm512_rsqrt28_ps is available for AVX512ER only

ggael · ggael · commit 40b4bf3d3226 · 2018-04-03T14:36:27.000+02:00
diff --git a/Eigen/Core b/Eigen/Core
@@ -183,6 +183,9 @@
       #ifdef __AVX512DQ__
         #define EIGEN_VECTORIZE_AVX512DQ
       #endif
+      #ifdef __AVX512ER__
+        #define EIGEN_VECTORIZE_AVX512ER
+      #endif
     #endif
 
     // include files
diff --git a/Eigen/src/Core/arch/AVX512/MathFunctions.h b/Eigen/src/Core/arch/AVX512/MathFunctions.h
@@ -375,7 +375,7 @@ prsqrt<Packet8d>(const Packet8d& _x) {
   // Insert NaNs and Infs in all the right places.
   return _mm512_mask_blend_pd(le_zero_mask, x, infs_and_nans);
 }
-#else
+#elif defined(EIGEN_VECTORIZE_AVX512ER)
 template <>
 EIGEN_STRONG_INLINE Packet16f prsqrt<Packet16f>(const Packet16f& x) {
   return _mm512_rsqrt28_ps(x);

Original file line number	Diff line number	Diff line change
`@@ -375,7 +375,7 @@ prsqrt<Packet8d>(const Packet8d& _x) {`
`375`	`375`	`// Insert NaNs and Infs in all the right places.`
`376`	`376`	`return _mm512_mask_blend_pd(le_zero_mask, x, infs_and_nans);`
`377`	`377`	`}`
`378`		`-#else`
	`378`	`+#elif defined(EIGEN_VECTORIZE_AVX512ER)`
`379`	`379`	`template <>`
`380`	`380`	`EIGEN_STRONG_INLINE Packet16f prsqrt<Packet16f>(const Packet16f& x) {`
`381`	`381`	`return _mm512_rsqrt28_ps(x);`