liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-15 00:56:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improved check [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-17 18:16:58 -07:00
parent c5372e493b
commit 52a81b3b9d
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/halfutils.c
View File

@@ -19,6 +19,7 @@
#endif #endif
#endif #endif
float (*HalfvecL2SquaredDistance) (int dim, half * ax, half * bx); float (*HalfvecL2SquaredDistance) (int dim, half * ax, half * bx);
float (*HalfvecInnerProduct) (int dim, half * ax, half * bx); float (*HalfvecInnerProduct) (int dim, half * ax, half * bx);
double (*HalfvecCosineSimilarity) (int dim, half * ax, half * bx); double (*HalfvecCosineSimilarity) (int dim, half * ax, half * bx);
@@ -196,7 +197,13 @@ HalfvecCosineSimilarityF16cFma(int dim, half * ax, half * bx)
#define CPU_FEATURE_OSXSAVE (1 << 27) #define CPU_FEATURE_OSXSAVE (1 << 27)
#define CPU_FEATURE_F16C (1 << 29) #define CPU_FEATURE_F16C (1 << 29)
static bool #ifdef _MSC_VER
#define TARGET_XSAVE
#else
#define TARGET_XSAVE __attribute__((target("xsave")))
#endif
TARGET_XSAVE static bool
SupportsCpuFeature(unsigned int feature) SupportsCpuFeature(unsigned int feature)
{ {
unsigned int exx[4] = {0, 0, 0, 0}; unsigned int exx[4] = {0, 0, 0, 0};