AVX512 内置函数采用掩码(可以是 -1)。
可移植的内在 _mm512_fmadd_ps (#include <immintrin.h>) 在 GCC9.1 的标头中定义为:
extern __inline __m512
__attribute__ ((__gnu_inline__, __always_inline__, __artificial__))
_mm512_fmadd_ps (__m512 __A, __m512 __B, __m512 __C)
{
return (__m512) __builtin_ia32_vfmaddps512_mask ((__v16sf) __A,
(__v16sf) __B,
(__v16sf) __C,
(__mmask16) -1,
_MM_FROUND_CUR_DIRECTION);
}
我通过在我的系统上查看/usr/lib/gcc/x86_64-pc-linux-gnu/9.1.0/include/avx512*.h 找到了这个。 (不要直接包含那些,仅来自immintrin.h)
IDK 为什么要使用 __builtin_ia32_vfmaddps512_mask 而不是 Intel 的内部函数之一,例如 _mm512_mask_fmadd_ps(合并到第一个操作数)或 _mm512_mask3_fmadd_ps(合并到 +c 操作数)或 _mm512_maskz_fmadd_ps(零屏蔽)。
甚至是完整的_mm512_maskz_fmadd_round_ps,它还允许指定舍入覆盖和屏蔽。
但无论如何,您可以通过这种方式在任何英特尔内部函数(如果有的话)下找到真正的 GCC 内置函数的名称。