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

Improved code for feature detection

Browse Source
This commit is contained in:
Andrew Kane
2024-04-08 14:40:12 -07:00
parent 862f17c1de
commit 3eba34e5e3
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/halfutils.c
View File

@@ -118,8 +118,11 @@ HalfvecInnerProductF16cFma(int dim, half * ax, half * bx)
#endif #endif
#ifdef HALFVEC_DISPATCH #ifdef HALFVEC_DISPATCH
#define FEATURE_FMA (1 << 12)
#define FEATURE_F16C (1 << 29)
static bool static bool
F16cFmaAvailable() SupportsFeature(unsigned int feature)
{ {
unsigned int exx[4] = {0, 0, 0, 0}; unsigned int exx[4] = {0, 0, 0, 0};
@@ -129,8 +132,7 @@ F16cFmaAvailable()
__cpuid(exx, 1); __cpuid(exx, 1);
#endif #endif
/* FMA = 12, F16C = 29 */ return (exx[2] & feature) == feature;
return (exx[2] & (1 << 12)) != 0 && (exx[2] & (1 << 29)) != 0;
} }
#endif #endif
@@ -145,7 +147,7 @@ HalfvecInit(void)
HalfvecInnerProduct = HalfvecInnerProductDefault; HalfvecInnerProduct = HalfvecInnerProductDefault;
#ifdef HALFVEC_DISPATCH #ifdef HALFVEC_DISPATCH
if (F16cFmaAvailable()) if (SupportsFeature(FEATURE_FMA | FEATURE_F16C))
{ {
HalfvecL2DistanceSquared = HalfvecL2DistanceSquaredF16cFma; HalfvecL2DistanceSquared = HalfvecL2DistanceSquaredF16cFma;
HalfvecInnerProduct = HalfvecInnerProductF16cFma; HalfvecInnerProduct = HalfvecInnerProductF16cFma;