devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add support for AVX512-FP16 for vectorizing half precision math

Browse Source
This commit is contained in:
Matthew Sterrett
2022-08-17 18:15:21 +00:00
committed by Rasmus Munk Larsen
parent 76a669fb45
commit 7a3b667c43
7 changed files with 985 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Eigen/src/Core/util/ConfigureVectorization.h
View File

@@ -270,6 +270,17 @@
#ifdef __AVX512BF16__
#define EIGEN_VECTORIZE_AVX512BF16
#endif
#ifdef __AVX512FP16__
#ifdef __AVX512VL__
#define EIGEN_VECTORIZE_AVX512FP16
#else
#if EIGEN_COMP_GNUC
#error Please add -mavx512vl to your compiler flags: compiling with -mavx512fp16 alone without AVX512-VL is not supported.
#else
#error Please enable AVX512-VL in your compiler flags (e.g. -mavx512vl): compiling with AVX512-FP16 alone without AVX512-VL is not supported.
#endif
#endif
#endif
#endif
#endif