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

Fixed alignment for bit in IVFFlat - fixes #520

Browse Source
This commit is contained in:
Andrew Kane
2024-04-17 00:05:31 -07:00
parent c361f80465
commit cf570810eb
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/ivfbuild.c
View File

@@ -337,14 +337,20 @@ GetMaxDimensions(IvfflatType type)
static Size
GetItemSize(IvfflatType type, int dimensions)
{
Size size;
if (type == IVFFLAT_TYPE_VECTOR)
return VECTOR_SIZE(dimensions);
size = VECTOR_SIZE(dimensions);
else if (type == IVFFLAT_TYPE_HALFVEC)
return HALFVEC_SIZE(dimensions);
size = HALFVEC_SIZE(dimensions);
else if (type == IVFFLAT_TYPE_BIT)
return VARBITTOTALLEN(dimensions);
size = VARBITTOTALLEN(dimensions);
else
elog(ERROR, "Unsupported type");
/* Ensure items are aligned to prevent UB */
/* Only need 4 byte alignment, but prefer MAXALIGN */
return MAXALIGN(size);
}
/*