From 30d9ed95bb17c9e9092a7fa27dca049d34a9fe3b Mon Sep 17 00:00:00 2001
From: sin3point14 <manaschaudhary2000@gmail.com>
Date: Thu, 22 May 2025 03:38:01 +0530
Subject: [PATCH] Add avx512 checks

---
 cmake/find/FindAVX.cmake | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/cmake/find/FindAVX.cmake b/cmake/find/FindAVX.cmake
index 1f4b24b..c457274 100644
--- a/cmake/find/FindAVX.cmake
+++ b/cmake/find/FindAVX.cmake
@@ -116,8 +116,14 @@ if(CMAKE_COMPILER_IS_GNUCC OR CMAKE_COMPILER_IS_GNUCXX OR CMAKE_CXX_COMPILER_ID
   elseif(DETECTED_AVX_10)
       SET(AVX_FLAGS "${AVX_FLAGS} -mavx")
   endif()
-  # TODO: Check for AVX512 support
-  SET(AVX_FLAGS "${AVX_FLAGS} -mno-avx512f -mno-avx512pf -mno-avx512er -mno-avx512cd")
+  # Only add -mno-avx512* if the compiler accepts them
+  foreach(flag -mno-avx512f -mno-avx512pf -mno-avx512er -mno-avx512cd)
+    string(REPLACE "-" "_" safe_flag ${flag})
+    check_cxx_compiler_flag("${flag}" HAS_FLAG_${safe_flag})
+    if(HAS_FLAG_${safe_flag})
+      set(AVX_FLAGS "${AVX_FLAGS} ${flag}")
+    endif()
+  endforeach()
 elseif(CMAKE_CXX_COMPILER_ID MATCHES "Intel")
   set(AVX_FLAGS "-xHost")
 elseif(MSVC)