/ lib / numpy / distutils / checks / cpu_fma3.c
cpu_fma3.c
 1  #if defined(DETECT_FEATURES) && defined(__INTEL_COMPILER)
 2      /*
 3       * Unlike GCC and CLANG, Intel Compiler exposes all supported intrinsics,
 4       * whether or not the build options for those features are specified.
 5       * Therefore, we must test #definitions of CPU features when option native/host
 6       * is enabled via `--cpu-baseline` or through env var `CFLAGS` otherwise
 7       * the test will be broken and leads to enable all possible features.
 8       */
 9      #if !defined(__FMA__) && !defined(__AVX2__)
10          #error "HOST/ARCH doesn't support FMA3"
11      #endif
12  #endif
13  
14  #include <xmmintrin.h>
15  #include <immintrin.h>
16  
17  int main(int argc, char **argv)
18  {
19      __m256 a = _mm256_loadu_ps((const float*)argv[argc-1]);
20             a = _mm256_fmadd_ps(a, a, a);
21      return (int)_mm_cvtss_f32(_mm256_castps256_ps128(a));
22  }