MSVC多年来支持AVX / AVX2指令,根据
this msdn blog post,它可以自动生成
fused-multiply-add (FMA)指令.
但是以下两个函数都没有编译为FMA指令：
float func1(float x,float y,float z)
{
return x * y + z;
}
float func2(float x,float z)
{
return std::fma(x,y,z);
}
更糟糕的是,std :: fma没有实现为单个FMA指令,它执行速度非常快,比普通的x * yz慢得多(如果实现不依赖于FMA指令,则预期std :: fma的性能很差) .
我用/ arch编译：AVX2 / O2 / Qvec标志.
还尝试用/ fp：快速,没有成功.
所以问题是MSVC如何强制自动发出FMA指令？
UPDATE
有一个#pragma fp_contract (on|off),(看起来像)什么都不做.
解决方法
MSVC 2015确实为标量操作生成fma指令,但不为向量操作生成(除非您明确使用fma内在函数).
我编译了以下代码
//foo.cpp
float mul_add(float a,float b,float c) {
return a*b + c;
}
//MSVC cannot handle vectors as function parameters so use const references
__m256 mul_addv(__m256 const &a,__m256 const &b,__m256 const &c) {
return _mm256_add_ps(_mm256_mul_ps(a,b),c);
}
同
cl /c /O2 /arch:AVX2 /fp:fast /FA foo.cpp
在MSVC2015中它产生了以下组件
;mul_add vmovaps xmm3,xmm1 vfmadd213ss xmm3,xmm0,xmm2 vmovaps xmm0,xmm3
和
;mul_addv vmovups ymm0,YMMWORD PTR [rcx] vmulps ymm1,ymm0,YMMWORD PTR [rdx] vaddps ymm0,ymm1,YMMWORD PTR [r8]
