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

Added separate define for __get_cpuid

Browse Source
This commit is contained in:
Andrew Kane
2024-05-07 16:55:21 -07:00
parent 9894ca3e4e
commit 9b269e2612
3 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/bitutils.c
View File

@@ -11,7 +11,7 @@
#ifdef BIT_DISPATCH
#include <immintrin.h>
#if defined(HAVE__GET_CPUID)
#if defined(USE__GET_CPUID)
#include <cpuid.h>
#else
#include <intrin.h>
@@ -173,7 +173,7 @@ SupportsAvx512Popcount()
{
unsigned int exx[4] = {0, 0, 0, 0};
#if defined(HAVE__GET_CPUID)
#if defined(USE__GET_CPUID)
__get_cpuid(1, &exx[0], &exx[1], &exx[2], &exx[3]);
#else
__cpuid(exx, 1);
@@ -187,7 +187,7 @@ SupportsAvx512Popcount()
if ((_xgetbv(0) & 0xe6) != 0xe6)
return false;
#if defined(HAVE__GET_CPUID)
#if defined(USE__GET_CPUID)
__get_cpuid_count(7, 0, &exx[0], &exx[1], &exx[2], &exx[3]);
#else
__cpuidex(exx, 7, 0);

4
src/halfutils.c
View File

@@ -6,7 +6,7 @@
#ifdef HALFVEC_DISPATCH
#include <immintrin.h>
#if defined(HAVE__GET_CPUID)
#if defined(USE__GET_CPUID)
#include <cpuid.h>
#else
#include <intrin.h>
@@ -254,7 +254,7 @@ SupportsCpuFeature(unsigned int feature)
{
unsigned int exx[4] = {0, 0, 0, 0};
#if defined(HAVE__GET_CPUID)
#if defined(USE__GET_CPUID)
__get_cpuid(1, &exx[0], &exx[1], &exx[2], &exx[3]);
#else
__cpuid(exx, 1);

4
src/halfvec.h
View File

@@ -24,8 +24,8 @@
#endif
/* Needed for cross-compiling / universal binaries on Mac */
#if defined(USE_DISPATCH) && !defined(HAVE__GET_CPUID) && defined(__APPLE_CC__)
#define HAVE__GET_CPUID
#if defined(USE_DISPATCH) && (defined(HAVE__GET_CPUID) || defined(__APPLE_CC__))
#define USE__GET_CPUID
#endif
#if defined(USE_DISPATCH)