olg gcc guard

Author: martindurant

.github/workflows/test_wheel.yaml

@@ -9,7 +9,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-22.04, windows-2022]
+        os: [ubuntu-24.04, windows-2022]
         architecture: ["x64"]
         linux_archs: ["native"]
         numpy_version: ["numpy#latest"]

setup.py

@@ -43,7 +43,15 @@ def fix_exts(sources):
 
     import platform
     _machine = platform.machine()
-    _march = ['-march=x86-64-v2'] if _machine == 'x86_64' else []
+    if _machine == 'x86_64':
+        # x86-64-v2 requires GCC >= 11; probe before using it
+        probe = subprocess.run(
+            ['cc', '-march=x86-64-v2', '-x', 'c', '-', '-fsyntax-only'],
+            input=b'', capture_output=True
+        )
+        _march = ['-march=x86-64-v2'] if probe.returncode == 0 else []
+    else:
+        _march = []
     modules = [
         Extension(mod, fix_exts(sources), extra_compile_args=['-O3'] + _march)
         for mod, sources in modules_to_build.items()]