liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-06-29 17:21:16 +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
#ifdef HALFVEC_DISPATCH
#define FEATURE_FMA (1 << 12)
#define FEATURE_F16C (1 << 29)
static bool
F16cFmaAvailable()
SupportsFeature(unsigned int feature)
{
unsigned int exx[4] = {0, 0, 0, 0};
@@ -129,8 +132,7 @@ F16cFmaAvailable()
__cpuid(exx, 1);
#endif
/* FMA = 12, F16C = 29 */
return (exx[2] & (1 << 12)) != 0 && (exx[2] & (1 << 29)) != 0;
return (exx[2] & feature) == feature;
}
#endif
@@ -145,7 +147,7 @@ HalfvecInit(void)
HalfvecInnerProduct = HalfvecInnerProductDefault;
#ifdef HALFVEC_DISPATCH
if (F16cFmaAvailable())
if (SupportsFeature(FEATURE_FMA | FEATURE_F16C))
{
HalfvecL2DistanceSquared = HalfvecL2DistanceSquaredF16cFma;
HalfvecInnerProduct = HalfvecInnerProductF16cFma;