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

Improved code [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-08 15:32:16 -07:00
parent 191c8e1cca
commit ab382a2053
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/halfutils.c
View File

@@ -118,11 +118,11 @@ HalfvecInnerProductF16cFma(int dim, half * ax, half * bx)
#endif
#ifdef HALFVEC_DISPATCH
#define FEATURE_FMA (1 << 12)
#define FEATURE_F16C (1 << 29)
#define CPU_FEATURE_FMA (1 << 12)
#define CPU_FEATURE_F16C (1 << 29)
static bool
SupportsFeature(unsigned int feature)
SupportsCpuFeature(unsigned int feature)
{
unsigned int exx[4] = {0, 0, 0, 0};
@@ -147,7 +147,7 @@ HalfvecInit(void)
HalfvecInnerProduct = HalfvecInnerProductDefault;
#ifdef HALFVEC_DISPATCH
if (SupportsFeature(FEATURE_FMA | FEATURE_F16C))
if (SupportsCpuFeature(CPU_FEATURE_FMA | CPU_FEATURE_F16C))
{
HalfvecL2SquaredDistance = HalfvecL2SquaredDistanceF16cFma;
HalfvecInnerProduct = HalfvecInnerProductF16cFma;