Fix issue #175

saleyn · Mar 29, 2024 · 2e151f2 · 2e151f2
1 parent 5150d2b
commit 2e151f2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/c_src/Makefile b/c_src/Makefile
@@ -70,8 +70,14 @@ HAVE_FORMAT ?= $(C20_FEATURES)
 HAVE_SRCLOC ?= $(C20_FEATURES)
 
 ifeq ($(UNAME_SYS),linux)
+  ifeq ($(shell gcc -march=native -dM -E - </dev/null | awk '/__AVX2__/ { print $$3 }'),1)
+    CXXFLAGS += -mavx2
+  else ifeq ($(shell gcc -march=native -dM -E - </dev/null | awk '/__AVX__/ { print $$3 }'),1)
+    CXXFLAGS += -mavx
+  endif
+
   SYSROOT  := $(shell $(CXX) -print-sysroot)
-  CXXFLAGS += -mavx2 -DHAVE_SETRESUID -DHAVE_PIPE2 $(X64)
+  CXXFLAGS += -DHAVE_SETRESUID -DHAVE_PIPE2 $(X64)
   # Check for Linux capability API (Install package: libcap-devel)
   ifneq ($(wildcard $(SYSROOT)/usr/include/sys/capability.h),)
     CXXFLAGS += -DHAVE_CAP