[NeoMathEngine] AVX512 alignment

Signed-off-by: Kirill Golikov <[email protected]>
neoml-lib · Jan 18, 2024 · 28f5d52 · 28f5d52
1 parent 6d85c03
commit 28f5d52
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/NeoMathEngine/src/CPU/CPUInfo.h b/NeoMathEngine/src/CPU/CPUInfo.h
@@ -149,6 +149,9 @@ struct CCPUInfo final {
 #error "Platform isn't supported!"
 #endif // ERROR FINE_PLATFORM
 		}
+		if( HasAvx512And2Fma ) {
+			floatAlignment = 16;
+		}
 
 		return floatAlignment;
 #endif // !NEOML_USE_NEON

diff --git a/NeoMathEngine/src/CPU/x86/avx512/Avx512Functions.h b/NeoMathEngine/src/CPU/x86/avx512/Avx512Functions.h
@@ -24,7 +24,7 @@ namespace NeoML {
 namespace Avx512 {
 
 // The minimum vector size recommended for using AVX512 vector functions
-static constexpr int VectorMathMinSize = 32;
+static constexpr int VectorMathMinSize = 16;
 
 void dataCopy( float* dst, const float* src, int vectorSize );