[NeoMathEngine] AVX512 alignment

favorart · favorart · commit 28f5d52075d4 · 2024-01-18T23:38:27.000+01:00
Signed-off-by: Kirill Golikov &lt;kirill.golikov@abbyy.com&gt;
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 );
 

Original file line number	Diff line number	Diff line change
`@@ -149,6 +149,9 @@ struct CCPUInfo final {`
`149`	`149`	`#error "Platform isn't supported!"`
`150`	`150`	`#endif // ERROR FINE_PLATFORM`
`151`	`151`	`}`
	`152`	`+ if( HasAvx512And2Fma ) {`
	`153`	`+ floatAlignment = 16;`
	`154`	`+ }`
`152`	`155`
`153`	`156`	`return floatAlignment;`
`154`	`157`	`#endif // !NEOML_USE_NEON`