From e9d5a0b1aa26103b8cc7e39d4a53ba0ed199b85b Mon Sep 17 00:00:00 2001
From: Andrew Kane <andrew@ankane.org>
Date: Wed, 17 Apr 2024 18:19:03 -0700
Subject: [PATCH] Improved check [skip ci]

---
 src/halfutils.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/halfutils.c b/src/halfutils.c
index a540168..86babbf 100644
--- a/src/halfutils.c
+++ b/src/halfutils.c
@@ -195,6 +195,7 @@ HalfvecCosineSimilarityF16cFma(int dim, half * ax, half * bx)
 #ifdef HALFVEC_DISPATCH
 #define CPU_FEATURE_FMA     (1 << 12)
 #define CPU_FEATURE_OSXSAVE (1 << 27)
+#define CPU_FEATURE_AVX     (1 << 28)
 #define CPU_FEATURE_F16C    (1 << 29)
 
 #ifdef _MSC_VER
@@ -236,7 +237,7 @@ HalfvecInit(void)
 	HalfvecCosineSimilarity = HalfvecCosineSimilarityDefault;
 
 #ifdef HALFVEC_DISPATCH
-	if (SupportsCpuFeature(CPU_FEATURE_FMA | CPU_FEATURE_F16C))
+	if (SupportsCpuFeature(CPU_FEATURE_AVX | CPU_FEATURE_FMA | CPU_FEATURE_F16C))
 	{
 		HalfvecL2SquaredDistance = HalfvecL2SquaredDistanceF16cFma;
 		HalfvecInnerProduct = HalfvecInnerProductF16cFma;