From 99d6a4c7094422cd530360de405890e716997cb1 Mon Sep 17 00:00:00 2001
From: Ayan Shafqat <ayan.x.shafqat@gmail.com>
Date: Sat, 4 Feb 2023 10:59:59 -0500
Subject: [PATCH] Update wscript for AVX512F support

Accurately determine if AVX512F has been supported by the compiler
---
 wscript | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wscript b/wscript
index e8ef1fcb3a..d25958b06d 100644
--- a/wscript
+++ b/wscript
@@ -521,7 +521,7 @@ int main() { return 0; }''',
             if re.search ('x86_64-w64', str(conf.env['CC'])) is not None:
                 conf.define ('FPU_AVX_FMA_SUPPORT', 1)
         elif conf.env['build_target'] == 'i386' or conf.env['build_target'] == 'i686' or conf.env['build_target'] == 'x86_64':
-            conf.check_cxx(fragment = "#include <immintrin.h>\nint main(void) { __m128 a; _mm_fmadd_ss(a, a, a); return 0; }\n",
+            conf.check_cxx(fragment = "#include <immintrin.h>\nint main(void) { __m512 a; _mm512_fmadd_ps(a, a, a); return 0; }\n",
                            features  = ['cxx'],
                            cxxflags  = [ conf.env['compiler_flags_dict']['avx512f'], conf.env['compiler_flags_dict']['fma'], conf.env['compiler_flags_dict']['avx'] ],
                            mandatory = False,