FMA3 в GCC: как включить
У меня есть i5-4250U, который имеет AVX2 и FMA3. Я тестирую код плотного матричного умножения в GCC 4.8.1 для Linux, который я написал. Ниже приведен список трех разных способов компиляции.
SSE2: gcc matrix.cpp -o matrix_gcc -O3 -msse2 -fopenmp
AVX: gcc matrix.cpp -o matrix_gcc -O3 -mavx -fopenmp
AVX2+FMA: gcc matrix.cpp -o matrix_gcc -O3 -march=native -fopenmp -ffast-math
Версии SSE2 и AVX явно отличаются по производительности. Тем не менее, AVX2 + FMA не лучше, чем версия AVX. Я не понимаю этого. Я получаю более 80% пиковых провалов процессора, предполагая, что FMA не существует, но я думаю, что у меня получится намного лучше с FMA. Умножение матриц должно извлекать непосредственную выгоду из FMA. Я действительно делаю восемь точечных продуктов одновременно в AVX. Когда я проверяюmarch=native
это дает:
cc -march=native -E -v - </dev/null 2>&1 | grep cc1 | grep fma
...-march=core-avx2 -mavx -mavx2 -mfma -mno-fma4 -msse4.2 -msse4.1 ...
Так что я вижу, что он включен (просто чтобы убедиться, что я добавил-mfma
но это не имеет значения).ffast-math
должна позволять модель с плавающей запятойКак использовать инструкции Fused Multiply-Add (FMA) с SSE / AVX
Редактировать:
Основываясь на комментариях Mysticial, я пошел дальше и использовал _mm256_fmadd_ps, и теперь версия AVX2 + FMA стала быстрее.Я не уверен, почему компилятор не сделает это для меня. Сейчас я получаю около 80 GFLOPS (110% пиковых флопов без FMA) для более 1000x1000 матриц. В случае, если кто-то не доверяет моему расчету пикового флопа, вот что я сделал.
peak flops (no FMA) = frequency * simd_width * ILP * cores
= 2.3GHZ * 8 * 2 * 2 = 73.2 GFLOPS
peak flops (with FMA) = 2 * peak flops (no FMA) = 146.2 GFLOPS
Мой процессор в турбо-режиме при использовании обоих ядер составляет 2,3 ГГц. Я получаю 2 за ILP, потому что Ivy Bridge может выполнять одно умножение AVX и одно сложение AVX одновременно (и я развернул цикл несколько раз, чтобы убедиться в этом).
Я получаю только около 55% пиковых флопов (с FMA). Я не уверен, почему, но, по крайней мере, сейчас я что-то вижу.
Одним из побочных эффектов является то, что теперь я получаю небольшую ошибку при сравнении с простым алгоритмом умножения матриц, которому я доверяю. Я думаю, это связано с тем, что FMA имеет только один режим округления вместо того, что обычно было бы двумя (что по иронии судьбы нарушает правила IEEE с плавающей запятой, даже если это, вероятно, лучше).
Редактировать:
Кто-то должен повторитьКак мне достичь теоретического максимума 4 FLOP за цикл? но сделайте 8 двойных операций с плавающей запятой за цикл с Haswell.
редактировать
Фактически, Mysticial обновил свой проект для поддержки FMA3 (см. Его ответ в ссылке выше). Я запустил его код в Windows8 с MSVC2012 (потому что версия Linux не компилировалась с поддержкой FMA). Вот результаты.
Testing AVX Mul + Add:
Seconds = 22.7417
FP Ops = 768000000000
FLOPs = 3.37705e+010
sum = 17.8122
Testing FMA3 FMA:
Seconds = 22.1389
FP Ops = 1536000000000
FLOPs = 6.938e+010
sum = 333.309
Это 69,38 GFLOPS для FMA3 для двойной плавающей запятой. Для одного с плавающей запятой мне нужно удвоить его, так что это 138,76 SP GFLOPS. Я рассчитываю, что мой пик составляет 146,2 SP GFLOPS.Это 95% от пика! Другими словами, я смогу немного улучшить свой код GEMM (хотя он уже немного быстрее, чем Eigen).