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

Improved consistency of CPU dispatching code

Browse Source
This commit is contained in:
Andrew Kane
2024-04-18 14:13:12 -07:00
parent fb3c964ac2
commit 4e093f95be
3 changed files with 17 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/bitutils.h
View File

@@ -2,19 +2,24 @@
#define BITUTILS_H #define BITUTILS_H
/* Only enable for more recent compilers */ /* Only enable for more recent compilers */
/* TODO Move to better place */
#if defined(__x86_64__) && defined(__GNUC__) && __GNUC__ >= 8 #if defined(__x86_64__) && defined(__GNUC__) && __GNUC__ >= 8
#define BIT_DISPATCH #define USE_DISPATCH
#elif defined(__x86_64__) && defined(__clang_major__) && __clang_major__ >= 7 #elif defined(__x86_64__) && defined(__clang_major__) && __clang_major__ >= 7
#define BIT_DISPATCH #define USE_DISPATCH
#elif defined(_M_AMD64) && defined(_MSC_VER) && _MSC_VER >= 1920 #elif defined(_M_AMD64) && defined(_MSC_VER) && _MSC_VER >= 1920
#define BIT_DISPATCH #define USE_DISPATCH
#endif #endif
/* target_clones requires glibc */ /* target_clones requires glibc */
#if defined(BIT_DISPATCH) && defined(__gnu_linux__) #if defined(USE_DISPATCH) && defined(__gnu_linux__)
#define USE_TARGET_CLONES #define USE_TARGET_CLONES
#endif #endif
#if defined(USE_DISPATCH)
#define BIT_DISPATCH
#endif
extern uint64 (*BitHammingDistance) (uint32 bytes, unsigned char *ax, unsigned char *bx, uint64 distance); extern uint64 (*BitHammingDistance) (uint32 bytes, unsigned char *ax, unsigned char *bx, uint64 distance);
extern double (*BitJaccardDistance) (uint32 bytes, unsigned char *ax, unsigned char *bx, uint64 ab, uint64 aa, uint64 bb); extern double (*BitJaccardDistance) (uint32 bytes, unsigned char *ax, unsigned char *bx, uint64 ab, uint64 aa, uint64 bb);

3
src/halfvec.h
View File

@@ -5,10 +5,11 @@
#include <float.h> #include <float.h>
#include "bitutils.h"
#include "fmgr.h" #include "fmgr.h"
#include "vector.h" #include "vector.h"
#if defined(__x86_64__) || defined(_M_AMD64) #if defined(USE_DISPATCH)
#define HALFVEC_DISPATCH #define HALFVEC_DISPATCH
#endif #endif

20
src/vector.c
View File

@@ -34,18 +34,10 @@
#define STATE_DIMS(x) (ARR_DIMS(x)[0] - 1) #define STATE_DIMS(x) (ARR_DIMS(x)[0] - 1)
#define CreateStateDatums(dim) palloc(sizeof(Datum) * (dim + 1)) #define CreateStateDatums(dim) palloc(sizeof(Datum) * (dim + 1))
/* target_clones requires glibc */ #if defined(USE_TARGET_CLONES) && !defined(__FMA__)
#if defined(__gnu_linux__) && defined(__has_attribute) #define VECTOR_TARGET_CLONES __attribute__((target_clones("default", "fma")))
/* Use separate line for portability */
#if __has_attribute(target_clones)
#define HAVE_TARGET_CLONES
#endif
#endif
#if defined(__x86_64__) && defined(HAVE_TARGET_CLONES) && !defined(__FMA__)
#define VECTOR_DISPATCH __attribute__((target_clones("default", "fma")))
#else #else
#define VECTOR_DISPATCH #define VECTOR_TARGET_CLONES
#endif #endif
PG_MODULE_MAGIC; PG_MODULE_MAGIC;
@@ -573,7 +565,7 @@ halfvec_to_vector(PG_FUNCTION_ARGS)
PG_RETURN_POINTER(result); PG_RETURN_POINTER(result);
} }
VECTOR_DISPATCH static float VECTOR_TARGET_CLONES static float
VectorL2SquaredDistance(int dim, float *ax, float *bx) VectorL2SquaredDistance(int dim, float *ax, float *bx)
{ {
float distance = 0.0; float distance = 0.0;
@@ -620,7 +612,7 @@ vector_l2_squared_distance(PG_FUNCTION_ARGS)
PG_RETURN_FLOAT8((double) VectorL2SquaredDistance(a->dim, a->x, b->x)); PG_RETURN_FLOAT8((double) VectorL2SquaredDistance(a->dim, a->x, b->x));
} }
VECTOR_DISPATCH static float VECTOR_TARGET_CLONES static float
VectorInnerProduct(int dim, float *ax, float *bx) VectorInnerProduct(int dim, float *ax, float *bx)
{ {
float distance = 0.0; float distance = 0.0;
@@ -662,7 +654,7 @@ vector_negative_inner_product(PG_FUNCTION_ARGS)
PG_RETURN_FLOAT8((double) -VectorInnerProduct(a->dim, a->x, b->x)); PG_RETURN_FLOAT8((double) -VectorInnerProduct(a->dim, a->x, b->x));
} }
VECTOR_DISPATCH static double VECTOR_TARGET_CLONES static double
VectorCosineSimilarity(int dim, float *ax, float *bx) VectorCosineSimilarity(int dim, float *ax, float *bx)
{ {
float similarity = 0.0; float similarity = 0.0;