From: Chidi Williams <redacted>
Date: Wed, 9 Nov 2022 00:28:36 +0000 (+0000)
Subject: Check for AVX and AVX2 on Darwin
X-Git-Tag: upstream/1.7.4~1841
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=9e700e18212c63c2c1b935352817dc873e40b4ef;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

Check for AVX and AVX2 on Darwin
---

diff --git a/Makefile b/Makefile
index c20bd551..fb98dcdb 100644
--- a/Makefile
+++ b/Makefile
@@ -50,7 +50,19 @@ endif
 # TODO: probably these flags need to be tweaked on some architectures
 #       feel free to update the Makefile for your architecture and send a pull request or issue
 ifeq ($(UNAME_M),x86_64)
-	CFLAGS += -mavx -mavx2 -mfma -mf16c
+	CFLAGS += -mfma -mf16c
+	ifeq ($(UNAME_S),Darwin)
+		AVX1_M := $(shell sysctl machdep.cpu.features)
+		ifneq (,$(findstring AVX1.0,$(AVX1_M)))
+			CFLAGS += -mavx
+		endif
+		AVX2_M := $(shell sysctl machdep.cpu.leaf7_features)
+		ifneq (,$(findstring AVX2,$(AVX2_M)))
+			CFLAGS += -mavx2
+		endif
+	else
+		CFLAGS += -mavx -mavx2
+	endif
 endif
 ifeq ($(UNAME_M),amd64)
 	CFLAGS += -mavx -mavx2 -mfma -mf16c