Conditionally enable avx2 support

adgear · Jun 3, 2024 · d4ff942 · d4ff942
1 parent 43bfc6d
commit d4ff942
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/c_src/Makefile b/c_src/Makefile
@@ -19,7 +19,7 @@ endif
 
 CC = gcc
 
-GEN_CFLAGS = -finline-functions -mavx2 -fPIC \
+GEN_CFLAGS = -finline-functions -fPIC \
 						 -std=gnu99 -Wall -Wextra -Wno-missing-field-initializers
 
 UNAME_SYS := $(shell uname -s)
@@ -28,6 +28,17 @@ ifeq ($(UNAME_SYS), Darwin)
 	LDFLAGS ?= -flat_namespace -undefined suppress
 endif
 
+HAS_AVX2 := $(shell \
+	if $(CC) -O -mavx2 -o $(BASEDIR)/../test_for_rdrand ./test_for_rdrand.c 2>/dev/null; then \
+		echo 1; \
+	else \
+		echo 0; \
+	fi; \
+)
+ifeq ($(HAS_AVX2), 1)
+	GEN_CFLAGS += -mavx2
+endif
+
 HAS_RDRAND := $(shell \
 	if $(CC) -O -o $(BASEDIR)/../test_for_rdrand ./test_for_rdrand.c 2>/dev/null; then \
 		$(BASEDIR)/../test_for_rdrand && echo 1; \