Я пытаюсь оптимизировать некоторые вычисления матрицы, и мне было интересно узнать, удалось ли обнаружить во время компиляции, если SSE или/или AVX или/и AVX2 или/и AVX-512 включены компилятором? Идеально для GCC и Clang, но я могу справиться только с одним из них.
Я не уверен, что это возможно, и, возможно, я буду использовать свой собственный макрос, но я предпочел бы его обнаружить и попросить пользователя его выбрать.
Большинство компиляторов автоматически определит:
__SSE__
__SSE2__
__SSE3__
__AVX__
__AVX2__
и т.д., в соответствии с любыми передаваемыми вами командами. Вы можете легко проверить это с помощью gcc (или gcc-совместимых компиляторов, таких как clang), например:
$ gcc -msse3 -dM -E - < /dev/null | egrep "SSE|AVX" | sort
#define __SSE__ 1
#define __SSE2__ 1
#define __SSE2_MATH__ 1
#define __SSE3__ 1
#define __SSE_MATH__ 1
или
$ gcc -mavx2 -dM -E - < /dev/null | egrep "SSE|AVX" | sort
#define __AVX__ 1
#define __AVX2__ 1
#define __SSE__ 1
#define __SSE2__ 1
#define __SSE2_MATH__ 1
#define __SSE3__ 1
#define __SSE4_1__ 1
#define __SSE4_2__ 1
#define __SSE_MATH__ 1
#define __SSSE3__ 1
или просто проверить заранее определенные макросы для построения по умолчанию на вашей конкретной платформе:
$ gcc -dM -E - < /dev/null | egrep "SSE|AVX" | sort
#define __SSE2_MATH__ 1
#define __SSE2__ 1
#define __SSE3__ 1
#define __SSE_MATH__ 1
#define __SSE__ 1
#define __SSSE3__ 1
Более свежие процессоры Intel поддерживают AVX-512, который не является монолитным набором команд. Можно увидеть поддержку, доступную от GCC (версия 6.2), для двух примеров ниже.
Вот приземление рыцарей:
$ gcc -march=knl -dM -E - < /dev/null | egrep "SSE|AVX" | sort
#define __AVX__ 1
#define __AVX2__ 1
#define __AVX512CD__ 1
#define __AVX512ER__ 1
#define __AVX512F__ 1
#define __AVX512PF__ 1
#define __SSE__ 1
#define __SSE2__ 1
#define __SSE2_MATH__ 1
#define __SSE3__ 1
#define __SSE4_1__ 1
#define __SSE4_2__ 1
#define __SSE_MATH__ 1
#define __SSSE3__ 1
Вот Skylake AVX-512:
$ gcc -march=skylake-avx512 -dM -E - < /dev/null | egrep "SSE|AVX" | sort
#define __AVX__ 1
#define __AVX2__ 1
#define __AVX512BW__ 1
#define __AVX512CD__ 1
#define __AVX512DQ__ 1
#define __AVX512F__ 1
#define __AVX512VL__ 1
#define __SSE__ 1
#define __SSE2__ 1
#define __SSE2_MATH__ 1
#define __SSE3__ 1
#define __SSE4_1__ 1
#define __SSE4_2__ 1
#define __SSE_MATH__ 1
#define __SSSE3__ 1
Intel раскрыла дополнительные подмножества AVX-512 (см. расширения ISA). GCC (версия 7) поддерживает флаги компилятора и символы препроцессора, связанные с подмножествами 4FMAPS, 4VNNIW, IFMA, VBMI и VPOPCNTDQ AVX-512:
for i in 4fmaps 4vnniw ifma vbmi vpopcntdq ; do echo "==== $i ====" ; gcc -mavx512$i -dM -E - < /dev/null | egrep "AVX512" | sort ; done
==== 4fmaps ====
#define __AVX5124FMAPS__ 1
#define __AVX512F__ 1
==== 4vnniw ====
#define __AVX5124VNNIW__ 1
#define __AVX512F__ 1
==== ifma ====
#define __AVX512F__ 1
#define __AVX512IFMA__ 1
==== vbmi ====
#define __AVX512BW__ 1
#define __AVX512F__ 1
#define __AVX512VBMI__ 1
==== vpopcntdq ====
#define __AVX512F__ 1
#define __AVX512VPOPCNTDQ__ 1